ReactSeq / infer.sh
Oopstom's picture
Upload 313 files
c668e80 verified
raw
history blame
258 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