Hi,
I was just wondering, there is a small difference whenever we save state dictionary of a model and reload it back. Can you please point if I am doing some mistake or if this is an insignificant issue?
Regards,
P.S: I have pasted sample code below
from torchvision import models
import torch
model = models.resnet50(pretrained=True)
model.eval()
stateDict = model.state_dict()
torch.save(model.state_dict(), “test.pth”)
reloadDict = torch.load(“test.pth”)
diff = stateDict.items() - reloadDict.items()