【colab pytorch】设置随机种子
【colab pytorch】设置随机种子
西西嘛呦
发布于 2020-08-26 10:15:22
发布于 2020-08-26 10:15:22
代码可运行
运行总次数：0
代码可运行
为了保证在本机上的可复现性，需要设置随机种子：
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本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
原始发表：2020-03-06 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
腾讯云开发者
