wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix hanging

This commit is contained in:
Simon Alibert 2024-11-03 18:07:55 +01:00
parent e4ba084e25
commit 16103cb8b7
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/datasets/image_writer.py
View File

@ -80,6 +80,7 @@ def worker_process(queue: queue.Queue, num_threads: int):
threads = [] threads = []
for _ in range(num_threads): for _ in range(num_threads):
t = threading.Thread(target=worker_thread_process, args=(queue,)) t = threading.Thread(target=worker_thread_process, args=(queue,))
t.daemon = True
t.start() t.start()
threads.append(t) threads.append(t)
for t in threads: for t in threads:
@ -129,6 +130,7 @@ class ImageWriter:
self.queue = multiprocessing.JoinableQueue() self.queue = multiprocessing.JoinableQueue()
for _ in range(self.num_processes): for _ in range(self.num_processes):
p = multiprocessing.Process(target=worker_process, args=(self.queue, self.num_threads)) p = multiprocessing.Process(target=worker_process, args=(self.queue, self.num_threads))
p.daemon = True
p.start() p.start()
self.processes.append(p) self.processes.append(p)