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

reduce batch size

Browse Source
This commit is contained in:
Yann Ahlgrim
2026-06-22 13:38:06 +02:00
parent 164e25fe55
commit b2d553c20d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/grids/lp_grid.yaml
+2 -2
View File
@@ -10,13 +10,13 @@ grid:
optimization.momentum: [0.9] optimization.momentum: [0.9]
optimization.lr_schedule: [cosine] optimization.lr_schedule: [cosine]
optimization.gradient_accumulation_steps: [8, 16] optimization.gradient_accumulation_steps: [8, 16]
data.batch_size: [64] data.batch_size: [16]
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]
launch: launch:
folder: submitit_logs/ folder: submitit_logs/
partition: gpu8 partition: gpu1
nodes: 1 nodes: 1
tasks_per_node: 1 tasks_per_node: 1
time: 180 time: 180