Получение AttributeError: невозможно получить атрибут «MulticlassRegrade» в <модуле «модели» при выполнении инструкций р ⇐ Python

[Anonymous]

Я пытаюсь запустить этот репозиторий GitHub локально. Однако, когда я пытаюсь запустить их тестовую команду (с их тестовыми изображениями), я сталкиваюсь с этой ошибкой:

AttributeError: Невозможно получить атрибут «MulticlassRegrade» ' на '. Фраза MulticlassReгрессия существует только в файлах ./micronuclAI_model/micronuclai.pt и ./src/model/__pycache__/models.cpython-311.pyc.

обратная трассировка

Код: Выделить всё

Traceback (most recent call last):
File ".../micronuclAI/./src/model/prediction2.py", line 152, in
main(args)
File ".../micronuclAI/./src/model/prediction2.py", line 100, in main
model = torch.load(args.model, map_location=args.device)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "user/.local/lib/python3.11/site-packages/torch/serialization.py", line 1025, in load
return _load(opened_zipfile,
^^^^^^^^^^^^^^^^^^^^^
File "user/.local/lib/python3.11/site-packages/torch/serialization.py", line 1446, in _load
result = unpickler.load()
^^^^^^^^^^^^^^^^
File "user/.local/lib/python3.11/site-packages/torch/serialization.py", line 1439, in find_class
return super().find_class(mod_name, name)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: Can't get attribute 'MulticlassRegression' on 

ОБНОВЛЕНИЕ
Я обнаружил, что переименовал класс micronuclAI в MulticlassReгрессия. Авторы, вероятно, переименовали класс в конце, после создания файла .pt.
Авторы исправили эту ошибку отдельно, но я столкнулся с другой проблемой в их новом коде, о которой я уже упоминал. в выпуске о репозитории.

Подробнее здесь: https://stackoverflow.com/questions/793 ... n-module-m