Fix gpu training loggins