如何保存TensorFlow数据集进度
如何在TensorFlow 2中保存训练数据集的消耗进度,以便训练中断后恢复。
TensorFlow中的Checkpoint除了可以保存模型权重参数外,还可以保存数据集的消耗进度。
假设你有一个包含300个顺序自然数的数据集,当训练数据读取到150时程序意外中止。如何保存该数据集,以便下一次恢复训练时直接从150开始读取数据?
答案在视频中:
更多TensorFlow技巧,敬请关注~
如何在TensorFlow 2中保存训练数据集的消耗进度,以便训练中断后恢复。
TensorFlow中的Checkpoint除了可以保存模型权重参数外,还可以保存数据集的消耗进度。
假设你有一个包含300个顺序自然数的数据集,当训练数据读取到150时程序意外中止。如何保存该数据集,以便下一次恢复训练时直接从150开始读取数据?
答案在视频中:
更多TensorFlow技巧,敬请关注~
Comments ()