Fix: Conflict of formats between Data prep and Training, fix Tensor size 83d4d7f ra1425 commited on Nov 25