debug

core/models.py

@@ -48,10 +48,11 @@ class LTRFM_NeRF(nn.Module):
             app_n_comp=opt.app_n_comp,
             density_n_comp=opt.density_n_comp,
         )
-           
-        aabb = torch.tensor([[-1, -1, -1], [1, 1, 1]]).to(device)
-        grid_size = torch.tensor([opt.splat_size, opt.splat_size, opt.splat_size]).to(device)
-        near_far =torch.tensor([opt.znear, opt.zfar]).to(device)
+        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()
+        near_far =torch.tensor([opt.znear, opt.zfar]).cuda()
         
         print(device)
         print('aabb')

core/tensorBase.py

@@ -243,8 +243,8 @@ class TensorBase(torch.nn.Module):
         print(self.aabbSize.device)
         self.invaabbSize = 2.0/self.aabbSize
         print(self.invaabbSize.device)
-        self.invaabbSize = self.invaabbSize.to(self.aabb.device)
-        print(self.invaabbSize.device)
+        # self.invaabbSize = self.invaabbSize.to(self.aabb.device)
+        # print(self.invaabbSize.device)
         self.gridSize= gridSize.float()
         self.units=self.aabbSize / (self.gridSize-1)
         self.stepSize=torch.mean(self.units)*self.step_ratio  # TBD step_ratio? why so small 0.5