RuntimeError: numel: integer multiplication overflow

[anddyhzw]

trafficstars

Thank you for your wonderfull work！Could you please help me to solve the below problem. "RuntimeError: numel: integer multiplication overflow" problem occured when I train my data with the diff_gaussian_rasterization moudle . The detailed info is below:

An error occured in forward. Please forward snapshot_fw.dump for debugging. Traceback (most recent call last): File "main_avatar.py", line 855, in trainer.train() File "main_avatar.py", line 405, in train total_loss, batch_losses = self.forward_one_pass(items) File "main_avatar.py", line 207, in forward_one_pass render_output = self.avatar_net.render(items, self.bg_color) File "/home/hezhiwu/git/AnimatableGaussians/network/avatar.py", line 215, in render render_ret = render3( File "/home/hezhiwu/git/AnimatableGaussians/gaussians/gaussian_renderer.py", line 87, in render3 rendered_image, radii, rendered_depth, rendered_alpha = rasterizer( File "/home/hezhiwu/.conda/envs/AnimatableGaussians/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl return forward_call(*args, **kwargs) File "/home/hezhiwu/.conda/envs/AnimatableGaussians/lib/python3.8/site-packages/diff_gaussian_rasterization_depth_alpha-0.0.0-py3.8-linux-x86_64.egg/diff_gaussian_rasterization_depth_alpha/init.py", line 213, in forward (AnimatableGaussians) hezhiwu@ai-devserver-03:~/git/AnimatableGaussians$ snapshot_fw.dump File "/home/hezhiwu/.conda/envs/AnimatableGaussians/lib/python3.8/site-packages/diff_gaussian_rasterization_depth_alpha-0.0.0-py3.8-linux-x86_64.egg/diff_gaussian_rasterization_depth_alpha/init.py", line 32, in rasterize_gaussians return _RasterizeGaussians.apply( File "/home/hezhiwu/.conda/envs/AnimatableGaussians/lib/python3.8/site-packages/torch/autograd/function.py", line 506, in apply return super().apply(*args, **kwargs) # type: ignore[misc] File "/home/hezhiwu/.conda/envs/AnimatableGaussians/lib/python3.8/site-packages/diff_gaussian_rasterization_depth_alpha-0.0.0-py3.8-linux-x86_64.egg/diff_gaussian_rasterization_depth_alpha/init.py", line 90, in forward raise ex File "/home/hezhiwu/.conda/envs/AnimatableGaussians/lib/python3.8/site-packages/diff_gaussian_rasterization_depth_alpha-0.0.0-py3.8-linux-x86_64.egg/diff_gaussian_rasterization_depth_alpha/init.py", line 86, in forward num_rendered, color, depth, alpha, radii, geomBuffer, binningBuffer, imgBuffer = _C.rasterize_gaussians(*args) RuntimeError: numel: integer multiplication overflow