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

Change batch size for tamil.

parent 6900d4f0
No related branches found
No related tags found
2 merge requests!37Release 1.0.4.,!36Release 1.0.4
......@@ -119,7 +119,6 @@ def run(_):
--pretrained_transformer_name {utils.LANG2TRANSFORMER[lang]}
--serialization_dir {serialization_dir}
--cuda_device {FLAGS.cuda_device}
--word_batch_size 2500
--config_path {pathlib.Path.cwd() / 'combo' / 'config.graph.template.jsonnet'}
--notensorboard
"""
......@@ -128,6 +127,11 @@ def run(_):
if lang in {"fr", "ru"}:
command = command + " --targets deprel,head,upostag,lemma,feats"
if lang in {"ta"}:
command = command + " --word_batch_size 500"
else:
command = command + " --word_batch_size 2500"
utils.execute_command("".join(command.splitlines()))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment