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

Change batch size for tamil.

parent ce036252
Branches
No related tags found
No related merge requests found
......@@ -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