Agent Skills
返回列表
💻

神经网络设计与训练助手

开发编程 更新于 2026.08.30

将以下提示词粘贴到你的 AI 对话框中:

请根据 https://skillhub.cn/install/skillhub.md,安装 @user_3c6cb52e/self-ent-tech-neural-network。

技能介绍

要解决的问题

模型能跑,不等于训练得好。常见的困境是:任务刚到手,CNN、RNN、Transformer 该选哪一个没有依据;写一段 PyTorch 或 TensorFlow 代码后,loss 曲线抖动、验证集上不去、一收敛就过拟合;照着论文复现时,超参数和训练配置又被大量省略,最终效果对不上。

技能如何工作

这个技能把流程拆成可执行的步骤:

  1. 分析任务需求:先确认是图像、NLP、时序还是生成任务。
  2. 选择网络架构:在 CNN、RNN、Transformer、扩散模型之间做取舍。
  3. 设计网络层与参数:给出可落地的层配置,而不是只给名称。
  4. 编写训练代码:生成 Model(nn.Module) 之类的 PyTorch 结构,或对应的 TensorFlow 实现。
  5. 调优策略:覆盖学习率调度、正则化、混合精度,而不是只换优化器。

训练配置示例:优化器 AdamW,学习率 3e-4,Epochs 100。技能约束包括:代码可直接训练、包含训练曲线监控、讨论过拟合应对策略。互动上,每 2 到 3 个步骤会确认理解或邀请参与,并用动手挑战把知识转化为操作。进度系统分为 🌱 初学者、🌿 进阶者、🌳 熟练者、🌟 专家级,逐步解锁更深层内容。

适用边界

它更擅长架构设计、训练代码组织和调优思路讨论;不负责替代数据清洗、分布式系统部署,也不自动保证某个指标达标。请输入“退出”结束当前会话。

使用场景

  • 接到图像分类任务,需要先判断用 CNN 还是 Transformer 并写出可跑通的 PyTorch/TensorFlow 训练代码
  • 已有训练脚本但效果不佳,需要逐项检查学习率调度、正则化与混合精度等调优策略
  • 想复现论文中的模型结构,需要把架构描述转成具体网络层、参数与训练配置
  • 训练时过拟合明显,需要结合训练曲线监控来调整模型容量和正则化方式

适合人员

  • 刚上手深度学习的算法工程师,需要从零搭出可训练的 PyTorch/TensorFlow 模型
  • 训练效果不稳定、想系统排查学习率和正则化策略的模型工程师
  • 正在复现论文并需要把文字描述转成网络层与参数设置的研究员
  • 负责完整模型训练闭环、需要训练曲线监控与过拟合应对的工程团队