update debug info

app.py

@@ -82,6 +82,8 @@ if opt.resume is not None:
     print(f'[INFO] load resume success!')
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+print('first device')
+print(device)
 model = model.half().to(device)
 model.eval()
 

core/models.py

@@ -52,6 +52,9 @@ class LTRFM_NeRF(nn.Module):
         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(device)
+        print('aabb')
         # 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)