解决DataLoader产生的数据全是零的问题

产生这个的主要原因是DataLoader无法处理GPU内的数据,只能使用CPU处理,故在Dataset中的__getitem__函数的返回值只能是cpu中的数据

你可能感兴趣的:(计算机视觉,pytorch)