Tensorflow渴望者的评估模式

Question

在Pytorch中，有一个eval()模式的概念，当设置该模式时，它会影响模型的某些层，例如禁用退出。

我正在将Tensorflow 1.9与eager / tf.data.Dataset / tf.data.Dataset模型子类一起使用，我想知道是否存在等效项，因为我不想在验证或测试阶段退出/批量处理规范。

谢谢。

Answer 1

如果您使用的是Keras，则存在学习阶段的类似概念，如果通过set_learning_phase设置为1（即“测试”），则禁用某些层，例如辍学。