How to save FAST AI resnet50 model to Google Drive after training in google colab notebook

Part 1 (2019)
11

Hi!
The saving part is all nice
However the loading part by learn.save(dest/'filename') leads to an error in Google Colab (and not in Paperspace Gradient)

RuntimeError: Error(s) in loading state_dict for SequentialRNN:
	size mismatch for 0.encoder.weight: copying a param with shape torch.Size([40504, 400]) from checkpoint, the shape in current model is torch.Size([40288, 400]).
	size mismatch for 0.encoder_dp.emb.weight: copying a param with shape torch.Size([40504, 400]) from checkpoint, the shape in current model is torch.Size([40288, 400]).
	size mismatch for 1.decoder.weight: copying a param with shape torch.Size([40504, 400]) from checkpoint, the shape in current model is torch.Size([40288, 400]).
	size mismatch for 1.decoder.bias: copying a param with shape torch.Size([40504]) from checkpoint, the shape in current model is torch.Size([40288]).


What do you think is causing this problem