Just checking

demo/gs_train.py

@@ -280,6 +280,8 @@ def train(
             view.projection_matrix = getProjectionMatrix(znear=view.znear, zfar=view.zfar, fovX=view.FoVx, fovY=view.FoVy).transpose(0,1).cuda().float()
             view.full_proj_transform = (view.world_view_transform.unsqueeze(0).bmm(view.projection_matrix.unsqueeze(0))).squeeze(0)
 
+            print("background.device: ", background.device)
+            print("view.device: ", view.original_image.device)
             render_pkg = render(view, gaussians, pipeline, background)
             rendering = render_pkg["render"]
             torchvision.utils.save_image(rendering, os.path.join(render_path, '{0:05d}'.format(idx) + ".png"))