neural_compressor.experimental.common.torch_utils

This is an utility file for PyTorch distillation.

Module Contents

Functions

record_output(output, name, output_process[, student])	Record layers output.
get_activation(name[, output_process, student])	Get a hook for getting activation.

neural_compressor.experimental.common.torch_utils.record_output(output, name, output_process, student=False)

Record layers output.

It is a help function.

neural_compressor.experimental.common.torch_utils.get_activation(name, output_process='', student=False)

Get a hook for getting activation.