知己AIModel Training · seb1n/awesome-ai-agent-skills
端到端训练机器学习模型
覆盖数据加载、预处理、架构选择、训练循环、验证和保存检查点的完整流程，支持PyTorch、TensorFlow等主流框架，适合搭建或优化训练流水线。
适合做什么
- 搭建新模型的训练流水线
- 为现有训练循环添加验证和早停
- 导出训练好的模型用于部署
- 分类
- 开发
开源技能由原作者维护，知己AI 按原样收录并注明来源。效果取决于任务与资料是否匹配。
试试这样开始
帮我给这批标注图片训练一个分类模型，加上验证并保存最优检查点。