模型的保存:
model_dict = model.state_dict()
fluid.save_dygraph(model_dict, model_path)
#model_path即为存储位置,文件名的部分不用加后缀
模型的读取
model_state_dict = fluid.load_dygraph(model_path)
model.load_dict(model_state_dict[0])
#这里注意model_path文件名部分不要加后缀,否则会报错
模型的保存:
model_dict = model.state_dict()
fluid.save_dygraph(model_dict, model_path)
#model_path即为存储位置,文件名的部分不用加后缀
模型的读取
model_state_dict = fluid.load_dygraph(model_path)
model.load_dict(model_state_dict[0])
#这里注意model_path文件名部分不要加后缀,否则会报错