Training tutorial quantitative eval

[priyasundaresan]

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:

Thank you very much in advance for the help!!

[manuelli]

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).