reduce batch sizes (OOM)

2026-05-21 10:56:23 +02:00
parent 0fcab1d3c7
commit 4c13856af2
6 changed files with 12 additions and 12 deletions
@@ -8,7 +8,7 @@ meta:
  num_classes: 182

 data:
-  batch_size: 16384
+  batch_size: 256
  root_path: ./wilds_data
  num_workers: 10
  pin_mem: true
@@ -52,5 +52,5 @@ validation:
  eval_every: 1

 logging:
-  folder: experiment_logs/supervised-vith14.224-bs.16384-ep.50-lp-lars-lr0.01-wd0.0005/
+  folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.01-wd0.0005/
  write_tag: linear_probe-lars-lr0.01-wd0.0005