Skip to content

Commit 61fb731

Browse files
committed
add name to kernel_mask
1 parent fc71ed7 commit 61fb731

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

rasa/utils/tensorflow/layers.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,9 @@ def build(self, input_shape: tf.TensorShape) -> None:
7777
kernel_mask = tf.cast(
7878
tf.greater_equal(kernel_mask, self.sparsity), self.kernel.dtype
7979
)
80-
self.kernel_mask = tf.Variable(initial_value=kernel_mask, trainable=False)
80+
self.kernel_mask = tf.Variable(
81+
initial_value=kernel_mask, trainable=False, name="kernel_mask"
82+
)
8183

8284
def call(self, inputs: tf.Tensor) -> tf.Tensor:
8385
# set some weights to 0 according to precomputed mask

0 commit comments

Comments
 (0)