首页 > 解决方案 > 如何在 Tensorflow 的第一个纪元后保存进度

问题描述

我有一个模型,启用急切执行需要 10 秒/纪元来训练,而图形执行需要 1 秒/纪元。然而,随着图形的执行,它在第一个 epoch 上花费了 90 秒 - 大概是创建/优化图形。

有没有办法保存/加载此步骤,以便我可以快速重新训练模型以使用新数据,而无需每次等待 90 秒来创建图形?

我尝试保存/加载模型,但没有任何效果。

标签: tensorflowkeras

解决方案


推荐阅读