paddlepaddle吧 关注:215贴子:314
  • 0回复贴,共1

在对一个简单的3分类模型进行预测时,使用load_params和使用load

取消只看楼主收藏回复

训练的时候保存模型是这样的model_param_path="./model_parameter/params"+str(epoch_id).zfill(3)train_param_path="./train_parameter/persistables"+str(epoch_id).zfill(3)fluid.io.save_params(executor=exe,dirname=model_param_path,main_program=train_program)fluid.io.save_persistables(executor=exe,dirname=train_param_path,main_program=train_program)测试的时候加载模型时是这样的#fluid.io.load_persistables(executor=exe,dirname='../train_parameter/persistables873',main_program=startup_program)fluid.io.load_params(executor=exe,dirname='../model_parameter/params873',main_program=startup_program)


1楼2019-06-09 16:08回复