- 🍨 本文为🔗365天深度学习训练营 中的学习记录博客
- 🍦 参考文章:365天深度学习训练营 | 第8周:猫狗识别
- 🍖 原作者:K同学啊|接辅导、项目定制
🏡 我的环境:
- 语言环境:Python3.6.5
- 编译器:jupyter lab
- 深度学习环境:TensorFlow2.4.1
- 数据集:参加训练营可获取
这篇文章中我放弃了以往的model.fit()训练方法,改用model.train_on_batch方法。两种方法的比较:
model.fit()
:用起来十分简单,对新手非常友好model.train_on_batch()
:封装程度更低,可以玩更多花样。
此外我也引入了进度条的显示方式,更加方便我们及时查看模型训练过程中的情况,可以及时打印各项指标。
🏡 我的环境:
- 语言环境:Python3.6.5
- 编译器:jupyter notebook
- 深度学习环境:TensorFlow2.4.1