证明出错在dataloader里面
在pytorch当中，float16和half是一样的数据结构，都是属于half操作，
然后dataloader不能返回half值，所以在dataloader里面，要把float16改成float32即可返回
pytorch---之halfTensor
猜你喜欢
转载自blog.csdn.net/zxyhhjs2017/article/details/88546423
今日推荐
周排行