From e95865c48ebcf41cbd082357d7c3cd6a1d14d5f2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Patrik=20Pavli=C5=A1in?= <patrik.pavlisin@student.tuke.sk>
Date: Tue, 15 Feb 2022 11:39:19 +0000
Subject: [PATCH] Update
 'pages/students/2016/patrik_pavlisin/dp2022/train1.yaml'

---
 pages/students/2016/patrik_pavlisin/dp2022/train1.yaml | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pages/students/2016/patrik_pavlisin/dp2022/train1.yaml b/pages/students/2016/patrik_pavlisin/dp2022/train1.yaml
index 4e5b184efb..0ff6bd986f 100644
--- a/pages/students/2016/patrik_pavlisin/dp2022/train1.yaml
+++ b/pages/students/2016/patrik_pavlisin/dp2022/train1.yaml
@@ -22,9 +22,12 @@ data:
 src_vocab: dp2022/run2/example.vocab.src
 tgt_vocab: dp2022/run2/example.vocab.tgt
 
+# Train on a single GPU
+world_size: 1
+gpu_ranks: [0]
+
 # Where to save the checkpoints
 save_model: dp2022/run2/model
-
 save_checkpoint_steps: 1000
 train_steps: 20000
 valid_steps: 10000
\ No newline at end of file