TensorFlow推荐使用[tf.train.MonitoredSession](https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/train/MonitoredSession)
,[tf.train.MonitoredTrainingSession](https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/train/MonitoredTrainingSession)
进行训练,并使用[tf.train.SessionRunHook](https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/train/SessionRunHook)
进行训练流程控制。为方便此类用户使用,我们提供了相关压缩工具的hook,初始化hook之后,即可使用。
这些hook的构造方法和使用的压缩算法文档中一致,在此不再详述。
注意,有些hook需要在构造反向图之前创建,例如量化的hook。目前提供的hook有:
- SNIPLevelPrunerHook
- SIMDOCPrunerHook
- TaylorFOChannelPrunerHook
- LowRankDecompositionHook
- EMAQuantizerHook
- OAQQuantizerHook
- LSQQuantizerHook