in floresv1/scripts/translate.py [0:0]
def translate_files_local(args, cmds):
m = mp.Manager()
gpu_queue = m.Queue()
for i in args.cuda_visible_device_ids:
gpu_queue.put(i)
with mp.Pool(processes=len(args.cuda_visible_device_ids)) as pool:
for _ in tqdm.tqdm(pool.imap_unordered(translate, [(gpu_queue, cmd) for cmd in cmds]), total=len(cmds)):
pass