Error when loading the model

huster-wgm / geoseg

Geoseg - A Computer Vision Package for Automatic Building Segmentation and Outline extraction

MIT License

104 stars 30 forks source link

Open skyimager opened 5 years ago

skyimager commented 5 years ago

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 commented 5 years ago

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