change grid

This commit is contained in:
YannAhlgrim
2026-06-02 12:01:59 +02:00
parent 150c2a7253
commit 628daab3b7
+2 -2
View File
@@ -7,9 +7,9 @@ grid:
optimization.optimizer: [adamw] optimization.optimizer: [adamw]
optimization.lr: [0.01, 0.05, 0.001] optimization.lr: [0.01, 0.05, 0.001]
optimization.weight_decay: [5.0e-4, 0] optimization.weight_decay: [5.0e-4, 0]
optimization.momentum: [0.9, 0.99, 0.5] optimization.momentum: [0.9, 0.5]
optimization.lr_schedule: [cosine, step] optimization.lr_schedule: [cosine, step]
data.batch_size: [32, 64, 128, 256] data.batch_size: [64, 128, 256]
meta.representation_type: [last_avgpool, last4_avgpool_concat] meta.representation_type: [last_avgpool, last4_avgpool_concat]
meta.head_type: [linear, bn_linear] meta.head_type: [linear, bn_linear]