def worker_init_fn(worker_id, num_workers, rank, seed):
    worker_seed = (((num_workers * rank) + worker_id) + seed)
    np.random.seed(worker_seed)
    random.seed(worker_seed)
def worker_init_fn(worker_id, num_workers, rank, seed):
    worker_seed = (((num_workers * rank) + worker_id) + seed)
    np.random.seed(worker_seed)
    random.seed(worker_seed)