Running the training script