_copy_param_info_from should be invoked with two program, with represent the same topology
收藏
_copy_param_info_from should be invoked with two program, with represent the same topology
收藏
0
收藏
全部评论(4)
你可以这样写
# 定义预测的一个程序,这个是在主程序中获取的一个程序,专门用来做测试的,这个定义要放在定义方法之前,因为测试程序是训练程序的前半部分(不包括优化器和backward),所以要定义在优化方法之前。
test_program = fluid.default_main_program().clone(for_test=True) # for_test=True,表示测试过程中使用训练模型中的BN
# Initialize optimizer
optimizer = optimizer_setting(args, params)
optimizer.minimize(loss)
place = fluid.CUDAPlace(0) if args.use_gpu else fluid.CPUPlace()
exe = fluid.Executor(place)
exe.run(fluid.default_startup_program())
0
请登录后评论
============
代码如上,提示在infer_program = train_prog.clone(for_test=True)这句报错了,错误信息如标题。
我看了一下模型库的例子,好像结构是类似的,为什么会报这个错误呢?用法上有问题吗?