train.py error ModelDescBase？

[xianzhe-741]

Traceback (most recent call last):
File "train.py", line 81, in
train()
File "train.py", line 64, in train
starting_epoch=cfg.starting_epoch
File "/usr/local/lib/python3.6/dist-packages/tensorpack/train/config.py", line 119, in init
assert_type(model, ModelDescBase, 'model')
File "/usr/local/lib/python3.6/dist-packages/tensorpack/train/config.py", line 107, in assert_type
name, tp.name, v.class.name)
AssertionError: model has to be type 'ModelDescBase', but an object of type 'AttentionOCR' found.