From 771ed0e29e552202f0836c44db06e2053daae93c Mon Sep 17 00:00:00 2001 From: younssas <161091103+younssas@users.noreply.github.com> Date: Wed, 1 Jan 2025 16:48:48 +0100 Subject: [PATCH] Update factorized_top_k.py Bug fix, add_weight expects shape as first argument, and takes 'counter' as shape. Specified "counter" as name argument. --- tensorflow_recommenders/layers/factorized_top_k.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_recommenders/layers/factorized_top_k.py b/tensorflow_recommenders/layers/factorized_top_k.py index b6a7f45..6159803 100644 --- 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,