YannAhlgrim/wilds-ijepa
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

reduce batch sizes (OOM)

Browse Source
This commit is contained in:
YannAhlgrim
2026-05-21 10:56:23 +02:00
parent 0fcab1d3c7
commit 4c13856af2
6 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/supervised_wilds_vith14_ep300-lp.yaml
+2 -2
View File
@@ -8,7 +8,7 @@ meta:
num_classes: 182 num_classes: 182
data: data:
batch_size: 16384 batch_size: 256
root_path: ./wilds_data root_path: ./wilds_data
num_workers: 10 num_workers: 10
pin_mem: true pin_mem: true
@@ -52,5 +52,5 @@ validation:
eval_every: 1 eval_every: 1
logging: 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 write_tag: linear_probe-lars-lr0.01-wd0.0005
configs/supervised_wilds_vith14_ep50-lp-lars-lr0.001-wd0.0.yaml
+2 -2
View File
@@ -8,7 +8,7 @@ meta:
num_classes: 182 num_classes: 182
data: data:
batch_size: 16384 batch_size: 256
root_path: ./wilds_data root_path: ./wilds_data
num_workers: 10 num_workers: 10
pin_mem: true pin_mem: true
@@ -52,5 +52,5 @@ validation:
eval_every: 1 eval_every: 1
logging: logging:
folder: experiment_logs/supervised-vith14.224-bs.16384-ep.50-lp-lars-lr0.001-wd0.0/ folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.001-wd0.0/
write_tag: linear_probe-lars-lr0.001-wd0.0 write_tag: linear_probe-lars-lr0.001-wd0.0
configs/supervised_wilds_vith14_ep50-lp-lars-lr0.001-wd0.0005.yaml
+2 -2
View File
@@ -8,7 +8,7 @@ meta:
num_classes: 182 num_classes: 182
data: data:
batch_size: 16384 batch_size: 256
root_path: ./wilds_data root_path: ./wilds_data
num_workers: 10 num_workers: 10
pin_mem: true pin_mem: true
@@ -52,5 +52,5 @@ validation:
eval_every: 1 eval_every: 1
logging: logging:
folder: experiment_logs/supervised-vith14.224-bs.16384-ep.50-lp-lars-lr0.001-wd0.0005/ folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.001-wd0.0005/
write_tag: linear_probe-lars-lr0.001-wd0.0005 write_tag: linear_probe-lars-lr0.001-wd0.0005
configs/supervised_wilds_vith14_ep50-lp-lars-lr0.01-wd0.0.yaml
+2 -2
View File
@@ -8,7 +8,7 @@ meta:
num_classes: 182 num_classes: 182
data: data:
batch_size: 16384 batch_size: 256
root_path: ./wilds_data root_path: ./wilds_data
num_workers: 10 num_workers: 10
pin_mem: true pin_mem: true
@@ -52,5 +52,5 @@ validation:
eval_every: 1 eval_every: 1
logging: logging:
folder: experiment_logs/supervised-vith14.224-bs.16384-ep.50-lp-lars-lr0.01-wd0.0/ folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.01-wd0.0/
write_tag: linear_probe-lars-lr0.01-wd0.0 write_tag: linear_probe-lars-lr0.01-wd0.0
configs/supervised_wilds_vith14_ep50-lp-lars-lr0.05-wd0.0.yaml
+2 -2
View File
@@ -8,7 +8,7 @@ meta:
num_classes: 182 num_classes: 182
data: data:
batch_size: 16384 batch_size: 256
root_path: ./wilds_data root_path: ./wilds_data
num_workers: 10 num_workers: 10
pin_mem: true pin_mem: true
@@ -52,5 +52,5 @@ validation:
eval_every: 1 eval_every: 1
logging: logging:
folder: experiment_logs/supervised-vith14.224-bs.16384-ep.50-lp-lars-lr0.05-wd0.0/ folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.05-wd0.0/
write_tag: linear_probe-lars-lr0.05-wd0.0 write_tag: linear_probe-lars-lr0.05-wd0.0
configs/supervised_wilds_vith14_ep50-lp-lars-lr0.05-wd0.0005.yaml
+2 -2
View File
@@ -8,7 +8,7 @@ meta:
num_classes: 182 num_classes: 182
data: data:
batch_size: 16384 batch_size: 256
root_path: ./wilds_data root_path: ./wilds_data
num_workers: 10 num_workers: 10
pin_mem: true pin_mem: true
@@ -52,5 +52,5 @@ validation:
eval_every: 1 eval_every: 1
logging: logging:
folder: experiment_logs/supervised-vith14.224-bs.16384-ep.50-lp-lars-lr0.05-wd0.0005/ folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.05-wd0.0005/
write_tag: linear_probe-lars-lr0.05-wd0.0005 write_tag: linear_probe-lars-lr0.05-wd0.0005