debug

core/models.py

@@ -122,6 +122,8 @@ class LTRFM_NeRF(nn.Module):
         xyz_samples=xyz_samples.permute(0,2,3,4,1)
         xyz_samples=xyz_samples.view(1,-1,1,3)
         
+        print("plane"+ planes.device)
+        print("xyz_samples"+ xyz_samples.device)
 
         grid_out = self.tensorRF.predict_sdf(planes,xyz_samples)
         grid_out['sigma']=grid_out['sigma'].view(grid_size,grid_size,grid_size).float()