(seed)
| 27 | |
| 28 | |
| 29 | def seed_all(seed): |
| 30 | random.seed(seed) |
| 31 | os.environ["PYTHONHASHSEED"] = str(seed) |
| 32 | np.random.seed(seed) |
| 33 | torch.manual_seed(seed) |
| 34 | torch_device_module.manual_seed(seed) |
| 35 | torch_device_module.manual_seed_all(seed) |
| 36 | torch.backends.cudnn.benchmark = False |
| 37 | torch.backends.cudnn.deterministic = True |
| 38 | |
| 39 | |
| 40 | def save_videos_grid(videos: torch.Tensor, path: str, rescale=False, n_rows=1, fps=24): |