Can't load the pretrained model

clcarwin / sphereface_pytorch

A PyTorch Implementation of SphereFace.

MIT License

714 stars 172 forks source link

Can't load the pretrained model #60

Open lychenyoko opened 3 years ago

lychenyoko commented 3 years ago

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!

AlonZolfi commented 3 years ago

You can unzip the file using 7z software. the zip file contains .pth file inside, which can be loaded regularly by pytorch.