Update 'pages/students/2016/patrik_pavlisin/dp2022/train1.yaml'
This commit is contained in:
parent
a71d71e63f
commit
e95865c48e
@ -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
|
Loading…
Reference in New Issue
Block a user