Open
Description
Hi, I am testing the code under the environment of:
pytorch == 1.9.0
torchvision == 0.10.0
python == 3.6.5
When I try torch.load(args.model)
it returns an error:
File "/usr/local/cuda/lib/python3.6/site-packages/torch/serialization.py", line 777, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, '7'.
Could you help me with this? Thanks!
Metadata
Metadata
Assignees
Labels
No labels