项目概览
ML Forge 是一款可视化 PyTorch 流程编辑器,用户无需手写训练代码,就能在画布上连接数据预处理、模型层和训练配置节点。它会生成并运行图像分类训练流程,实时显示损失曲线,支持保存检查点、对训练后模型执行推理,并导出为清晰的 PyTorch 项目。使用前需安装 Python 3.10 以上版本,且 PyTorch 与 torchvision 需要单独预装。
仓库事实
- 主要语言
- Python
- 许可证
- MIT
- 仓库更新时间
- 2026年7月11日
- 默认分支
- main
资源类型
桌面应用
使用场景
软件开发
运行方式
桌面端
适合人群
开发者