Internal change

PiperOrigin-RevId: 294999858 Change-Id: I83d5b1a410a8b3745126cfa4bf6ebfb6b4938d46
xzllxls · Feb 13, 2020 · 4e2bdbf · 4e2bdbf
1 parent 2abdae7
commit 4e2bdbf
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tf_slim/learning.py b/tf_slim/learning.py
@@ -263,6 +263,7 @@ def InitAssignFn(sess):
 from tensorflow.python.framework import errors
 from tensorflow.python.framework import ops
 from tensorflow.python.lib.io import file_io
+from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import clip_ops
 from tensorflow.python.ops import control_flow_ops
 from tensorflow.python.ops import lookup_ops
@@ -680,8 +681,8 @@ def train(train_op,
           else:
             local_init_op = control_flow_ops.group(
                 *[opt.local_step_init_op for opt in sync_optimizer])
-        ready_for_local_init_op = control_flow_ops.group(
-            *[opt.ready_for_local_init_op for opt in sync_optimizer])
+        ready_for_local_init_op = array_ops.concat(
+            [opt.ready_for_local_init_op for opt in sync_optimizer], axis=0)
       else:
         ready_for_local_init_op = None