Training tutorial quantitative eval

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

559 stars 133 forks source link

Training tutorial quantitative eval #195

Closed priyasundaresan closed 5 years ago

priyasundaresan commented 5 years ago

Hi there, any idea what would be causing this error in the last cell of training_tutorial.ipynb?

Can't seem to track down where the indexing issue is and didn't see any similar issues posted:

Screenshot from 2019-05-27 13-05-46

Thank you very much in advance for the help!!

manuelli commented 5 years ago

I actually just fixed this issue in a private branch that updates the entire repo to pytorch 1.1 (PR coming soon). In the mean time this is just like a 2 line fix. See the screenshot below (sorry to do a screenshot but it's in another repo and this is the easiest way to link it).