如何保存TensorFlow数据集进度

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

如何保存TensorFlow数据集进度

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

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

答案在视频中:

TensorFlow 2如何保存训练数据集进度_哔哩哔哩 (゜-゜)つロ 干杯~-bilibili
%E5%A6%82%E4%BD%95%E5%9C%A8TensorFlow%202%E4%B8%AD%E4%BF%9D%E5%AD%98%E8%AE%AD%E7%BB%83%E6%95%B0%E6%8D%AE%E9%9B%86%E7%9A%84%E6%B6%88%E8%80%97%E8%BF%9B%E5%BA%A6%EF%BC%8C%E4%BB%A5%E4%BE%BF%E8%AE%AD%E7%BB%83%E4%B8%AD%E6%96%AD%E5%90%8E%E6%81%A2%E5%A4%8D%E3%80%82

更多TensorFlow技巧,敬请关注~