用Checkpoint替换TensorFlow Saver

时间:2019-04-01 17:48:54

标签: tensorflow

我一直在使用TensorFlow的Saver类来保存模型参数,但是该类在TensorFlow 2中已经消失了,因此我需要用Checkpoint替换它。我不知道该怎么做。 Checkpoint文档中的所有示例均假设您要保存tf.keras.Model。我没有使用Keras,所以这并不适用。

Saver只需要保存一个变量列表,所以这就是我要开始的内容。如何将其传递给Checkpoint?它期望您将每个可检查点对象作为命名参数传递。我希望我只能说variables=[var1, var2, ...],但它不接受列表。我可以将每个变量作为一个单独的参数传递,但是我应该使用什么作为名称?变量名?这破坏了检查点的整个目的,即不依赖变量名而变得更强大。在不使用Keras的代码中编写检查点的预期方式是什么?

0 个答案:

没有答案