reduce batch size

2026-05-18 11:04:47 +02:00
parent bd8debf88f
commit 8e5815923a
1 changed files with 2 additions and 2 deletions
@@ -1,5 +1,5 @@
 data:
-  batch_size: 128
+  batch_size: 16
  color_jitter_strength: 0.0
  crop_scale:
    - 1.0
@@ -43,7 +43,7 @@ optimization:
  ema:
    - 0.996
    - 1.0
-  epochs: 300
+  epochs: 600
  final_lr: 1.0e-06
  final_weight_decay: 0.4
  ipe_scale: 1.0