color/cond_train.py