aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/model_pruning
diff options
context:
space:
mode:
authorGravatar Jesse <jessehagenaars@gmail.com>2018-06-05 14:49:04 +0200
committerGravatar GitHub <noreply@github.com>2018-06-05 14:49:04 +0200
commitf9c7fe82cb930ee26d281e4bf21211ed352d176e (patch)
tree084fcc4a12dea8648c8d21e0a5df61c18a001295 /tensorflow/contrib/model_pruning
parentf2e22502fd58e8d81c9e080b9242375fbf2bc772 (diff)
Put some emphasis on incrementing global step
Pruning will not work if the global step is not incremented
Diffstat (limited to 'tensorflow/contrib/model_pruning')
-rw-r--r--tensorflow/contrib/model_pruning/README.md1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/model_pruning/README.md b/tensorflow/contrib/model_pruning/README.md
index 50e7e5d7cd..9143d082bf 100644
--- a/tensorflow/contrib/model_pruning/README.md
+++ b/tensorflow/contrib/model_pruning/README.md
@@ -103,6 +103,7 @@ with tf.graph.as_default():
mon_sess.run(mask_update_op)
```
+Ensure that `global_step` is being [incremented](https://www.tensorflow.org/api_docs/python/tf/train/Optimizer#minimize), otherwise pruning will not work!
## Example: Pruning and training deep CNNs on the cifar10 dataset