运行classnet训练代码有import的报错

[foww-0001]

运行classnet训练代码有import的报错：

python3 easyai/train_task.py -t classify -i /home/wfw/data/VOCdevkit/caltech100_classify/ImageSets/train.txt -v /home/wfw/data/VOCdevkit/caltech100_classify/ImageSets/val.txt -m classnet

报错如下：

Traceback (most recent call last):
  File "easyai/train_task.py", line 5, in <module>
    from easyai.tasks.utility.registry import REGISTERED_TRAIN_TASK
  File "/home/wfw/workspace/EDGE/easy_ai/easyai/tasks/__init__.py", line 1, in <module>
    from . import cls
  File "/home/wfw/workspace/EDGE/easy_ai/easyai/tasks/cls/__init__.py", line 3, in <module>
    from . import classify_train
  File "/home/wfw/workspace/EDGE/easy_ai/easyai/tasks/cls/classify_train.py", line 8, in <module>
    from easyai.tasks.utility.common_train import CommonTrain
  File "/home/wfw/workspace/EDGE/easy_ai/easyai/tasks/utility/common_train.py", line 8, in <module>
    from easyai.tasks.utility.base_train import BaseTrain
  File "/home/wfw/workspace/EDGE/easy_ai/easyai/tasks/utility/base_train.py", line 11, in <module>
    from easyai.solver.utility.freeze_process import FreezePorcess
  File "/home/wfw/workspace/EDGE/easy_ai/easyai/solver/utility/freeze_process.py", line 5, in <module>
    from easyai.torch_utility.torch_freeze_layer import TorchFreezeLayer
  File "/home/wfw/workspace/EDGE/easy_ai/easyai/torch_utility/torch_freeze_layer.py", line 6, in <module>
    from easyai.name_manager import BlockType
ImportError: cannot import name 'BlockType'

[lpj0822]

bug已经修改