Trained models are stored in `output` directory. It's better to store them in `pre_trained_model` instead? Or in some new directory...