fix some name bugs in training