corrected issue accessing factorized top-k counter, described here: t…

…ensorflow#712 (comment)
messiest · Oct 22, 2024 · 27b1e6b · 27b1e6b
1 parent 685694e
commit 27b1e6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_recommenders/layers/factorized_top_k.py b/tensorflow_recommenders/layers/factorized_top_k.py
@@ -373,7 +373,7 @@ def __init__(self,
     self._num_parallel_calls = num_parallel_calls
     self._sorted = sorted_order
 
-    self._counter = self.add_weight("counter", dtype=tf.int32, trainable=False)
+    self._counter = self.add_weight(name="counter", dtype=tf.int32, trainable=False)
 
   def index_from_dataset(
       self,