Description
Using the python code from the demo, I attempted to load the isnet-general-use.pth file. The link in the demo code (https://drive.google.com/uc?id=1KyMpRjewZdyYfxHPYcd-ZbanIXtin0Sn) resulted in a 404, so I looked around the repo and found the following link: https://drive.google.com/file/d/1XHIzgTzY5BQHw140EDIgwIb53K659ENH/view. I was able to successfully download this file and place it in the saved_models directory under the isnet.pth file name just like in the demo.
When I run the entire script, however, I get the following error:
/home/dis/anaconda3/envs/dis_env/lib/python3.7/site-packages/torch/nn/_reduction.py:42: UserWarning: size_average and reduce args will be deprecated, please use reduction='mean' instead.
warnings.warn(warning.format(ret))
Traceback (most recent call last):
File "impl.py", line 119, in <module>
net = build_model(hypar, device)
File "impl.py", line 66, in build_model
net.load_state_dict(torch.load(hypar["model_path"]+"/"+hypar["restore_model"],map_location=device))
File "/home/dis/anaconda3/envs/dis_env/lib/python3.7/site-packages/torch/serialization.py", line 593, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/home/dis/anaconda3/envs/dis_env/lib/python3.7/site-packages/torch/serialization.py", line 762, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, '<'.
I'm using the CPU for device, and I setup a conda environment with Python 3.7
and PyTorch 1.8
.
I was thinking maybe the PyTorch version I am using is not compatible with the pth file I am downloading, but I can't seem to get any more information about the issue.
Any help is much appreciated, and let me know if I need to provide more information!
Activity