Changed label for R6 and R9

martius-lab · Mar 28, 2024 · 64e80ac · 64e80ac
1 parent 09a30a1
commit 64e80ac
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/hitchhiking_rotations/cfgs/cfg_pose_to_fourier.py b/hitchhiking_rotations/cfgs/cfg_pose_to_fourier.py
@@ -21,7 +21,7 @@ def get_cfg_pose_to_fourier(device, nb, nf):
     return {
         "verbose": False,
         "batch_size": 64,
-        "epochs": 300,
+        "epochs": 400,
         "training_data": {
             "_target_": "hitchhiking_rotations.datasets.PoseToFourierDataset",
             "mode": "train",
@@ -41,7 +41,7 @@ def get_cfg_pose_to_fourier(device, nb, nf):
         "val_data": {
             "_target_": "hitchhiking_rotations.datasets.PoseToFourierDataset",
             "mode": "val",
-            "dataset_size": 400,
+            "dataset_size": 200,
             "device": device,
             "nb": nb,
             "nf": nf,

diff --git a/visu/figure_14.py b/visu/figure_14.py
@@ -62,8 +62,8 @@
         "quat_rf": RotRep.QUAT_RF,
         "quat_c": RotRep.QUAT_C,
         "quat_aug": RotRep.QUAT_AUG,
-        "r6": RotRep.GSO,
-        "r9": RotRep.SVD,
+        "r6": RotRep.RSIX,
+        "r9": RotRep.ROTMAT,
     }
 
     training_metric = "l2"