pytorch---之halfTensor

zxyhhjs2017發表於2019-03-20

證明出錯在dataloader裡面

在pytorch當中,float16和half是一樣的資料結構,都是屬於half操作,

然後dataloader不能返回half值,所以在dataloader裡面,要把float16改成float32即可返回

相關文章