reduce grid params

2026-06-17 15:53:52 +02:00
parent c154456e00
commit a97f9a7190
1 changed files with 2 additions and 2 deletions
@@ -7,9 +7,9 @@ grid:
  optimization.optimizer: [adamw]
  optimization.lr: [0.01, 0.05, 0.001]
  optimization.weight_decay: [5.0e-4, 0]
-  optimization.momentum: [0.9, 0.5]
+  optimization.momentum: [0.9]
  optimization.lr_schedule: [cosine]
-  data.batch_size: [256, 512, 1024]
+  data.batch_size: [512, 1024]
  meta.representation_type: [last_avgpool, last4_avgpool_concat]
  meta.head_type: [linear, bn_linear]