workers_per_gpu=0

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)

posted @ 2023-03-06 18:20  kuaqi  阅读(446)  评论(0编辑  收藏  举报