(worker_id, num_workers, rank, seed)
| 61 | |
| 62 | |
| 63 | def worker_init_fn(worker_id, num_workers, rank, seed): |
| 64 | # The seed of each worker equals to |
| 65 | # num_worker * rank + worker_id + user_seed |
| 66 | worker_seed = num_workers * rank + worker_id + seed |
| 67 | np.random.seed(worker_seed) |
| 68 | random.seed(worker_seed) |
| 69 | |
| 70 | |
| 71 | class AverageMeter(object): |
nothing calls this directly
no outgoing calls
no test coverage detected