problem while running training.ipynb -help required

RobotLocomotion / pytorch-dense-correspondence

Code for "Dense Object Nets: Learning Dense Visual Object Descriptors By and For Robotic Manipulation"

https://arxiv.org/pdf/1806.08756.pdf

Other

557 stars 133 forks source link

problem while running training.ipynb -help required #197

Closed Amulya21 closed 5 years ago

Amulya21 commented 5 years ago

Hii, I am facing a problem while training, it is working fine it 371st line of training.py (i found this using print statements) later using pdb i found that it is going to the path (python2.7/dist-packages/torch/nn/functional.py )in this functional.py at line 90 return f(input,weight,bias) there is some issue i guess. Please help Thanks

manuelli commented 5 years ago

Hi,

Sorry you ran into issues. PR #199 just updates the repo to pytorch 1.1. I ran through the entire tutorial from a fresh clone and docker build and didn't encounter any issues. Note that you should run the notebook training_tutorial.ipynb, i'm not sure which file you meant by training.ipynb.

Hopefully with these updates if you follow the tutorial start again from the beginning you shouldn't encounter any issues. But please let us know if you do.

Amulya21 commented 5 years ago

Thanks for the reply. I actually did run training_tutorial.ipynb itself but i was unable to train it that time but the problem got resolved with older version of your code itself and i was able train the model.