Error when loading the model

[skyimager]

When loading the model using the below code:

checkpoint_file = "BR-Net_iter_5000.pth"
model = torch.load(checkpoint_file)
model.eval()

on Pytorch == 0.4.1

I am getting error as:

key error: "track running status"

However on pytorch==0.3.1 the model loads fine. Seems that the model is not compatible with pytorch version 0.4.1

[huster-wgm]

Yes, you are right. After updating to 0.4.1, I just save the params. I will update these checkpoints soon.