question-mark
Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Error when loading the model

See original GitHub issue

When I run the testing code

mmskl configs/recognition/st_gcn_aaai18/kinetics-skeleton/test.yaml --checkpoint ./work_dir/recognition/ST_GCN_18/kinetics-skeleton/epoch_50.pth --gpus 2

I found this error

RuntimeError: don’t know how to restore data location of torch.FloatStorage (tagged with gpu)

Do you have any suggestions? The code is running on my pc with 2 GPUs (RTX2080ti)

Thankyou!

Issue Analytics

  • State:closed
  • Created 4 years ago
  • Reactions:1
  • Comments:5

github_iconTop GitHub Comments

3reactions
eng100200commented, Sep 7, 2020

@ferisetiawan what changes you made in map_location?

1reaction
ferisetiawancommented, Feb 17, 2020

Problem solved. I did some changes on the map_location parameter during the load_checkpoint.

Read more comments on GitHub >

github_iconTop Results From Across the Web

Error in loading the model - jit - PyTorch Forums
Since the error message says “error loading the model”, it is likely the program failed to load the model file in torch::jit::load(path) ....
Read more >
Error when loading trained model with tf.keras ... - GitHub
The model will be loaded without errors. ... Note that if model.fit is not called, the model will be loaded fine. Also, if...
Read more >
An error occurred while loading the model. Verify
I'm trying the new composite model feature released in the Dec 2020 update but keep getting "An error occurred while loading the model....
Read more >
Error loading the model because it is being used
My team is trying to use the collaborative feature of Bizagi. We have been able to successfully share a process diagram with two...
Read more >
How to solve error while loading model with keras
I have a flask app, and i am using tensorflow and keras to load the model that i am going to use.Below is...
Read more >

github_iconTop Related Medium Post

No results found

github_iconTop Related StackOverflow Question

No results found

github_iconTroubleshoot Live Code

Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start Free

github_iconTop Related Reddit Thread

No results found

github_iconTop Related Hackernoon Post

No results found

github_iconTop Related Tweet

No results found

github_iconTop Related Dev.to Post

No results found

github_iconTop Related Hashnode Post

No results found