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 |