当前位置:   article > 正文

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

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
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 问题复现:安装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)
  • 1
  • 2
  • 3
  • 4
  • 5
  • 问题解决:在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)
  • 1
  • 2
  • 3
  • 4
  • 5
  • 问题分析:在对于高层API,PaddlePaddle在版本1.0之后做了很大的改动,比如就修改了高层API所在的位置。同时也完善了高层API的很多功能。高层API虽然没有底层API灵活,但高层API使用更简单,非常适合初学者使用。
声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号