AttributeError: '_MultiProcessingDataLoaderIter' object has no attribute 'next'

[krammnic]

PyTorch version: 1.13.0.dev20220810+cu116 Train command: python train.py --train_data data_lmdb_release/training/MJ/MJ_train --valid_data data_lmdb_release/training/MJ/MJ_valid --select_data MJ-MJ --batch_ratio 0.5-0.5 --Transformation TPS --FeatureExtraction ResNet --SequenceModeling BiLSTM --Prediction Attn Full log:

Traceback (most recent call last):
  File "/home/krammnic/PycharmProjects/yolov6/deep-text-recognition-benchmark/train.py", line 318, in <module>
    train(opt)
  File "/home/krammnic/PycharmProjects/yolov6/deep-text-recognition-benchmark/train.py", line 147, in train
    image_tensors, labels = train_dataset.get_batch()
  File "/home/krammnic/PycharmProjects/yolov6/deep-text-recognition-benchmark/dataset.py", line 88, in get_batch
    image, text = data_loader_iter.next()  # .next()
AttributeError: '_MultiProcessingDataLoaderIter' object has no attribute 'next'

[scuxxs]

I have the same issue

[scuxxs]

have you solved it？

[jkwhuer]

I would appreciate it if you could tell me how to solve this problem

[import-max]

This error is an error of next() function of Python. You should edit the details of dataset.py imported in train.py

#  line87
#                image, text = data_loader_iter.next()
                image, text = next(data_loader_iter)

#  line92
#                image, text = self.dataloader_iter_list[i].next()
                image, text = next(self.dataloader_iter_list[i])

data_loader_iter.next() ---> next(data_loader_iter)
self.dataloader_iter_list[i].next() ---> next(self.dataloader_iter_list[i])

[HkDzl]

data_loader_iter.next()--->next(data_loader_iter)