More flexible transfer learning: Hacking pretrained models

[Pomo (Malcolm McLean)]

Modifying pretrained resnet - does it "just work"?

Hello PyTorch experts. I understand that PyTorch is able to track and update weights and gradients automatically. However, because I do not understand exactly how it does this magic I am also not confident I won’t break it. I would like to modify an existing Resnet model, and have tried to do so by imitating the code in fastai. Would some please check my code? The task: create a pretrained model with create_cnn the custom head created by fastai looks like this… (1): Sequential( (0): Ad…