first commit

jkwang93 · May 20, 2021 · 9fc4597 · 9fc4597
1 parent aa0f0b0
commit 9fc4597
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -17,7 +17,10 @@ The default task of our code is the DRD2 target. Users can customize their own t
 
 ```
 python 1_train_prior_Transformer.py --train-data {your_training_data_path} --valid-data {your_valid_data_path} --save-prior-path {path_to_save_prior_model}
+
 python 2_generator_Transformer.py --prior {piror_model_path} --save_molecules_path {save_molecules_path}
+
 python 3_train_middle_model_dm.py --train-data {your_training_data_path} --save-middle-path {path_to_save_middle_model}
+
 python 4_train_agent_save_smiles.py  --num-steps 5000 --batch-size 128 --middle {path_of_middle_model} --agent {path_to_save_agent_model} ---save-file-path{save_smiles}
 ```