ram overflow

2026-07-17 10:04:19 +02:00
parent 08ab1b15b7
commit 35d92f1219
1 changed files with 2 additions and 2 deletions
@@ -1,5 +1,5 @@
 data:
-  batch_size: 256
+  batch_size: 128
  color_jitter_strength: 0.0
  crop_scale:
    - 1.0
@@ -40,7 +40,7 @@ meta:
  use_bfloat16: true
  use_gradient_checkpointing: true
 optimization:
-  gradient_accumulation_steps: 1
+  gradient_accumulation_steps: 2
  ema:
    - 0.996
    - 1.0