fix

core/models.py

@@ -48,7 +48,7 @@ class LTRFM_NeRF(nn.Module):
             app_n_comp=opt.app_n_comp,
             density_n_comp=opt.density_n_comp,
         )
-        print(aabb.device)
+        
         aabb = torch.tensor([[-1, -1, -1], [1, 1, 1]]).cuda()
         print(aabb.device)
         grid_size = torch.tensor([opt.splat_size, opt.splat_size, opt.splat_size]).cuda()