产生这个的主要原因是DataLoader
无法处理GPU内的数据,只能使用CPU处理,故在Dataset中的__getitem__
函数的返回值只能是cpu中的数据
解决DataLoader产生的数据全是零的问题
最新推荐文章于 2024-04-17 14:43:51 发布
产生这个的主要原因是DataLoader
无法处理GPU内的数据,只能使用CPU处理,故在Dataset中的__getitem__
函数的返回值只能是cpu中的数据