如何保存TensorFlow数据集进度

如何在TensorFlow 2中保存训练数据集的消耗进度,以便训练中断后恢复。

如何保存TensorFlow数据集进度

TensorFlow中的Checkpoint除了可以保存模型权重参数外,还可以保存数据集的消耗进度。

假设你有一个包含300个顺序自然数的数据集,当训练数据读取到150时程序意外中止。如何保存该数据集,以便下一次恢复训练时直接从150开始读取数据?

答案在视频中:

TensorFlow 2如何保存训练数据集进度_哔哩哔哩_bilibili
如何在TensorFlow 2中保存训练数据集的消耗进度,以便训练中断后恢复。

更多TensorFlow技巧,敬请关注~

转发至

微信扫一扫分享

WeChat QR Code

有AI相关的想法,但不知道怎么落地?

我是国冰,独立AI开发者。十多年技术一线——半导体显示、智慧城市到自动驾驶;现在一个人+AI,接AI落地项目,从梳理需求到系统落地,陪你走通。

微信聊聊 微信二维码 看我能做什么