记一次训练实操
前情
了解了一次 datawhale 的4月活动《免费使用GPU资源》 。本着能薅就薅的原则进来了。操作手册很简单,有一份很详细的操作指南。小白也能轻松上手。
这个指南提供了几个方面。
- 一个有免费金的线上运行环境
- 环境上提供了GPU资源
- 环境上提供了一个简单的Python 机器学习代码用于实践
- 环境上准备好了用于训练的数据
- …
简单看下代码
这是一个用于猫狗分类的训练模型。数据是通过path 打好了标签。类似于 dog1.jpg, cat1.jpg
使用的 tensorflow 机器学习框架。tensflow可以产生 model(模型)。调节执行次数,最后对比模型训练结果。
整体代码 tensorflow
的api。基本不需要其他额外代码实现。
TensorFlow 首先处理数据,然后建立模型,进行训练,反馈训练结果等。基本上的功能都包全了。
最后
因为我是小白嘛,预期真的不高,作为一个爱好者,能跑起来就行。整个流程跑下来不难,可能后续优化起来还需要 TensorFlow
的相关知识去完成进阶。
引入
https://datawhaler.feishu.cn/docx/QaujdWRcJoFi6yxr16DciPThndd