From 00c388ed8c24d36c1dbba5c3d38ad03b8d53eade Mon Sep 17 00:00:00 2001
From: Mateusz Klimaszewski <mk.klimaszewski@gmail.com>
Date: Mon, 10 May 2021 14:07:09 +0200
Subject: [PATCH] Fix fr and ru models training.
---
scripts/train_iwpt21.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/scripts/train_iwpt21.py b/scripts/train_iwpt21.py
index b5838c4..39054e4 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:
--
GitLab