run trainer with gpu by default

MaloOLIVIER · Dec 19, 2024 · 2ed24ab · 2ed24ab
1 parent 9570f56
commit 2ed24ab
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/configs/trainer/ddp.yaml b/configs/trainer/ddp.yaml
@@ -1,11 +1,10 @@
 _target_: lightning.Trainer
 
 # Hardware
-accelerator: cpu
-# accelerator: gpu
-# num_nodes: 1
-# devices: '0,'
-# strategy: 'ddp_find_unused_parameters_true'
+accelerator: gpu
+num_nodes: 1
+devices: '0,'
+strategy: 'ddp_find_unused_parameters_true'
 
 # Epochs and batch sizes
 max_epochs: ${nb_epochs}