small fix in training scripts