NameError: name 'task_cls' is not defined

[smurf-1119]

When I run the code "python tools.py --keyword update,raw --mf weights/det-resnet18/mf.txt --mt weights/det-resnet18/mt.txt --old weights/pytorch-resnet18/resnet18-5c106cde.pth --new weights", I encounter the issue as follows: How can I solve it?

[smurf-1119]

More details: (STR) root@node03:~/data/qu/model-quantization# python tools.py --keyword update,raw --mf weights/det-resnet18/mf.txt --mt weights/det-resnet18/mt.txt --old weights/pytorch-resnet18/resnet18-5c106cde.pth --new weights Failing to import plugin, ModuleNotFoundError("No module named 'plugin'",) loading third party model failed cannot import name 'model_zoo' Traceback (most recent call last): File "tools.py", line 11, in import main as entry File "/root/data/qu/model-quantization/main.py", line 4, in task_cls.py NameError: name 'task_cls' is not defined