We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fc71ed7 commit 61fb731Copy full SHA for 61fb731
rasa/utils/tensorflow/layers.py
@@ -77,7 +77,9 @@ def build(self, input_shape: tf.TensorShape) -> None:
77
kernel_mask = tf.cast(
78
tf.greater_equal(kernel_mask, self.sparsity), self.kernel.dtype
79
)
80
- self.kernel_mask = tf.Variable(initial_value=kernel_mask, trainable=False)
+ self.kernel_mask = tf.Variable(
81
+ initial_value=kernel_mask, trainable=False, name="kernel_mask"
82
+ )
83
84
def call(self, inputs: tf.Tensor) -> tf.Tensor:
85
# set some weights to 0 according to precomputed mask
0 commit comments