Tensorflow渴望者的评估模式

Question

我Pytorch有eval()模式的概念，该概念在设置后会影响模型的某些层，例如停用退出功能。

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

谢谢。

Answer 1

如果您使用的是Keras，则learning phase有一个类似的概念，如果通过set_learning_phase将该值设置为1（即“测试”），则会禁用某些层，例如辍学。