Fastai 1.0.x использует модель torchvision

Question

Может кто-нибудь показать мне пример использования модели torchvision в fastai без изменений (т.е. без изменений сети, выполненных create_cnn)? Я пробовал это, но получаю RuntimeError: size mismatch, Я думаю, что входной размер alexnet должен быть 224, поэтому не уверен, как отлаживать.

data = ImageDataBunch.from_name_re(path_img, 
                                   get_image_files(path/'images'), 
                                   r'/([^/]+)_\d+.jpg$', 
                                   ds_tfms=None, size=224, bs=16)

learn = create_cnn(data, models.alexnet, metrics=error_rate, 
                   custom_head=models.alexnet(num_classes=data.c).classifier)

print(learn.summary())