ReactSeq / infer.sh
Oopstom's picture
change beam size
e76a5fb
raw
history blame
268 Bytes
python inference.py \
--model trained_models/retrosnyhesis_ReactSeq_prompt_model_on_50k_aug100.pt \
--src ./tmp_data/src.txt \
--output ./tmp_data/tgt.txt \
--beam_size 10 \
--n_best 10 \
--batch_size 16384 \
--batch_type tokens \
--max_length 500 \
--seed 0 \
--gpu 0