Skip to content

Assertion Error for En-De Translation using latent search and pretrained model . #5

@spprabhu

Description

@spprabhu

Hi Raphael I am getting following results on WMT En-De Translation for

  1. Fast Translation
    2.One REfinement step

python run.py --opt_dtok wmt14_ende --use_pretrain --test --evaluate
[OPTS] Model tag: dtok-wmt14_ende
Running on 1 GPUs
loading pretrained model in ./mydata/shu_trained_wmt14_ende.pt
translating: 100.0%
Average decoding time: 15ms, std: 1
BLEU = 22.304035169763978

python run.py --opt_dtok wmt14_ende --use_pretrain --opt_Trefine_steps 1 --test --evaluate
[OPTS] Model tag: dtok-wmt14_ende
Running on 1 GPUs
loading pretrained model in ./mydata/shu_trained_wmt14_ende.pt
translating: 100.0%
Average decoding time: 38ms, std: 1
BLEU = 24.147135058514433

They are inline with the specified results.However I am not able to implement the model using Latent Search as it is giving Assertion Error.

python run.py --opt_dtok wmt14_ende --use_pretrain --opt_Trefine_steps 1 --opt_Tlatent_search --opt_Tteacher_rescore --test --evaluate
[OPTS] Model tag: dtok-wmt14_ende
Running on 1 GPUs
Traceback (most recent call last):
File "run.py", line 207, in
assert os.path.exists(pretrained_autoregressive_path)
AssertionError

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions