You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
(grasr) marcin@marcin-B550-AORUS-ELITE-AX-V2:~/GraSR$ python main.py -m ./t -q ./q -o result
Traceback (most recent call last):
File "main.py", line 40, in <module>
main()
File "main.py", line 31, in main
q_descriptors = get_descriptors(args.model, args.query_protein, qd_out_path)
File "/home/marcin/anaconda3/envs/grasr/lib/python3.6/site-packages/torch/autograd/grad_mode.py", line 43, in decorate_no_grad
return func(*args, **kwargs)
File "/home/marcin/GraSR/encode.py", line 100, in get_descriptors
net_list = [load_model(mp) for mp in model_path_list]
File "/home/marcin/GraSR/encode.py", line 100, in <listcomp>
net_list = [load_model(mp) for mp in model_path_list]
File "/home/marcin/GraSR/encode.py", line 28, in load_model
state_dict = {k.replace("module.", ""): v for k, v in torch.load(model_path, map_location='cpu').items()}
File "/home/marcin/anaconda3/envs/grasr/lib/python3.6/site-packages/torch/serialization.py", line 387, in load
return _load(f, map_location, pickle_module, **pickle_load_args)
File "/home/marcin/anaconda3/envs/grasr/lib/python3.6/site-packages/torch/serialization.py", line 564, in _load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, 'H'.
The text was updated successfully, but these errors were encountered:
hey do you know what could be wrong?
cheers
The text was updated successfully, but these errors were encountered: