fix bug traininfo