AttributeError: Can't pickle local object 'setup_data_loader.<locals>.seed_worker'

Hi all,
I hope everybody reading this is having a great day.

So I have a problem when reproducing the experiments of Large Language Models are Zero-Shot Reasoners.
I have installed all required package and equiped the environment.
But when I run main.py, it outputs the following error:

Traceback (most recent call last):
File “”, line 1, in
File “C:\ProgramData\anaconda3\Lib\multiprocessing\spawn.py”, line 120, in spawn_main
Traceback (most recent call last):
File “D:\GitHub project\zero_shot_cot\main.py”, line 217, in
exitcode = _main(fd, parent_sentinel)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File “C:\ProgramData\anaconda3\Lib\multiprocessing\spawn.py”, line 130, in _main
self = reduction.pickle.load(from_parent)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
EOFError: Ran out of input
main()
File “D:\GitHub project\zero_shot_cot\main.py”, line 36, in main
for i, data in enumerate(dataloader):
^^^^^^^^^^^^^^^^^^^^^
File “C:\ProgramData\anaconda3\Lib\site-packages\torch\utils\data\dataloader.py”, line 441, in iter
return self._get_iterator()
^^^^^^^^^^^^^^^^^^^^
File “C:\ProgramData\anaconda3\Lib\site-packages\torch\utils\data\dataloader.py”, line 388, in _get_iterator
return _MultiProcessingDataLoaderIter(self)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File “C:\ProgramData\anaconda3\Lib\site-packages\torch\utils\data\dataloader.py”, line 1042, in init
w.start()
File “C:\ProgramData\anaconda3\Lib\multiprocessing\process.py”, line 121, in start
self._popen = self._Popen(self)
^^^^^^^^^^^^^^^^^
File “C:\ProgramData\anaconda3\Lib\multiprocessing\context.py”, line 224, in _Popen
return _default_context.get_context().Process._Popen(process_obj)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File “C:\ProgramData\anaconda3\Lib\multiprocessing\context.py”, line 336, in _Popen
return Popen(process_obj)
^^^^^^^^^^^^^^^^^^
File “C:\ProgramData\anaconda3\Lib\multiprocessing\popen_spawn_win32.py”, line 94, in init
reduction.dump(process_obj, to_child)
File “C:\ProgramData\anaconda3\Lib\multiprocessing\reduction.py”, line 60, in dump
ForkingPickler(file, protocol).dump(obj)
AttributeError: Can’t pickle local object ‘setup_data_loader..seed_worker’

Would anybody be kind enough to tell me how to solve this problem

Hi, have you solved the problem? I met the same issue.

Hi,
I am also having same issue. Were you able to resolve the issue?
@namespace @zzy20020802