neural_compressor.compression.distillation.utility
This is an utility file for PyTorch distillation.
Module Contents
Functions
|
Record layers output. |
|
Get a hook for getting activation. |