| 28 | |
| 29 | |
| 30 | def init_logging(rank, models_root): |
| 31 | if rank == 0: |
| 32 | log_root = logging.getLogger() |
| 33 | log_root.setLevel(logging.INFO) |
| 34 | formatter = logging.Formatter("Training: %(asctime)s-%(message)s") |
| 35 | handler_file = logging.FileHandler(os.path.join(models_root, "training.log")) |
| 36 | handler_stream = logging.StreamHandler(sys.stdout) |
| 37 | handler_file.setFormatter(formatter) |
| 38 | handler_stream.setFormatter(formatter) |
| 39 | log_root.addHandler(handler_file) |
| 40 | log_root.addHandler(handler_stream) |
| 41 | log_root.info('rank_id: %d' % rank) |