code/original_training.py