Added exception handling in grad_total_variation

ashawkey / nerf2mesh

[ICCV2023] Delicate Textured Mesh Recovery from NeRF via Adaptive Surface Refinement

https://me.kiui.moe/nerf2mesh/

MIT License

898 stars 88 forks source link

Closed gr3atchoi closed 1 year ago

gr3atchoi commented 1 year ago

Exception handling added when size of input tensor is (0, 3) it happened when I set --bound >= 4.

In that case, "RuntimeError: CUDA error: invalid configuration argument" is raised.

gr3atchoi commented 1 year ago

fixed.

ashawkey commented 1 year ago

Thanks for your contribution!