'Can't load a checkpoint with pytorch, bug in my code or repository?

I downloaded a checkpoint for ResNet-50 (pre-trained with VICReg) through the provided link:

https://dl.fbaipublicfiles.com/vicreg/resnet50_fullckpt.pth

But upon loading the checkpoint, following error appears:

>>> torch.load("/home/user/path/to/resnet50_fullckpt.pth")

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/user/anaconda3/envs/pytorch_env/lib/python3.8/site-packages/torch/serialization.py", line 594, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "/home/user/anaconda3/envs/pytorch_env/lib/python3.8/site-packages/torch/serialization.py", line 853, in _load
    result = unpickler.load()
AttributeError: Can't get attribute 'exclude_bias_and_norm' on <module '__main__' (built-in)>

I am new to pytorch so I am not sure if I should be loading this checkpoint in a different way? Or is it an issue with how the model was saved in the first place?

Sources

This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.

Source: Stack Overflow

Solution	Source

'Can't load a checkpoint with pytorch, bug in my code or repository?

Sources

Related Questions