For anyone who encounters a problem when using pytorch 1.0. saying that _worker_manager_loop is not found. There is also a same issue occuring in proSR repository. You can find the issue here: https://github.com/fperazzi/proSR/issues/31 . This issue tells us to change _worker_memory_loop to _pin_memory_loop to fix the problem.
There are some additional steps that need to be change after changing from _worker_manager_loop to _pin_memory_loop.
go to code/dataloader.py and change:
self.worker_result_queue = multiprocessing.SimpleQueue()
to
self.worker_result_queue = multiprocessing.Queue()
For anyone who encounters a problem when using pytorch 1.0. saying that _worker_manager_loop is not found. There is also a same issue occuring in proSR repository. You can find the issue here: https://github.com/fperazzi/proSR/issues/31 . This issue tells us to change _worker_memory_loop to _pin_memory_loop to fix the problem.
There are some additional steps that need to be change after changing from _worker_manager_loop to _pin_memory_loop.
go to code/dataloader.py and change:
self.worker_result_queue = multiprocessing.SimpleQueue()
toself.worker_result_queue = multiprocessing.Queue()
then change:
self.worker_manager_thread = threading.Thread( target=_worker_manager_loop, args=(self.worker_result_queue, self.data_queue, self.done_event, self.pin_memory, maybe_device_id)) self.worker_manager_thread.daemon = True self.worker_manager_thread.start()
toself.pin_memory_thread = threading.Thread( target=_pin_memory_loop, args=(self.worker_result_queue, self.data_queue, maybe_device_id, self.done_event )) self.pin_memory_thread.daemon = True self.pin_memory_thread.start()