xrg commited on
Commit
98fc8ee
·
1 Parent(s): cbf7cc7

add dbug info

Browse files
Files changed (1) hide show
  1. core/tensorBase.py +6 -0
core/tensorBase.py CHANGED
@@ -262,6 +262,12 @@ class TensorBase(torch.nn.Module):
262
  pass
263
 
264
  def normalize_coord(self, xyz_sampled):
 
 
 
 
 
 
265
  return (xyz_sampled-self.aabb[0]) * self.invaabbSize - 1
266
 
267
  def get_optparam_groups(self, lr_init_spatial = 0.02, lr_init_network = 0.001):
 
262
  pass
263
 
264
  def normalize_coord(self, xyz_sampled):
265
+ print("debug")
266
+
267
+ print(xyz_samples.device)
268
+ print(self.aabb[0].device)
269
+ print(self.invaabbSize.device)
270
+
271
  return (xyz_sampled-self.aabb[0]) * self.invaabbSize - 1
272
 
273
  def get_optparam_groups(self, lr_init_spatial = 0.02, lr_init_network = 0.001):