Skip to content

Commit

Permalink
fix(snakefile): correctly locate files to translate (#402)
Browse files Browse the repository at this point in the history
  • Loading branch information
AmitMY authored Jan 29, 2024
1 parent 4db5fd7 commit 225b30d
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions Snakefile
Original file line number Diff line number Diff line change
Expand Up @@ -452,7 +452,7 @@ rule translate_mono_trg:
threads: gpus_num * 2
resources: gpu=gpus_num
input:
bin=ancient(decoder), file=f'{translated}/mono_trg/file.{{part}}',
bin=ancient(decoder), file=f'{translated}/mono_trg/file.{{part}}.gz',
vocab=vocab_path, model=f'{backward_dir}/{best_model}'
output: f'{translated}/mono_trg/file.{{part}}.out'
params: args = get_args("decoding-backward")
Expand Down Expand Up @@ -546,7 +546,7 @@ rule translate_corpus:
resources: gpu=gpus_num
input:
ancient(decoder),
file=f'{translated}/corpus/file.{{part}}',
file=f'{translated}/corpus/file.{{part}}.gz',
vocab=vocab_path,
teacher_models=expand(f"{teacher_base_dir}{{ens}}/{best_model}",ens=ensemble)
output: f'{translated}/corpus/file.{{part}}.nbest'
Expand Down Expand Up @@ -599,7 +599,7 @@ rule translate_mono_src:
threads: gpus_num*2
resources: gpu=gpus_num
input:
file=f'{translated}/mono_src/file.{{part}}',vocab=vocab_path,
file=f'{translated}/mono_src/file.{{part}}.gz',vocab=vocab_path,
teacher_models=expand(f"{teacher_base_dir}{{ens}}/{best_model}",ens=ensemble),
bin=ancient(decoder)
output: f'{translated}/mono_src/file.{{part}}.out'
Expand Down

0 comments on commit 225b30d

Please sign in to comment.