1. 创建log日志.txt文件并在其中保存option.py中初始化的所有参数
with open(os.path.join(args.save_dir, args.exp_name, args.dataset_name, args.log_dir), "a") as args_log:
for k, v in sorted(vars(args).items()):
print('%s: %s ' % (str(k), str(v)))
args_log.write('%s: %s \n' % (str(k), str(v)))
展示结果:
2. 每次保存训练参数时,再次写入log日志时,新的内容添加在分割线下方
with open(os.path.join(args.save_dir, args.exp_name, args.dataset_name, args.log_dir), "a") as args_log:
# 在新日志条目之前添加一个新行
args_log.write('\n' + '*'*100 + '\n')
for k, v in sorted(vars(args).items()):
print('%s: %s ' % (str(k), str(v)))
args_log.write('%s: %s \n' % (str(k), str(v)))