3.将模型的参数写入到文件里面

第一步:构造argv参数

第二步:使用vars(argv) 构造字典

第三步:循环vars(argv), 将用f.write() 将键和值进行写入

import argparse, sys, os

def parse_arguments(argv):
    parser = argparse.ArgumentParser()

    parser.add_argument('--log_dir', type=str, help='file dir is open', default='picture/')
    parser.add_argument('--model_dir', type=str, help='model dir is save', default='model/')
    return parser.parse_args(argv)

args = parse_arguments(sys.argv[1:])

log_path = os.path.join(os.path.expanduser(args.log_dir), 'argument.txt')
with open(log_path, 'w') as f:
    for key, value in vars(args).items():
        f.write('%s:%s
'%(key, value))
        print(key, value)
原文地址:https://www.cnblogs.com/my-love-is-python/p/11018466.html