np.random.seed(1) torch.manual_seed(1) torch.cuda.manual_seed_all(1) torch.backends.cudnn.deterministic = True # 保证每次结果一样 start_time = time.time()
时间: 2024-04-28 16:21:04 浏览: 141
np.random.seed() 的使用详解
这段代码的作用是设置随机数种子以保证实验的可重复性,具体来说:
- `np.random.seed(1)` 设置 NumPy 的随机数种子,确保每次生成的随机数相同。
- `torch.manual_seed(1)` 设置 PyTorch 的随机数种子,确保每次生成的随机数相同。
- `torch.cuda.manual_seed_all(1)` 设置所有可用的 CUDA 设备的随机数种子,确保每次生成的随机数相同。
- `torch.backends.cudnn.deterministic = True` 设置 CuDNN 的随机数种子,确保每次生成的随机数相同。
- `start_time = time.time()` 记录程序的开始时间。
阅读全文