fix the problem that program hang after training