运行resnet18的classify训练

[MiniBullLab]

2021-10-12 13:36:30,113 ERROR   [backbone_factory.py, 47] Traceback (most recent call last):
  File "/easy_data/easy_ai/easyai/model_block/utility/backbone_factory.py", line 34, in get_backbone_model
    result = self.get_backbone_from_name(model_args)
  File "/easy_data/easy_ai/easyai/model_block/utility/backbone_factory.py", line 66, in get_backbone_from_name
    result = build_from_cfg(model_args, REGISTERED_CLS_BACKBONE)
  File "/easy_data/easy_ai/easyai/utility/registry.py", line 109, in build_from_cfg
    return obj_cls(**args)
TypeError: __init__() got an unexpected keyword argument 'class_number'

[MiniBullLab]

2021-10-12 13:36:30,113 ERROR   [train_task.py, 42] Traceback (most recent call last):
  File "easyai/train_task.py", line 37, in train
    task = build_from_cfg(task_args, REGISTERED_TRAIN_TASK)
  File "/easy_data/easy_ai/easyai/utility/registry.py", line 109, in build_from_cfg
    return obj_cls(**args)
  File "/easy_data/easy_ai/easyai/tasks/cls/classify_train.py", line 20, in __init__
    self.set_model(gpu_id=gpu_id)
  File "/easy_data/easy_ai/easyai/tasks/utility/base_train.py", line 82, in set_model
    self.model = self.torchModelProcess.create_model(self.model_args, gpu_id)
  File "/easy_data/easy_ai/easyai/torch_utility/torch_model_process.py", line 29, in create_model
    model = self.modelFactory.get_model(model_config)
  File "/easy_data/easy_ai/easyai/model/utility/model_factory.py", line 37, in get_model
    result = self.get_model_from_cfg(input_name, model_args)
  File "/easy_data/easy_ai/easyai/model/utility/model_factory.py", line 52, in get_model_from_cfg
    model = MyModel(model_define, path, default_args)
  File "/easy_data/easy_ai/easyai/model/common/my_model.py", line 25, in __init__
    self.create_block_list()
  File "/easy_data/easy_ai/easyai/model/common/my_model.py", line 31, in create_block_list
    base_out_channels = backbone_block.get_outchannel_list()
AttributeError: 'NoneType' object has no attribute 'get_outchannel_list'

[lpj0822]

已经修改