python - TensorFlow Eager 的评估模式

Question

我 Pytorch 有这个eval()模式的概念，当设置它时，会影响模型的某些层，例如停用 dropout。

我正在使用带有 Eager//Keras 模型子类化的 Tensorflow 1.9，tf.data.Dataset我想知道是否有它的等价物，因为我不想在验证或测试阶段退出/批处理规范。

谢谢你。

标签： pythontensorflow

Answer 1

如果您使用的是 Keras，则有一个类似的学习阶段概念，如果将 via 设置为 1（即“测试”），则会set_learning_phase禁用某些层，例如 dropout。