Joblib で tqdm を使って進行状況を表示

実は簡単

From https://github.com/joblib/joblib/issues/972#issuecomment-1623366702

import joblib
from tqdm import tqdm
import time


def workload(i):
    time.sleep(5)
    return i

n_iter = 4

result = [r for r in tqdm(joblib.Parallel(return_as="generator", n_jobs=2)(joblib.delayed(workload)(i) for i in range(n_iter)), total=n_iter)]
print(result)

コメント

コメントを残す

メールアドレスが公開されることはありません。 が付いている欄は必須項目です