Day44 训练和测试的规范写法

1. 彩色 / 灰度图的训练测试规范:"封装在函数里"

简单说就是把 "训练模型" 和 "测试模型" 的代码,分别写成两个独立的函数(比如叫train()test())。

  • 好处:代码更整洁,想训练就调用train(),想测试就调用test(),后续复用也方便。

2. 展平操作:"除 batchsize 外全部展平"

比如你有一张彩色图,形状是[batchsize, 高, 宽, 通道数](比如[32, 28, 28, 3],代表 32 张 28×28 的 3 通道彩色图)。"展平" 就是把高、宽、通道数这几个维度合并成一个维度,变成[32, 28×28×3](也就是[32, 2352])。

  • 目的:把图片的 "二维像素 + 通道" 变成 "一维向量",方便后续模型计算。

3. dropout 操作:"训练丢、测试关"

dropout 是防止模型 "学太死" 的技巧:

  • 训练阶段:随机让一部分神经元 "罢工"(不参与计算),避免模型过度依赖某些神经元。
  • 测试阶段 :要切换到eval模式,把 dropout 关掉(所有神经元都工作),这样才能得到稳定的测试结果。

@浙大疏锦行

相关推荐
tanis_207723 分钟前
学术论文 PDF 的版面自动还原:MinerU 对多栏排版、浮动图表与脚注区域的识别实战
人工智能·pdf·ocr
guo_xiao_xiao_27 分钟前
YOLOv11室内果蔬展示苹果目标检测数据集-37张-apple-1_3
人工智能·yolo·目标检测
AI学长33 分钟前
数据集|草莓成熟度目标检测数据集-3类530张图
人工智能·目标检测·草莓成熟度目标检测数据集
湘美书院--湘美谈教育37 分钟前
湘美书院谈AI教育经验集:如何用AI整理湖湘文化经义大略
大数据·人工智能·深度学习·神经网络·机器学习
月落归舟39 分钟前
深度讲解 AI Agent 完整运行全流程
人工智能
love在水一方43 分钟前
【翻译】NavDreamer: Video Models as Zero-Shot 3D Navig
人工智能·机器学习
byte轻骑兵43 分钟前
【HID】规范精讲[11]: 蓝牙HID设备信号交互流程深度拆解
人工智能·交互·hid·蓝牙键盘·蓝牙鼠标
AIDF20261 小时前
看破 AI 的“马甲”——从算子到 ChatGPT
人工智能·chatgpt·应用·模型·算子
IT大白鼠1 小时前
AIGC+教育:个性化学习、AI助教、内容生产,教育行业的变革路径
人工智能·学习·aigc
AI医影跨模态组学1 小时前
NPJ Precis Oncol(IF=8)哈尔滨医科大学附属肿瘤医院韩鹏等团队:一种可解释的深度学习生物标志物用于胃癌预后评估及辅助化疗获益预测
人工智能·深度学习·论文·医学·医学影像·影像组学