xrg commited on
Commit
5357224
1 Parent(s): 29c1deb
Files changed (1) hide show
  1. core/tensorBase.py +1 -1
core/tensorBase.py CHANGED
@@ -286,7 +286,7 @@ class TensorBase(torch.nn.Module):
286
  print(xyz_sampled.device)
287
  print(self.aabb[0].device)
288
  print(self.invaabbSize.device)
289
- if xyz_sampled.device!=invaabbSize.device:
290
  self.invaabbSize=self.invaabbSize.to(xyz_sampled.device)
291
 
292
  return (xyz_sampled-self.aabb[0]) * self.invaabbSize - 1
 
286
  print(xyz_sampled.device)
287
  print(self.aabb[0].device)
288
  print(self.invaabbSize.device)
289
+ if xyz_sampled.device!=self.invaabbSize.device:
290
  self.invaabbSize=self.invaabbSize.to(xyz_sampled.device)
291
 
292
  return (xyz_sampled-self.aabb[0]) * self.invaabbSize - 1