File size: 258 Bytes
c668e80
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
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