Skip to content

Commit a6c53b0

Browse files
CyberZHGseanpmorgan
authored andcommitted
Fix group operator for RectifiedAdam (#521)
1 parent 4cc2b70 commit a6c53b0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow_addons/optimizers/rectified_adam.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,7 @@ def _resource_apply_dense(self, grad, var):
203203
var_update = var.assign_sub(
204204
lr_t * var_t, use_locking=self._use_locking)
205205

206-
updates = [var_update + m_t, v_t]
206+
updates = [var_update, m_t, v_t]
207207
if self.amsgrad:
208208
updates.append(vhat_t)
209209
return tf.group(*updates)

0 commit comments

Comments
 (0)