保存模型 model_save_path="model/mnist-model/dygraph-mnist" fluid.save_dygraph(model.state_dict(),model_save_path)#保存模型参数 fluid.save_dygraph(optimaizer.state_dict(),model_save_path)#保存优化器参数 加载模型 params_dict, opt_dict = fluid.load_dygraph(model_save_path) model = MN…

2020-10-10 72点热度 0人点赞 阅读全文

paddle.fluid.dygraph.save_dygraph(state_dict, model_path) 该接口将传入的参数或优化器的 dict 保存到磁盘上。 state_dict 是通过 Layer 的 state_dict() 方法得到的。 注: model_path 不可以是一个目录。 该接口会根据 state_dict 的内容,自动给 model_path …

2020-10-10 117点热度 2人点赞 阅读全文