theshadow29 / fai-notes Goto Github PK

Hi, I'm trying to get cadene models working with fastai 1.0, but it seems there's no longer a ConvLearner. I've got some models working with create_cnn, but others fail. For example resnet18 works, but alexnet doesn't.

data = ImageDataBunch.from_name_re(path_img, fnames, pat, ds_tfms=get_transforms(),
                                   size=224, bs=64).normalize(imagenet_stats)
model_name = 'resnet18'
model = pretrainedmodels.__dict__[model_name](pretrained=None)
learn = create_cnn(data, lambda x: model, metrics=error_rate)

My main goal is to create a fastai Learner that does not use adaptive pooling so I can do graph optimizations on it (SageMaker Neo doesn't support pytorch adaptive pooling layers). Any idea how I may be able to get this to work?

I followed the same method as you stated but i get this error. Any ideas how to deal with this?