From 2babbf18b13e894b87ce6217966c338811b104d1 Mon Sep 17 00:00:00 2001
From: Paul Andrey <paul.andrey@inria.fr>
Date: Wed, 1 Mar 2023 16:15:26 +0100
Subject: [PATCH] Fix 'TensorflowModel.apply_updates'.
---
declearn/model/tensorflow/_model.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/declearn/model/tensorflow/_model.py b/declearn/model/tensorflow/_model.py
index 118f9b9d..c7c130c8 100644
--- a/declearn/model/tensorflow/_model.py
+++ b/declearn/model/tensorflow/_model.py
@@ -323,7 +323,7 @@ class TensorflowModel(Model):
for var in self._model.trainable_weights:
updt = updates.coefs[var.name]
if isinstance(updt, tf.IndexedSlices):
- var.scatter_add(updt, read_value=False)
+ var.scatter_add(updt)
else:
var.assign_add(updt, read_value=False)
--
GitLab