Set random seed for reproducability.
(seed)
| 454 | |
| 455 | |
| 456 | def set_random_seed(seed): |
| 457 | """Set random seed for reproducability.""" |
| 458 | |
| 459 | if seed is not None and seed > 0: |
| 460 | random.seed(seed) |
| 461 | np.random.seed(seed) |
| 462 | torch.manual_seed(seed) |
| 463 | mpu.model_parallel_cuda_manual_seed(seed) |
| 464 | |
| 465 | |
| 466 | def get_train_val_test_data(args): |