tensorflow静态图的dropout问题

Question

问一个问题：
dropout是在训练的时候用，那怎么才能在测试的时候关闭dropout呢？
tensorflow会自动关闭么？还是需要手动设置keep_prob=1.0 ?

如果我想把训练和测试分开，也就是训练完成后把模型保存到文件中，然后再从文件中读取模型。 这里有一个疑问是，这个图（graph）是在训练的时候建立的，那么如果用if语句把dropout关闭，可以么？

例如像下面这样：

if self.status == "training":
    keep_prob = 0.8
elif self.status == "test":
    keep_prob = 1.0
tensorflow.nn.dropout(x, keep_prob)

感觉有点问题，感觉这个改变了计算图，

Answer 1

不用,你在加载模型是加个training=False