| 21 | from .solver import Solver |
| 22 | |
| 23 | class WorkerInit(object): |
| 24 | def __init__(self, rank, num_workers): |
| 25 | self.rank = rank |
| 26 | self.num_workers = num_workers |
| 27 | def func(self, pid): |
| 28 | print(f'[rank{self.rank}] setting worker seed {self.rank*self.num_workers+pid}', flush=True) |
| 29 | np.random.seed(self.rank*self.num_workers+pid) |
| 30 | |
| 31 | class SolverDeter(Solver): |
| 32 |