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())