(rank)
| 301 | |
| 302 | |
| 303 | def _init_logging(rank): |
| 304 | # logging |
| 305 | if rank == 0: |
| 306 | # set format |
| 307 | logging.basicConfig( |
| 308 | level=logging.INFO, |
| 309 | format="[%(asctime)s] %(levelname)s: %(message)s", |
| 310 | handlers=[logging.StreamHandler(stream=sys.stdout)]) |
| 311 | else: |
| 312 | logging.basicConfig(level=logging.ERROR) |
| 313 | |
| 314 | |
| 315 | def generate(args): |