| 77 | |
| 78 | |
| 79 | def write_args(args, path): |
| 80 | args_dict = dict((name, getattr(args, name)) for name in dir(args) |
| 81 | if not name.startswith('_')) |
| 82 | with open(path, 'a') as args_file: |
| 83 | args_file.write('==> torch version: {}\n'.format(torch.__version__)) |
| 84 | args_file.write( |
| 85 | '==> cudnn version: {}\n'.format(torch.backends.cudnn.version())) |
| 86 | args_file.write('==> Cmd:\n') |
| 87 | args_file.write(str(sys.argv)) |
| 88 | args_file.write('\n==> args:\n') |
| 89 | for k, v in sorted(args_dict.items()): |
| 90 | args_file.write(' %s: %s\n' % (str(k), str(v))) |
| 91 | args_file.close() |
| 92 | |
| 93 | |
| 94 | class Logger(object): |