Fix for incorrect val IoU during training due to multi-GPU set-up

[seamie6]

Hello, please see my issue posted in the GaussianOcc repo. As both use the same codebase, they suffer from the same issue. Thank you.

GANWANSHUI/GaussianOcc#37