Upload model
Browse files- modeling.py +0 -1
modeling.py
CHANGED
|
@@ -106,7 +106,6 @@ class CayleySTRING(nn.Module):
|
|
| 106 |
positions ([b, n, pos_dim]): Positions tensor.
|
| 107 |
"""
|
| 108 |
# Compute (I + S)^-1 @ x
|
| 109 |
-
print(self.training)
|
| 110 |
if self.training:
|
| 111 |
# Use linalg.solve during training for numerical stability.
|
| 112 |
y = torch.linalg.solve(
|
|
|
|
| 106 |
positions ([b, n, pos_dim]): Positions tensor.
|
| 107 |
"""
|
| 108 |
# Compute (I + S)^-1 @ x
|
|
|
|
| 109 |
if self.training:
|
| 110 |
# Use linalg.solve during training for numerical stability.
|
| 111 |
y = torch.linalg.solve(
|