改一个参数跑一次模型,跑多了都记不得之前干了什么,用了哪些参数,需要每次记录导出成文件。
设置一个 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)
不过这样输出来的都没有顺序,看着还是很乱,后面再改吧,需要再设置一层大属性,分层存储输出会好看些
----------------
坚持每天学习一点点