Could you please tell about using custom models from either torchvision or custom with create_cnn ? create_body by default passes pretrained bool value (resnet in fastai for instance take (pretrained,**kwargs) but other models expect (input,**kwargs) input tensor to be passed. I also discussed this earlier ,here
1 Like