首页 > 解决方案 > Keras:如何在前向和反向传播过程中忽略特定的神经网络权重?

问题描述

对于每个密集层,在将其一些权重设置为 0.0 后,我希望它们在推理和训练期间都被忽略。它们不会影响模型的损失,我以后可以只使用剩余的(非忽略的)权重进行重新训练。

我知道 Keras 的Masking layer,但我不能让它只屏蔽一个层中的单个权重(那些等于一个值,比如 0.0)。显然,它要求一行中的所有列都等于该值(根据我的喜好为 0.0)以屏蔽该行,而我想屏蔽 0.0 的权重而不影响非 0.0 的权重。

标签: pythontensorflowkeras

解决方案


推荐阅读