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