color/pl_train.py