diff --git a/pages/students/2016/patrik_pavlisin/dp2022/train1.yaml b/pages/students/2016/patrik_pavlisin/dp2022/train1.yaml
index 4e5b184ef..0ff6bd986 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