Unpickling Error

[nandinibajaj]

Using Python3.8, PyTorch 1.6 and Cuda 10.2, as specified in the repository. While running the sample registration using pretrained weights provided in the link attached to the repository I encountered the following error :

`Traceback (most recent call last):
  File "inference.py", line 72, in <module>
    rma_net.load_state_dict(torch.load(args.weight),True)
  File "/home/gridraster/anaconda3/envs/rmanet/lib/python3.8/site-packages/torch/serialization.py", line 585, in load
    return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
  File "/home/gridraster/anaconda3/envs/rmanet/lib/python3.8/site-packages/torch/serialization.py", line 755, in _legacy_load
    magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: A load persistent id instruction was encountered,
but no persistent_load function was specified.

Can you help me fix this issue?
What is the pickle version used in the repository while pickling the pretrained weights?

[RainbowRui]

It looks like a version problem of PyTorch. Excuse me. Does installed PyTorch work well on some basic operations?

[nandinibajaj]

Hey, Thank you for the prompt response. I realised the issue. The weights need to be passed into the network as zipped .pt file and then it works. There is no version incompatibility issue. Thank you, Nandini Bajaj ᐧ

…

On Wed, Jul 14, 2021 at 8:20 AM Hongrui Cai ***@***.***> wrote: It looks like a version problem of PyTorch. Excuse me. Does installed PyTorch work well on some basic operations? — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub <#3 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/ALB2ERXDOLF3LVV4CGHI5FLTXT3P3ANCNFSM47TXXGWQ> .

-- *Nandini BajajThird Year UG Department of ***@***.*** ***@***.******@***.*** ***@***.***>* *LinkedIn <http://linkedin.com/in/nandini-bajaj-166327147> *

[RainbowRui]

Thanks! You are welcome!