train_CNN_stress.py