40692791b6d7be242aa96a375daf468f54250771,DataLoader.py,DataLoader,next,#DataLoader#,95

Before Change

                return (src_data, src_pos), (tgt_data, tgt_pos)

        else:
            self.shuffle()
            self._iter_count = 0
            raise StopIteration()

After Change

        else:

            if self._need_shuffle:
                self.shuffle()

            self._iter_count = 0
            raise StopIteration()

Instances