首页 > 解决方案 > 在 TensorFlow 中添加掩码以进行训练/推理

问题描述

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

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

标签: pythontensorflowmasking

解决方案


推荐阅读