-
Notifications
You must be signed in to change notification settings - Fork 4
Description
Hi Raphael I am getting following results on WMT En-De Translation for
- 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