深度学习简单的结果重定向

之前在训练的时候，有些开源没有配置完善的logger，没有加入tensorboard，只是通过控制台输出的方式显示每个迭代的结果，最近加个简单的结果重定向，仅修改了main的部分，可以直接将本次训练的结果保存在以时间命名的txt格式的日志中。

if __name__ == '__main__':
    torch.manual_seed(args.seed)
    struct_time = time.localtime()
    format_time = time.strftime("%Y-%m-%d-%H:%M:%S", struct_time)
    log_file_name = format_time
    log_file = open(log_file_name + '.txt', 'w')
    oldstdout = sys.stdout
    sys.stdout = log_file
    main(args)
    log_file.close()

足够简单，基本够用。