xrg commited on
Commit ·
5357224
1
Parent(s): 29c1deb
fix buf
Browse files- 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
|