训练与测试时同一组测试集结果不一致

您所在的位置:网站首页 数据集处理方式不一样跑出来的结果差别很大 训练与测试时同一组测试集结果不一致

训练与测试时同一组测试集结果不一致

2024-07-08 16:01:38| 来源: 网络整理| 查看: 265

训练与测试时同一组测试集结果不一致,相当于训练的模型不能后续使用了,这肯定不能容忍,所以耗费经历筛查:

1. 随机种子的设置问题

可在main函数加上这个:

SEED = 1 np.random.seed(SEED) torch.manual_seed(SEED) torch.cuda.manual_seed_all(SEED) torch.backends.cudnn.deterministic = True torch.backends.cudnn.benchmark = False

但是我在数据加载的地方也用到了np.random函数,所以要在数据加载的init函数里加上:

SEED = 1 np.random.seed(SEED)

2. 存储整个模型,而不是用state_dict()

torch.save(model, path) #存储 model = torch.load(path) #加载,不用事先定义模型

3.数据预处理不一致

数据预处理阶段用到了 np.random.normal,如果测试时重新加载数据的话,模型效果就会变差。

需要保证数据预处理是一致的,最好的方法就是只处理一次数据,然后将处理后的数据保存到文件,多次运行模型时直接调用该数据文件。

这个改正后,重新加载测试与train后测试的结果完全一样。

4. model.eval()

测试之前一定要用



【本文地址】

公司简介

联系我们

今日新闻


点击排行

实验室常用的仪器、试剂和
说到实验室常用到的东西,主要就分为仪器、试剂和耗
不用再找了,全球10大实验
01、赛默飞世尔科技(热电)Thermo Fisher Scientif
三代水柜的量产巅峰T-72坦
作者:寞寒最近,西边闹腾挺大,本来小寞以为忙完这
通风柜跟实验室通风系统有
说到通风柜跟实验室通风,不少人都纠结二者到底是不
集消毒杀菌、烘干收纳为一
厨房是家里细菌较多的地方,潮湿的环境、没有完全密
实验室设备之全钢实验台如
全钢实验台是实验室家具中较为重要的家具之一,很多

推荐新闻


图片新闻

实验室药品柜的特性有哪些
实验室药品柜是实验室家具的重要组成部分之一,主要
小学科学实验中有哪些教学
计算机 计算器 一般 打孔器 打气筒 仪器车 显微镜
实验室各种仪器原理动图讲
1.紫外分光光谱UV分析原理:吸收紫外光能量,引起分
高中化学常见仪器及实验装
1、可加热仪器:2、计量仪器:(1)仪器A的名称:量
微生物操作主要设备和器具
今天盘点一下微生物操作主要设备和器具,别嫌我啰嗦
浅谈通风柜使用基本常识
 众所周知,通风柜功能中最主要的就是排气功能。在

专题文章

    CopyRight 2018-2019 实验室设备网 版权所有 win10的实时保护怎么永久关闭