diff options
| author |  Mark Daoust <markdaoust@google.com> | 2018-07-02 13:37:38 -0700 |
|---|---|---|
| committer |  GitHub <noreply@github.com> | 2018-07-02 13:37:38 -0700 |
| commit | 11cd70438e7d7104904bf8f3b24fcaf6fd88eab5 (patch) | |
| tree | 29b1434fec70840fe55cce035e93a7d21f88a60c /tensorflow/contrib/model_pruning | |
| parent | 90b28b7316edb644b71b01edaaa8553d5913fc19 (diff) | |
Fix lint error.
Diffstat (limited to 'tensorflow/contrib/model_pruning')
| -rw-r--r-- | tensorflow/contrib/model_pruning/python/pruning.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/model_pruning/python/pruning.py b/tensorflow/contrib/model_pruning/python/pruning.py index d843fa26d5..da9d398cbc 100644 --- a/tensorflow/contrib/model_pruning/python/pruning.py +++ b/tensorflow/contrib/model_pruning/python/pruning.py @@ -520,7 +520,8 @@ class Pruning(object): thresholds = get_thresholds() for mask, threshold in zip(masks, thresholds): if not self._exists_in_do_not_prune_list(mask.name): - summary.scalar(mask.op.name + '/sparsity', nn_impl.zero_fraction(mask)) + summary.scalar(mask.op.name + '/sparsity', + nn_impl.zero_fraction(mask)) summary.scalar(threshold.op.name + '/threshold', threshold) def print_hparams(self): |