paddlepaddle吧 关注:213贴子:312
  • 2回复贴,共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回复
    定位为训练代码可测试代码均使用了optimizer.minimize方法,实际上都跑的是训练


    IP属地:北京2楼2019-06-09 16:08
    收起回复