Update clscl_train.py