python main.py train --train_dataset.use_generated_data true