-
Notifications
You must be signed in to change notification settings - Fork 2.6k
Description
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