Tensorflow正向传递与丢失

时间:2018-02-06 00:54:58

标签: tensorflow machine-learning computer-vision deep-learning computer-science

我正在尝试使用dropout来获取神经网络的误差估计值。这涉及在不仅训练而且还在测试中运行我的网络的多个前向传递,并且激活了丢失。辍学层似乎只在训练时激活,而不是测试。可以通过调用某些函数或修改某些参数在Tensorflow中完成吗?

1 个答案:

答案 0 :(得分:2)

是的,最简单的方法是使用tf.layers.dropout个{{3}}参数,该参数可以是张量,您可以trainingtrue定义任何特定参数会话运行:

false