INNER CODE UNIT · Python
worker_init_fn
piddnad/DDColor · basicsr/data/__init__.py:97
def worker_init_fn(worker_id, num_workers, rank, seed):
# Set the worker seed to num_workers * rank + worker_id + seed
worker_seed = num_workers * rank + worker_id + seed
np.random.seed(worker_seed)
random.seed(worker_seed)