Fix merge error in 688a03d639

2019-09-14 15:05:29 -07:00 · 2019-09-14 15:05:29 -07:00 · c09880bd0f
commit c09880bd0f
parent a1463b9d18
1 changed files with 1 additions and 1 deletions
--- a/tensorflow/python/training/moving_averages.py
+++ b/tensorflow/python/training/moving_averages.py
@ -444,7 +444,7 @@ class ExponentialMovingAverage(object):
                    "Variable", "VariableV2", "VarHandleOp"
                ]))
            if self._zero_debias:
-              zero_debias_true.add(avg)
+              zero_debias_true.add(avg.experimental_ref())
        self._averages[var.experimental_ref()] = avg

    with ops.name_scope(self.name) as scope: