我正在使用自动编码器。我的检查点包含网络的完整状态(即编码器、解码器、优化器等)。我想玩弄编码。因此,在我的评估模式中,我只需要网络的解码器部分。
如何仅从现有检查点读取一些特定变量,以便我可以在另一个模型中重用它们的值?
最佳答案
有list_variables
checkpoint_utils.py 中的方法它可以让你看到所有保存的变量。
但是,对于您的用例,使用 Saver
进行恢复可能更容易。 .如果您在保存检查点时知道变量的名称,则可以创建一个新的保存程序,并告诉它将这些名称初始化为新的 Variable
对象(可能具有不同的名称)。这在 CIFAR 示例中用于选择恢复 subset of variables .见 Choosing which Variables to Save and Restore在 Howto
关于variables - 如何在检查点中列出某些变量?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/38944238/