Commit a9ad1754 authored by Mateusz Klimaszewski's avatar Mateusz Klimaszewski

Change batch size for tamil.

parent 6900d4f0
......@@ -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()))
......
Markdown is supported
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