rasterizer bug

[123lcy123]

Hi!
Thank you for your excellent work! However, I encountered a problem. I can train the model normally using either N031 or my own dataset, but when I perform reenactment, the first time it works fine. However, during the second reenactment, at the following code segment:

Gaussian-Head-Avatar/lib/module/CameraModule.py

Line 169 in d006861

# Rasterize visible Gaussians to image, obtain their radii (on screen).

          render_images_b, radii_b = rasterizer(
               means3D = means3D,
               means2D = means2D,
               #shs = shs[b],
               colors_precomp = colors_precomp[b],
               opacities = opacity[b],
               scales = scales[b],
               rotations = rotations[b])

There is no return value, causing the rendering to fail. Have you encountered this issue before? What might be the cause?

[GasaiYU]

Hi, have you solved this question?