... update e t c

PiperOrigin-RevId: 353000226
keras-team · Jan 21, 2021 · 84806ce · 84806ce
1 parent 9941536
commit 84806ce
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/keras/optimizer_v2/optimizer_v2.py b/keras/optimizer_v2/optimizer_v2.py
@@ -651,6 +651,7 @@ def apply_gradients(self,
         grads_and_vars = self._aggregate_gradients(grads_and_vars)
       grads_and_vars = self._transform_gradients(grads_and_vars)
 
+      # TODO(wxinyi)
       return tf.distribute.get_replica_context().merge_call(
           functools.partial(self._distributed_apply, apply_state=apply_state),
           args=(grads_and_vars,),