1、workers_per_gpu设置为0,使得数据读取过程为单进程模式,这样才能调试,默认处于多进程模式,无法进入__getitem__函数。参考链接
2、如果直接将workers_per_gpu=0会报错:
'persistent_workers option needs num_workers > 0'
解决方法:需要将:persistent_workers=False。参考链接
for i, data_batch in enumerate(self.data_loader)