debug

core/models.py

@@ -55,6 +55,7 @@ class LTRFM_NeRF(nn.Module):
         
         print(device)
         print('aabb')
+        print(aabb.device)
         # tensorf Renderer
         self.tensorRF = TensorVMSplit_NeRF(aabb, grid_size, density_n_comp=opt.density_n_comp,appearance_n_comp=opt.app_n_comp,app_dim=opt.app_dim,\
             density_dim=opt.density_dim,near_far=near_far, shadingMode=opt.shadingMode, pos_pe=opt.pos_pe, view_pe=opt.view_pe, fea_pe=opt.fea_pe)

core/tensorBase.py

@@ -244,6 +244,7 @@ class TensorBase(torch.nn.Module):
         self.invaabbSize = 2.0/self.aabbSize
         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