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

@浙大疏锦行

相关推荐
是枚小菜鸡儿吖10 分钟前
卷不动了?带你拆解 2026 深度学习核心版图:CNN、Transformer 与扩散模型的实战进化
深度学习·cnn·transformer
泯泷11 分钟前
当AI排行榜成为一场数字游戏
人工智能·产品
神一样的老师12 分钟前
【RT-Thread Titan Board 开发板】家庭AI相框
人工智能
靴子学长25 分钟前
Decoder only 架构下 - KV cache 的理解
pytorch·深度学习·算法·大模型·kv
智算菩萨26 分钟前
【OpenGL】10 完整游戏开发实战:基于OpenGL的2D/3D游戏框架、物理引擎集成与AI辅助编程指南
人工智能·python·游戏·3d·矩阵·pygame·opengl
刘简爱学习36 分钟前
弱监督互斥多类脑肿瘤图像分割的类间可分离性损失
人工智能·深度学习·计算机视觉
AI英德西牛仔43 分钟前
AI复制的文字带星号
人工智能·ai·chatgpt·豆包·deepseek·ds随心转
卖报的大地主1 小时前
扩散薛定谔桥(Diffusion Schrödinger Bridge)
人工智能
向成科技1 小时前
当“超轻量AI”遇上“最强国产芯”
人工智能·物联网·ai·芯片·国产化·硬件·主板
远见阁1 小时前
智能体是如何“思考”的:ReAct模式
人工智能·ai·ai智能体