Ошибка быстрого AI с примером простого распознавания изображений ⇐ Python

[Anonymous]

Я прохожу первый урок быстрого курса по искусственному интеллекту и пробую пример «Это птица». С ноутбуком все работает нормально, но я начал рефакторинг компонентов в виде конвейера, и одним из шагов конвейера является обучение модели следующим образом:

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

def train_model_from_data():
learn = vision_learner(dls, resnet34, metrics=[partial(accuracy, axis=1)])
learn.fine_tune(6)
return learn

Когда я запустил файл Python, содержащий этот метод, я столкнулся с ошибкой, как показано ниже:

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

  File "/home/myuser/Projects/Private/ml-projects/nn-playground/src/models/is_it_a_bird/train_model.py", line 74, in download_train_dump_model
learner = train_model_from_data()
^^^^^^^^^^^^^^^^^^^^^^^
File "/home/myuser/Projects/Private/ml-projects/nn-playground/src/models/is_it_a_bird/train_model.py", line 62, in train_model_from_data
learn.fine_tune(6)
^^^^^^^^^^^^^^^
File "/home/myuser/Projects/Private/ml-projects/nn-playground/myenv/lib/python3.11/site-packages/fastcore/basics.py", line 503, in __getattr__
if attr is not None: return getattr(attr,k)

Есть идеи, в чем может быть ошибка?

Подробнее здесь: https://stackoverflow.com/questions/784 ... on-example