Mentions légales du service

Skip to content
Snippets Groups Projects
Verified Commit 2babbf18 authored by ANDREY Paul's avatar ANDREY Paul
Browse files

Fix 'TensorflowModel.apply_updates'.

parent 62683d63
No related branches found
No related tags found
No related merge requests found
Pipeline #764291 waiting for manual action
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment