Init random seed for each worker.
(worker_id: int, num_workers: int, rank: int, seed: int)
| 122 | |
| 123 | |
| 124 | def worker_init_fn(worker_id: int, num_workers: int, rank: int, seed: int): |
| 125 | """Init random seed for each worker.""" |
| 126 | # The seed of each worker equals to |
| 127 | # num_worker * rank + worker_id + user_seed |
| 128 | worker_seed = num_workers * rank + worker_id + seed |
| 129 | np.random.seed(worker_seed) |
| 130 | random.seed(worker_seed) |
nothing calls this directly
no outgoing calls
no test coverage detected