桌面应用软件开发

ml_forge

拖拽节点构建、训练并导出图像分类模型

Stars
417
Forks
50
License
MIT
更新于
更新于 2026年7月11日

正在检查仓库实时数据……

项目概览

ML Forge 是一款可视化 PyTorch 流程编辑器,用户无需手写训练代码,就能在画布上连接数据预处理、模型层和训练配置节点。它会生成并运行图像分类训练流程,实时显示损失曲线,支持保存检查点、对训练后模型执行推理,并导出为清晰的 PyTorch 项目。使用前需安装 Python 3.10 以上版本,且 PyTorch 与 torchvision 需要单独预装。

仓库事实

主要语言
Python
许可证
MIT
仓库更新时间
2026年7月11日
默认分支
main

资源类型

桌面应用

使用场景

软件开发

运行方式

桌面端

适合人群

开发者

相关项目

查看更多同类项目