i trained my model on cpu without a problem, but when i tried to do it on gpu i got abnormal logs and a