diff --git a/scripts/train_iwpt21.py b/scripts/train_iwpt21.py index b5838c491acb41390c1bd70329bced4b7f044df2..39054e4243783565086ada8b73f1ec173b2fce7d 100644 --- a/scripts/train_iwpt21.py +++ b/scripts/train_iwpt21.py @@ -125,8 +125,9 @@ def run(_): # Datasets without XPOS 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"}: command = command + " --word_batch_size 500" else: