adapt grid

2026-05-23 13:54:54 +02:00
parent 34a15d646d
commit 882c2b53b6
4 changed files with 7 additions and 6 deletions
@@ -4,13 +4,16 @@ constants:
  logging.write_tag: linear_probe

 grid:
-  optimization.lr: [0.01, 0.05]
+  optimization.optimizer: [adamw]
+  optimization.lr: [0.01, 0.05, 0.001]
  optimization.weight_decay: [0.0, 5.0e-4]
-  data.batch_size: [128, 256]
+  optimization.momentum: [0.9, 0.5]
+  optimization.lr_schedule: [step, cosine]
+  data.batch_size: [256, 128]

 launch:
  folder: submitit_logs/
  partition: gpu1
  nodes: 1
  tasks_per_node: 1
-  time: 300
+  time: 500