mks0601/ExAvatar_RELEASE

No gradient in stats['mean_2d'] when densifying and pruning scene Gaussian

Opened this issue · 1 comments

Liuyujian0408 commented

Thank you for your contribution, I met the following error when I ran the code and checked that there is no gradient in stats['mean_2d']. How to solve it?

stats['mean_2d_grad'] = torch.stack([x.grad.detach() for x in stats['mean_2d']])
AttributeError: 'NoneType' object has no attribute 'detach'
mks0601 commented

Please use export CUDA_VISIBLE_DEVICES=0 before running the training.

  • 👍1