Retrain the saved model

[arpittandon]

I think I found it. I used following for reference:

Transfer learning... twice?

So, in the end, this is how my code looks like: First, transfer learning (follows notebook from lesson3), Load Resnet50 with the weights from imagenet: from fastai.vision import * from fastai import * path = Config.data_path()/‘MYPATH’ path.mkdir(parents=True, exist_ok=True) df = pd.read_csv(path/‘train_v2.csv’) tfms = get_transforms(flip_vert=True, max_lighting=0.1, max_zoom=1.05, max_warp=0.) np.random.seed(42) src = (ImageList.from_csv(path, ‘train_v2.csv’, folder=‘train-jpg’, suf…

Though, instead of
learn.model[-1][-1]=nn.Linear(in_features=512,out_features=5, bias=True)

I did learn.model[-2] = nn.Linear(in_features=512,out_features=5, bias=True), as for me the last layer was softmax