在TensorFlow中添加遮罩以进行训练/推理

时间:2018-10-09 04:16:28

标签: python tensorflow masking

我正在使用tf.contrib.slim来训练和评估最新的CNN。我想知道是否有一种方法可以使用tf.selecttf.where创建一个掩码来忽略某些权重在确定(甚至推论)过程中的计算。说,我想忽略值为0的权重并继续训练模型。可能吗?

任何建议/代码将不胜感激。

0 个答案:

没有答案