python - Fastai 1.0.x は torchvision モデルを使用

Question

fastai で torchvision モデルを変更せずに (つまり、ネットワークを変更せずに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())