PaddlePaddle在定义训练器是出现NameError: name 'Trainer' is not defined

  • 关键字:训练器contribTrainer

  • 问题描述:在使用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
  • 问题复现:安装PaddlePaddle 1.0以上的版本,然后通过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)
  • 问题解决:在PaddlePaddle 1.0以上的版本,高级API已经迁移到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)
  • 问题分析:在对于高层API,PaddlePaddle在版本1.0之后做了很大的改动,比如就修改了高层API所在的位置。同时也完善了高层API的很多功能。高层API虽然没有底层API灵活,但高层API使用更简单,非常适合初学者使用。

猜你喜欢

转载自blog.csdn.net/PaddlePaddle/article/details/86740150