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

@浙大疏锦行

相关推荐
haluhalu.5 分钟前
prompts.chat:07-few-shot-prompting
人工智能·算法
weixin_4398575417 分钟前
在Ubuntu 26.04版本上借助本地AI助手在AMD显卡基础上安装并运行Qwen3.6-35B-A3B,作为本地编程AI
linux·人工智能·ubuntu
YOLO数据集集合21 分钟前
多模态AI道路病害检测系统:YOLO+DeepSeek融合实战
人工智能·yolo·目标检测·计算机视觉·目标跟踪
程序员脚趾8 小时前
GPT-5.6 上线:当 GPT 与 Codex 走向融合,AI 正从“回答问题”走向“完成工作”
人工智能
thubier(段新建)8 小时前
owtb 3pl 面向城市配送物流企业需求V0.2
大数据·人工智能
Token炼金师9 小时前
模型的防线:Prompt 注入防御、越狱攻击与对齐、红队测试、价值观对齐、对抗样本鲁棒性、安全评测与边界 —— 模型安全六防
人工智能·红队测试·prompt 注入防御·越狱攻击与对齐·价值观对齐·对抗样本鲁棒性·安全评测与边界
嘘神秘用9 小时前
布:AI 驱动的 Redis 客户端,更快、更直观
数据库·人工智能·redis
黒亱中旳9 小时前
Java AI 框架三国杀:Solon AI vs Spring AI vs LangChain4j 深度对比
java·人工智能·spring
小和尚同志9 小时前
前端 AI 单元测试思考与落地
前端·人工智能·aigc
alxraves10 小时前
医用超声远程会诊系统:会诊平台的核心架构与功能解析
java·人工智能·架构