Skip to content
Snippets Groups Projects
Commit 00c388ed authored by Mateusz Klimaszewski's avatar Mateusz Klimaszewski
Browse files

Fix fr and ru models training.

parent 2941cf06
2 merge requests!37Release 1.0.4.,!36Release 1.0.4
This commit is part of merge request !36. Comments created here will be created in the context of that merge request.
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 1 deletion
scripts/train_iwpt21.py
+ 2
− 1
View file @ 00c388ed
...@@ -125,8 +125,9 @@ def run(_): ...@@ -125,8 +125,9 @@ def run(_):
# Datasets without XPOS # Datasets without XPOS
if lang in {"fr", "ru"}: if lang in {"fr", "ru"}:
command = command + " --targets deprel,head,upostag,lemma,feats" command = command + " --targets deprel,head,upostag,lemma,feats,deps"
# Smaller dataset
if lang in {"ta"}: if lang in {"ta"}:
command = command + " --word_batch_size 500" command = command + " --word_batch_size 500"
else: else:
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment