Set aug threshhold to 0.1

martius-lab · Mar 26, 2024 · f50c2e9 · f50c2e9
1 parent c4f7fb5
commit f50c2e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hitchhiking_rotations/utils/conversions.py b/hitchhiking_rotations/utils/conversions.py
@@ -82,7 +82,7 @@ def rotmat_to_quaternion_aug(base: torch.Tensor, mode: str) -> torch.Tensor:
     rep = rotmat_to_quaternion_canonical(base)
 
     if mode == "train":
-        rep[torch.logical_and(torch.rand(rep.size(0), device=rep.device) < 0.5, rep[:, 3] < 0.3)] *= -1
+        rep[torch.logical_and(torch.rand(rep.size(0), device=rep.device) < 0.5, rep[:, 3] < 0.1)] *= -1
 
     return rep