Fix default configurations (rollback)

1pha · Mar 18, 2024 · 227a5a1 · 227a5a1
1 parent f9fe3f8
commit 227a5a1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/config/dataloader/default.yaml b/config/dataloader/default.yaml
@@ -1,5 +1,5 @@
 _target_: torch.utils.data.DataLoader
-batch_size: 4
-num_workers: 2
+batch_size: 32
+num_workers: 4
 pin_memory: True
-dataset: ${dataset}
+dataset: ${dataset}
diff --git a/config/trainer/default.yaml b/config/trainer/default.yaml
@@ -5,4 +5,4 @@ devices: 1
 accelerator: gpu
 gradient_clip_val: 1
 log_every_n_steps: 50
-accumulate_grad_batches: 4
+accumulate_grad_batches: 1