I found this solution from another page of Fasi AI forum. It seems the issue can be solced adding, “remove_module=True”. It seems there is a mismatch between the Tensor size of the new model I create from resnet18 and the one that I saved. The following code might help:
learn.load('YOUR_SAVED_MODEL',strict=False,remove_module=True)