【colab pytorch】设置随机种子

时间:2022-07-23
本文章向大家介绍【colab pytorch】设置随机种子，主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项，具有一定的参考价值，需要的朋友可以参考一下。

为了保证在本机上的可复现性，需要设置随机种子：

import numpy as np
import torch
np.random.seed(0)
torch.manual_seed(0)
torch.cuda.manual_seed_all(0)

torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False

随机文章
本站知识点必读
最近更新