当前位置:网站首页>深度学习随机数设置,保证实验的可重复性
深度学习随机数设置,保证实验的可重复性
2022-07-31 07:13:00 【Mick..】
在深度学习实验的时候,由于代码里面存在大量的随机操作,因此导致实验的可重复性比较差。为了保证实验的可重复性,我们需要对实验中一些随机操作进行设置。
1 python 和 numpy
如果读取数据的过程采用了随机预处理(如RandomCrop、RandomHorizontalFlip等),那么对python、numpy的随机数生成器也需要设置种子。
import random
import numpy as np
random.seed(seed)
np.random.seed(seed)
2 pytorch
torch.manual_seed(seed) # 为CPU设置随机种子
torch.cuda.manual_seed(seed) # 为当前GPU设置随机种子
torch.cuda.manual_seed_all(seed) # 为所有GPU设置随机种子
参考文献:
边栏推荐
猜你喜欢
随机推荐
2022.07.24_每日一题
个人报错问题 持续总结
【面试题】从输入URL到游览器渲染完成,经历了什么
《opencv学习笔记》-- 仿射变换
XSS靶场prompt.ml过关详解
2022.07.20_Daily Question
MySql数据库优化查询工具
2022.07.12 _ a day
初识NK-RTU980开发板
2022.07.18_每日一题
'vite' is not an internal or external command, nor is it a runnable program or batch file.
navicat 新建数据库
@Transactional注解的失效场景
进程调度的基本过程
2022.07.13 _ a day
手把手教你开发微信小程序自定义底部导航栏
循环结构--for循环
一文读懂Elephant Swap,为何为ePLATO带来如此高的溢价?
正则表达式绕过
Vscode: Project-tree plugin