ml_forge
拖拽节点构建、训练并导出图像分类模型
- Stars
- 417
- Forks
- 50
- 更新时间
- 更新于 2026年7月11日
拖拽节点构建、训练并导出图像分类模型
ML Forge 是一款可视化 PyTorch 流程编辑器,用户无需手写训练代码,就能在画布上连接数据预处理、模型层和训练配置节点。它会生成并运行图像分类训练流程,实时显示损失曲线,支持保存检查点、对训练后模型执行推理,并导出为清晰的 PyTorch 项目。使用前需安装 Python 3.10 以上版本,且 PyTorch 与 torchvision 需要单独预装。
资源类型
使用场景
运行方式
适合人群
GitHub 公开事实最近同步于 2026年7月14日。
使用 VS Code 的开源核心编辑、导航、调试并扩展代码工作流。
集中切换多款 AI 编程工具的供应商、MCP、提示词与技能配置
在终端环境中使用内置或第三方编码代理
用桌面、CLI 或 API 让 AI 安装、编辑、执行并测试代码