Closed JMBokhorst closed 1 year ago
It seems to be solved by saving the entire model and not just the state dict.
I am still curious about the issue, why this happened all of a sudden, and if there is a better way to solve it.
This issue is stale because it has been open 60 days with no activity. Remove stale label or comment or this will be closed in 7 days.
This issue was closed because it has been stalled for 7 days with no activity.
Hi,
I have a similar problem to this issue and hope you can help.
I'm using python3.8, PyTorch 1.9 and segmentation models 0.3.3 (installed via pip3 install git+https://github.com/qubvel/segmentation_models.pytorch).
I trained an Unet-plusplus model with a mobilenetv2 backbone, directly after I trained the model I wanted to apply it however I got this error:
I see that names are slightly different (x_0_0 vs 0 for example) and would expect that to happen when I would use different versions between the runs. However, everything is identical in terms of versions between training and inference.
Could you help?
Thanks in advance