如何把模型设置输出到一个文件

改一个参数跑一次模型,跑多了都记不得之前干了什么,用了哪些参数,需要每次记录导出成文件。

设置一个 Config 类,用于保存各个参数,这样写主函数的时候也不用每个超参数都再去相应的地方改,直接调用类里的值

保存文件用 json 的 dumps 方法,设置 indent 可以有缩进,否则所有东西都在一行里。文件扩展名随便设置(log, txt, json)都行

config_path = os.path.join(work_dir, 'config.log')
config_json = json.dumps(model_config.__dict__, indent=4)
with open(config_path,'w') as f:
    f.write(config_json)

不过这样输出来的都没有顺序,看着还是很乱,后面再改吧,需要再设置一层大属性,分层存储输出会好看些

---------------- 坚持每天学习一点点
原文地址:https://www.cnblogs.com/tccbj/p/10728393.html