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 关掉(所有神经元都工作),这样才能得到稳定的测试结果。

@浙大疏锦行

相关推荐
Spring AI学习2 小时前
Spring AI深度解析(11/50):异常处理与容错机制实战
java·人工智能·spring
小兔崽子去哪了2 小时前
机器学习,KNN 算法
后端·python·机器学习
木头左2 小时前
结合基本面分析的LSTM量化交易模型入参设计原则
人工智能·rnn·lstm
HuggingFace2 小时前
流式数据集:效率提升 100 倍!
人工智能
qq_12498707532 小时前
基于协同过滤算法的在线教育资源推荐平台的设计与实现(源码+论文+部署+安装)
java·大数据·人工智能·spring boot·spring·毕业设计
万俟淋曦2 小时前
【论文速递】2025年第38周(Sep-14-20)(Robotics/Embodied AI/LLM)
人工智能·深度学习·机器学习·机器人·大模型·论文·具身智能
一水鉴天2 小时前
整体设计 定稿 之7 共享给定表格文档的分析(豆包助手)
人工智能·架构
C嘎嘎嵌入式开发2 小时前
NLP 入门:从原理到实战的个人经验总结
人工智能·python·自然语言处理·nlp
水如烟3 小时前
孤能子视角:人工智能的“计算博弈“––“标量“即“矢量“
人工智能