experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

remove the added api on loss_scale

Browse Source
This commit is contained in:
zilinzhu 2020-03-30 09:25:43 +08:00
parent df6e5c6b59
commit ce592f32ee
2 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/training/experimental/loss_scale.py
View File

@ -197,10 +197,6 @@ class LossScale(trackable.Trackable):
"""Creates the LossScale from its config."""
return cls(**config)
def variables(self):
"""Returns the variables defined in this LossScale."""
return list(self._weights.values())
def get_loss_scale_weights(loss_scale):
return loss_scale._weights.values() # pylint: disable=protected-access

3
tensorflow/python/training/experimental/loss_scale_optimizer.py
View File

@ -246,4 +246,5 @@ class MixedPrecisionLossScaleOptimizer(optimizer.Optimizer):
def variables(self):
"""Returns the variables of the Optimizer."""
return self._optimizer.variables() + self._loss_scale.variables()
return self._optimizer.variables() + \
list(self._loss_scale._weights.values()) # pylint: disable=protected-access