Encode best LR for current Locs&Confs configs

PTG-Kitware · Dec 2, 2024 · 0f167e2 · 0f167e2
1 parent 4c499dc
commit 0f167e2
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 0 deletions.
diff --git a/configs/experiment/m2/feat_locsconfs.yaml b/configs/experiment/m2/feat_locsconfs.yaml
@@ -36,6 +36,8 @@ trainer:
 model:
   num_classes: 9  # number of activity classification classes
   compile: false
+  optimizer:
+    lr: 0.00005
   scheduler:
     # Code change to track train/loss instead of val/loss.
     factor: 0.9

diff --git a/configs/experiment/m2/feat_locsconfs_residualLinear.yaml b/configs/experiment/m2/feat_locsconfs_residualLinear.yaml
@@ -38,6 +38,8 @@ model:
   # This should stay false, related functionality is not available in torch
   # version used.
   compile: false
+  optimizer:
+    lr: 0.00005
   scheduler:
     # Code change to track train/loss instead of val/loss.
     factor: 0.9

diff --git a/configs/experiment/m3/feat_locsconfs.yaml b/configs/experiment/m3/feat_locsconfs.yaml
@@ -36,6 +36,8 @@ trainer:
 model:
   num_classes: 6  # number of activity classification classes
   compile: false
+  optimizer:
+    lr: 0.00005
   scheduler:
     # Code change to track train/loss instead of val/loss.
     factor: 0.9

diff --git a/configs/experiment/m3/feat_locsconfs_residualLinear.yaml b/configs/experiment/m3/feat_locsconfs_residualLinear.yaml
@@ -38,6 +38,8 @@ model:
   # This should stay false, related functionality is not available in torch
   # version used.
   compile: false
+  optimizer:
+    lr: 0.00005
   scheduler:
     # Code change to track train/loss instead of val/loss.
     factor: 0.9

diff --git a/configs/experiment/r18/feat_locsconfs.yaml b/configs/experiment/r18/feat_locsconfs.yaml
@@ -36,6 +36,8 @@ trainer:
 model:
   num_classes: 6  # number of activity classification classes
   compile: false
+  optimizer:
+    lr: 0.00005
   scheduler:
     # Code change to track train/loss instead of val/loss.
     factor: 0.9