ReactSeq / infer.sh
Oopstom's picture
123
94fd410
raw
history blame
259 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 30 \
--n_best 10 \
--batch_size 16384 \
--batch_type tokens \
--max_length 500 \
--seed 0