(rank)
| 252 | |
| 253 | |
| 254 | def _init_logging(rank): |
| 255 | # logging |
| 256 | if rank == 0: |
| 257 | # set format |
| 258 | logging.basicConfig( |
| 259 | level=logging.INFO, |
| 260 | format="[%(asctime)s] %(levelname)s: %(message)s", |
| 261 | handlers=[logging.StreamHandler(stream=sys.stdout)]) |
| 262 | else: |
| 263 | logging.basicConfig(level=logging.ERROR) |
| 264 | |
| 265 | |
| 266 | def generate(args): |