Skip to content

只能跑resnet,训练其他模型时,一直报错。 #5

@loganzhang4

Description

@loganzhang4

Traceback (most recent call last):
File "E:\Git\download\pytorch-classifier-master\main.py", line 151, in
opt, model, ema, train_dataset, test_dataset, optimizer, scaler, lr_scheduler, loss, DEVICE, CLASS_NUM, begin_epoch, best_metrice = parse_opt()
File "E:\Git\download\pytorch-classifier-master\main.py", line 113, in parse_opt
model = select_model(opt.model_name, CLASS_NUM, (opt.image_size, opt.image_size), opt.image_channel,
File "E:\Git\download\pytorch-classifier-master\utils\utils_model.py", line 59, in select_model
model = eval('models.{}(pretrained={})'.format(name, pretrained))
File "", line 1, in
TypeError: 'module' object is not callable
![Uploading P_0LYPL2Y825NUXPY_3{_8.png…

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions