grid for vit b ep 600

2026-06-15 13:52:07 +02:00
parent 8b714927ab
commit 1ceaf09a13
2 changed files with 4 additions and 4 deletions
@@ -8,8 +8,8 @@ grid:
  optimization.lr: [0.01, 0.05, 0.001]
  optimization.weight_decay: [5.0e-4, 0]
  optimization.momentum: [0.9, 0.5]
-  optimization.lr_schedule: [cosine, step]
-  data.batch_size: [64, 128, 256]
+  optimization.lr_schedule: [cosine]
+  data.batch_size: [256, 512, 1024]
  meta.representation_type: [last_avgpool, last4_avgpool_concat]
  meta.head_type: [linear, bn_linear]

@@ -18,4 +18,4 @@ launch:
  partition: gpu1
  nodes: 1
  tasks_per_node: 1
-  time: 180
+  time: 600