Hello I’ve trained ResNet18 model yesterday but can’t load it for inference today. What am I missing?
Ekran Alıntısı1452×806 179 KB
And when I change the last cell to: model.load_state_dict(checkpoint['state_dict']) it gives me this error: Ekran Alıntısı2857×264 26 KB