search

akanazawa / cmr

Project repo for Learning Category-Specific Mesh Reconstruction from Image Collections
MIT License
476 stars 85 forks source link

triangle_loss_fn returns 'nan' #11

Closed gsygsy96 closed 5 years ago

gsygsy96 commented 5 years ago

My pytorch vision is 1.1.0. Dose it matter?

hideonInternet commented 5 years ago

I also met the same problem with using pytorch 1.1.0

gsygsy96 commented 5 years ago

I figured it out. As mentioned in this issue, the gradient of 'sqrt' is not defined at 0.