| """随机种子设置""" | |
| import random | |
| import numpy as np | |
| import torch | |
| def set_seed(seed=42): | |
| """设置随机种子""" | |
| random.seed(seed) | |
| np.random.seed(seed) | |
| torch.manual_seed(seed) | |
| if torch.cuda.is_available(): | |
| torch.cuda.manual_seed_all(seed) | |
| """随机种子设置""" | |
| import random | |
| import numpy as np | |
| import torch | |
| def set_seed(seed=42): | |
| """设置随机种子""" | |
| random.seed(seed) | |
| np.random.seed(seed) | |
| torch.manual_seed(seed) | |
| if torch.cuda.is_available(): | |
| torch.cuda.manual_seed_all(seed) | |