search

chaoyuaw / pytorch-vcii

Video Compression through Image Interpolation (ECCV'18) [PyTorch]
https://chaoyuaw.github.io/vcii/
GNU Lesser General Public License v2.1
209 stars 38 forks source link

Invalid index of a 0-dim tensor #5

Closed 3bhady closed 5 years ago

3bhady commented 5 years ago

We should use loss.item() instead of loss.data[0]

chaoyuaw commented 5 years ago

Thanks!