Init random seed for each worker.
(worker_id: int, num_workers: int, rank: int, seed: int)
| 105 | |
| 106 | |
| 107 | def worker_init_fn(worker_id: int, num_workers: int, rank: int, seed: int): |
| 108 | """Init random seed for each worker.""" |
| 109 | # The seed of each worker equals to |
| 110 | # num_worker * rank + worker_id + user_seed |
| 111 | worker_seed = num_workers * rank + worker_id + seed |
| 112 | np.random.seed(worker_seed) |
| 113 | random.seed(worker_seed) |
nothing calls this directly
no outgoing calls
no test coverage detected