(seed: int, backend: str)
| 148 | |
| 149 | |
| 150 | def seed_all(seed: int, backend: str) -> None: |
| 151 | torch.manual_seed(seed) |
| 152 | np.random.seed(seed) |
| 153 | if backend == "cuda": |
| 154 | torch.cuda.manual_seed_all(seed) |
| 155 | |
| 156 | |
| 157 | def move_batch_to_device(batch: dict[str, Any], device: str) -> dict[str, Any]: |