training: fix type mismatch when training (#6) 94cf30b verified luofuli Jack477 commited on May 27, 2024