[ResNet/TF] Fix gradient calculation for sync variable

hXl3s · web-flow · commit 2d555548b698 · 2021-04-26T15:00:10.000+02:00
diff --git a/TensorFlow/Classification/ConvNets/model/resnet.py b/TensorFlow/Classification/ConvNets/model/resnet.py
@@ -239,7 +239,8 @@ def __call__(self, features, labels, mode, params):
 
         with tf.device("/cpu:0"):
             if hvd_utils.is_using_hvd():
-                sync_var = tf.Variable(initial_value=[0], dtype=tf.int32, name="signal_handler_var")
+                sync_var = tf.Variable(initial_value=[0], dtype=tf.int32, name="signal_handler_var",
+                                       trainable=False)
                 sync_var_assing = sync_var.assign([1], name="signal_handler_var_set")
                 sync_var_reset = sync_var.assign([0], name="signal_handler_var_reset")
                 sync_op = hvd.allreduce(sync_var, op=hvd.Sum, name="signal_handler_all_reduce")
