From e95865c48ebcf41cbd082357d7c3cd6a1d14d5f2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Patrik=20Pavli=C5=A1in?= 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 4e5b184e..0ff6bd98 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