如何在Tensorflow的新图中重用自动编码器的编码部分?

时间:2018-05-24 23:22:51

标签: tensorflow subgraph

我可以想象两种方法,但我不知道是否有更好的方法。我想象的两种方法是:

  1. 在同一个会话中,训练自动编码器后,只需使用自动编码器的编码子图作为输入构建一个新图形
  2. 训练自动编码器后,保存训练过的权重。这样,您就不必在同一会话中训练自动编码器和新的其他网络。 (方法1的变体的种类)

1 个答案:

答案 0 :(得分:0)

最容易想到的是像往常一样运行编码器(训练模式),但不向优化器提供sess.run()函数(这会影响训练后的编码器的权重。这样您就可以重复使用编码器而无需构造第二张图,或者您还拥有优点,那就是您可以在训练期间重用!