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
Tags
No related merge requests found
...@@ -119,7 +119,6 @@ def run(_): ...@@ -119,7 +119,6 @@ def run(_):
--pretrained_transformer_name {utils.LANG2TRANSFORMER[lang]} --pretrained_transformer_name {utils.LANG2TRANSFORMER[lang]}
--serialization_dir {serialization_dir} --serialization_dir {serialization_dir}
--cuda_device {FLAGS.cuda_device} --cuda_device {FLAGS.cuda_device}
--word_batch_size 2500
--config_path {pathlib.Path.cwd() / 'combo' / 'config.graph.template.jsonnet'} --config_path {pathlib.Path.cwd() / 'combo' / 'config.graph.template.jsonnet'}
--notensorboard --notensorboard
""" """
...@@ -128,6 +127,11 @@ def run(_): ...@@ -128,6 +127,11 @@ def run(_):
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"
if lang in {"ta"}:
command = command + " --word_batch_size 500"
else:
command = command + " --word_batch_size 2500"
utils.execute_command("".join(command.splitlines())) utils.execute_command("".join(command.splitlines()))
......
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