赞
踩
关键字:训练器
,contrib
,Trainer
问题描述:在使用Trainer函数创建训练器的时候,出现错误,错误提示NameError: name ‘Trainer’ is not defined。
报错信息:
<ipython-input-7-4908863d4852> in main()
9 place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace()
10
---> 11 trainer = Trainer(
12 train_func=train_program, place=place, optimizer_func=optimizer_program)
13
NameError: name 'Trainer' is not defined
from paddle.fluid.trainer import *
导入PaddlePaddle的高级API,之后使用Trainer
创建一个训练器,就会报错。错误代码如下:from paddle.fluid.trainer import *
from paddle.fluid.inferencer import *
······
trainer = Trainer(
train_func=train_program, place=place, optimizer_func=optimizer_program)
paddle.fluid.contrib
目录下,所以在导包的是应该要使用from paddle.fluid.contrib.trainer import *
的导包方式。正确代码如下:from paddle.fluid.contrib.trainer import *
from paddle.fluid.contrib.inferencer import *
······
trainer = Trainer(
train_func=train_program, place=place, optimizer_func=optimizer_program)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。