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

Fix lint errors

Browse Source
This commit is contained in:
Matt Conley 2019-04-16 11:28:38 -07:00
parent ac3775ae75
commit e73a77d19d
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -196,7 +196,7 @@ class FixedLossScale(LossScale):
def _is_all_finite(grads):
"""Returns a scalar boolean tensor indicating if all gradients are finite."""
is_finite_per_grad = [
math_ops.reduce_all(math_ops.is_finite(g)) for g in grads if g is not None
math_ops.reduce_all(math_ops.is_finite(g)) for g in grads if g is not None
]
return math_ops.reduce_all(is_finite_per_grad)

2
tensorflow/python/training/experimental/loss_scale_test.py
View File

@ -253,7 +253,7 @@ class DynamicLossScaleTest(test.TestCase, parameterized.TestCase):
self._test_helper(inputs, expected_outputs, init_loss_scale)
@test_util.run_in_graph_and_eager_modes
def test_update_with_None_gradients(self, strategy_fn):
def test_update_with_none_gradients(self, strategy_fn):
with strategy_fn().scope():
loss_scaler = loss_scale_module.DynamicLossScale()
loss_scaler.update([None])