Update 'pages/students/2016/patrik_pavlisin/dp2022/train1.yaml'

This commit is contained in:
Patrik Pavlišin 2022-02-15 11:39:19 +00:00
parent a71d71e63f
commit e95865c48e

View File

@ -22,9 +22,12 @@ data:
src_vocab: dp2022/run2/example.vocab.src src_vocab: dp2022/run2/example.vocab.src
tgt_vocab: dp2022/run2/example.vocab.tgt tgt_vocab: dp2022/run2/example.vocab.tgt
# Train on a single GPU
world_size: 1
gpu_ranks: [0]
# Where to save the checkpoints # Where to save the checkpoints
save_model: dp2022/run2/model save_model: dp2022/run2/model
save_checkpoint_steps: 1000 save_checkpoint_steps: 1000
train_steps: 20000 train_steps: 20000
valid_steps: 10000 valid_steps: 10000