[Solved] Using a fastai-trained model with plain Pytorch

[balnazzar (Andrea de Luca)]

@sgugger

You ave to redefine your model in plain PyTorch to be able to load those weights.

Thanks.

Ok, tell me just if I understood it right. I got to load a pretrained model from torchvision.models and then adding a custom head identical to fastai’s by using nn.Sequential(). Am I right?