From e2e6679ac3a7d8d747259dd7346253e5122fbe87 Mon Sep 17 00:00:00 2001
From: Zhao HG <853842+CyberZHG@users.noreply.github.com>
Date: Wed, 18 Sep 2019 19:49:06 +0800
Subject: [PATCH] Fix group operator for RectifiedAdam

---
 tensorflow_addons/optimizers/rectified_adam.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow_addons/optimizers/rectified_adam.py b/tensorflow_addons/optimizers/rectified_adam.py
index ff4d8b3bf4..46c0e42ae9 100644
--- a/tensorflow_addons/optimizers/rectified_adam.py
+++ b/tensorflow_addons/optimizers/rectified_adam.py
@@ -203,7 +203,7 @@ def _resource_apply_dense(self, grad, var):
         var_update = var.assign_sub(
             lr_t * var_t, use_locking=self._use_locking)
 
-        updates = [var_update + m_t, v_t]
+        updates = [var_update, m_t, v_t]
         if self.amsgrad:
             updates.append(vhat_t)
         return tf.group(*updates)