nan output from mesh model

[chenghsiuchen]

Hello, first of all, I really appreciate your team for open sourcing your amazing work.
However, when trying to reproduce the results, I am facing this strange issue that after several iterations of training, the output of the mesh deformation model become nan.
Here is an image of the training curve.

Do you have any insights regarding this issue ?
I'm running the code in a conda environment with pytorch 1.4.0 on a single GTX 1080 with batch_size=4.

[chenghsiuchen]

I think I figured it out, seems like it's a pytorch3D bug. Check here.