From 5a392332ba77b2b94703b1da98869cd97430ffc0 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Tue, 11 Oct 2022 21:56:35 +0530
Subject: [PATCH] ensure 2 gpus

---
 enhancer/cli/train_config/trainer/default.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml
index f50cae1..beb497c 100644
--- a/enhancer/cli/train_config/trainer/default.yaml
+++ b/enhancer/cli/train_config/trainer/default.yaml
@@ -9,7 +9,7 @@ benchmark: False
 check_val_every_n_epoch: 1
 detect_anomaly: False
 deterministic: False
-devices: -1
+devices: 2
 enable_checkpointing: True
 enable_model_summary: True
 enable_progress_bar: True