Pytorch 框架
seed = 123
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
TensorFLow / Keras 框架
seed = 123
tf.set_random_seed(seed)
np.random.seed(seed)
seed = 123
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
seed = 123
tf.set_random_seed(seed)
np.random.seed(seed)
1274
1万+
2247

被折叠的 条评论
为什么被折叠?