neural_compressor.compression.distillation.utility

This is an utility file for PyTorch distillation.

Functions

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

Module Contents

neural_compressor.compression.distillation.utility.record_output(output, name, output_process, student=False)

Record layers output.

It is a help function.

neural_compressor.compression.distillation.utility.get_activation(name, output_process='', student=False)

Get a hook for getting activation.