搭建一个深度学习项目时,模型只是其中一部分。数据读取、预处理、训练循环、配置管理、预测入口、日志和测试也都要逐项补齐。不同实验反复编写这些工程代码,很容易拖慢验证想法的速度,也会让项目结构和实验记录越来越不一致。

SuperDL-frame 是一个 Claude Code 插件,目标是把这部分重复工作变成一套可配置的生成流程:研究者描述模型、数据和项目需求,插件通过交互式步骤整理设计,再从蓝图生成一份完整的 Python 训练项目。

项目地址:SuShuheng/SuperDL-frame

它解决什么问题

SuperDL-frame 面向 PyTorch 机器学习项目,生成内容覆盖模型接口、数据预处理、数据集、训练与预测流程、配置、命令行入口、测试和打包文件。它不替用户设计模型,也不自动猜测数据语义;模型结构、数据格式和训练需求仍由使用者提供。

它更像一个可定制的工程起点:将常见项目结构和训练流程固化成模板,让新项目从一套相对完整的骨架开始,再由开发者补充具体算法和业务逻辑。

从需求到项目:两种引导流程

插件提供单模型和多模型两种脚手架流程。使用入口会根据项目由一个模型还是多个协作模型组成,引导到对应流程:

  • 单模型项目使用 /superdl-frame:scaffold-project,当前技能定义为 10 个阶段。
  • 多模型项目使用 /superdl-frame:scaffold-multi-project,当前技能定义为 11 个阶段。

流程会逐步梳理模型架构、预处理方式、项目结构、可选模块、命令行设计和开发环境等信息,并在生成前检查接口和模块依赖。多模型项目还会描述模型之间的依赖关系,由生成的流水线按依赖顺序组织训练和预测。

仓库首页 README 仍保留早期的“8 阶段”介绍;阅读当前插件能力时,应以仓库内的单模型和多模型技能文档为准。

蓝图如何生成代码

项目把待生成文件放在 blueprints/ 下,以 .bp 文件保存。蓝图中的插槽标记代表需要根据用户需求填入的内容;条件块控制可选模块对应的代码是否保留;重复块则用于按模型或其他输入列表展开代码。

生成过程可以概括为:

  1. 根据需求确定项目设计和模块组合。
  2. 检查必需信息、接口关系与模块依赖。
  3. 读取对应蓝图,填入用户确认的配置和实现片段。
  4. 根据模块选择保留或移除条件代码,并展开重复内容。
  5. 写出项目文件,再按工作流执行安装、命令行冒烟检查和测试等验证步骤。

这套方式把通用工程结构与项目特定代码分开维护。需要改变生成项目的默认行为时,可以调整对应蓝图或模块规则,而不必每次都从空目录重新搭建。

按需组合模块

生成项目包含模型接口、数据集、配置、训练器、预测器和命令行等基础部分。使用者还可以按项目需要选择内存预加载、断点续训、训练技巧、数据增强、测试时增强、后处理、Agent 训练工作流、实验管理命令和测试等功能。

模块不是互不相关的开关。比如测试时增强依赖数据增强,Agent 工作流需要训练状态和检查点能力。插件会处理模块依赖,对不完整或冲突的选择给出提示,再将最终组合用于生成。

多模型项目会额外生成流水线组件,根据模型之间的依赖关系确定执行顺序。这让“先分割、再分类”一类组合任务能沿用统一的工程组织方式,而不必把所有模型逻辑塞进一个训练脚本。

生成项目包含什么

具体文件取决于选择的模块。典型项目会包含:

  • src/:模型接口、数据处理、训练、预测、命令行和公共工具。
  • configs/:模型、训练和数据等配置文件。
  • tests/:所选测试模块对应的测试代码。
  • pyproject.toml 与 README.md:安装和项目说明文件。
  • 实验目录相关配置:保存训练配置快照、检查点、日志和结果,便于追踪实验。

当前插件还会为生成项目提供 doctor、version、update、tool 和 helpme 等标准命令。它们属于生成出来的项目 CLI;在 Claude Code 中启动脚手架则使用前面提到的插件技能命令。

可复现性方面,项目提供随机种子管理,并可将配置、检查点、日志和训练结果放在结构化目录中。这些机制有助于复现实验和比较结果,但最终复现质量仍取决于数据版本、依赖环境、硬件和训练实现是否一致。

安装与启动

推荐通过 Claude Code 插件市场安装:

1
2
claude plugin marketplace add SuShuheng/SuperDL-frame
claude plugin install superdl-frame

也可以将仓库克隆到 Claude Code 的插件目录。安装后,在 Claude Code 会话中启动单模型流程:

1
/superdl-frame:scaffold-project

如果需求包含两个或更多相互配合的模型,则启动多模型流程:

1
/superdl-frame:scaffold-multi-project

完成生成后,先检查模型接口、数据路径、配置项和依赖版本,再安装项目依赖并运行测试。自动生成的工程骨架可以减少重复搭建,但不能代替对训练逻辑和模型结果的审查。

适用场景与边界

SuperDL-frame 适合经常搭建 PyTorch 实验、希望统一项目结构,或需要组织多个模型训练与预测流程的研究者。它也适合把团队中反复出现的工程做法整理成可复用蓝图。

如果任务只需要一个短小的一次性脚本,完整项目骨架可能显得过重;如果数据格式或算法流程很特殊,也需要修改生成结果或扩展蓝图。使用者应把它看作工程模板和生成助手,而不是自动完成科研设计、保证模型效果的工具。

项目资料