Spaces:
Running
on
Zero
Running
on
Zero
Fix AngleChangeEvaluator
Browse files- estimator.py +1 -1
estimator.py
CHANGED
|
@@ -227,7 +227,7 @@ class AngleChangeEvaluator(BaseEvaluator):
|
|
| 227 |
cos_angle = (ba * bc).sum().clamp(-1.0, 1.0)
|
| 228 |
return torch.acos(cos_angle) * 180.0 / torch.pi
|
| 229 |
|
| 230 |
-
angle_diff = torch.abs(compute_angle(
|
| 231 |
return torch.tensor([angle_diff > self.threshold]) if self.greater_than else torch.tensor([angle_diff < self.threshold])
|
| 232 |
|
| 233 |
|
|
|
|
| 227 |
cos_angle = (ba * bc).sum().clamp(-1.0, 1.0)
|
| 228 |
return torch.acos(cos_angle) * 180.0 / torch.pi
|
| 229 |
|
| 230 |
+
angle_diff = torch.abs(compute_angle(-1) - compute_angle(0))
|
| 231 |
return torch.tensor([angle_diff > self.threshold]) if self.greater_than else torch.tensor([angle_diff < self.threshold])
|
| 232 |
|
| 233 |
|