color/pl_sd_train.py