Fastai 1.0.x использует модель torchvision - PullRequest
Новая
       5

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

0 голосов
/ 21 февраля 2019

Может кто-нибудь показать мне пример использования модели 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())
...