Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

UnpicklingError: invalid load key, 'H' #3

Open
Mar-Gra-creator opened this issue Dec 29, 2022 · 0 comments
Open

UnpicklingError: invalid load key, 'H' #3

Mar-Gra-creator opened this issue Dec 29, 2022 · 0 comments

Comments

@Mar-Gra-creator
Copy link

hey do you know what could be wrong?
cheers

(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'.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant