Я хочу продолжить обучение моей модели сохранения в fast-ai, я тренировался и моделировал с использованием (resenet34) в fast-ai, затем сохранял, используя lerner.save(“classification_fastai_model_1”). Он был успешно сохранен, и теперь я хочу больше тренироваться, используя новые изображения для такой же классификации.
from fastai import *
from fastai.vision import *
path = "PlantVillage2"
В этой папке 2 подпапки, одна для здоровой, а другая для нездоровой
data = ImageDataBunch.from_folder(path,
ds_tfms=get_transforms(do_flip=False),
bs=32,
size=224,
valid_pct=0.26)
data.show_batch(row=3, figsize=(8,9))
data.normalize(imagenet_stats)
learner = cnn_learner(data, models.resnet34, metrics=[accuracy, error_rate])
learner.load("classification_fastai_model_1")
learner.fit_one_cycle(3)
Я получаю эту ошибку сейчас; Я уже преобразовал все изображения в RGB, используя пакетную обработку Photoshop.
RuntimeError: Размер тензора a (2) должен соответствовать размеру тензора b (3) в не-одиночном измерении 0
Любая помощь от вас будет оценил, спасибо.