Index

A

AttentionHiddenStatesCallback (class in compressors.distillation.callbacks) (class in compressors.distillation.callbacks.hidden_states)

AttentionLoss (class in compressors.distillation.losses), [1]

B

BaseDistilModel (class in compressors.models.base_distil_model)

C

compressors.distillation.callbacks.hidden_states module compressors.distillation.callbacks.logits_diff module compressors.distillation.callbacks.wrappers module compressors.distillation.data.label_smoothing module compressors.distillation.losses module compressors.distillation.runners module compressors.models.base_distil_model module

compressors.models.cv module compressors.pruning.callbacks module compressors.pruning.runners module compressors.pruning.utils module compressors.runners module CosineHiddenStatesCallback (class in compressors.distillation.callbacks.hidden_states) CRDLoss (class in compressors.distillation.losses)

D

DistilRunner (class in compressors.distillation.runners)

E

EndToEndDistilRunner (class in compressors.distillation.runners)

F

forward() (compressors.distillation.losses.AttentionLoss method) (compressors.distillation.losses.CRDLoss method) (compressors.distillation.losses.KLDivLoss method) (compressors.distillation.losses.MSEHiddenStatesLoss method) (compressors.models.base_distil_model.BaseDistilModel method)

G

get_callbacks() (compressors.distillation.runners.EndToEndDistilRunner method)

get_stage_len() (compressors.distillation.runners.EndToEndDistilRunner method)

H

handle_batch() (compressors.pruning.runners.PruneRunner method) (compressors.runners.HFRunner method) HFDistilRunner (class in compressors.distillation.runners)

HFRunner (class in compressors.runners) HiddenStatesSelectCallback (class in compressors.distillation.callbacks) (class in compressors.distillation.callbacks.hidden_states)

K

kl_div_loss() (in module compressors.distillation.losses) KLDivCallback (class in compressors.distillation.callbacks) (class in compressors.distillation.callbacks.logits_diff)

KLDivLoss (class in compressors.distillation.losses) (class in compressors.distillation.losses._kl_loss)

L

LambdaPreprocessCallback (class in compressors.distillation.callbacks) (class in compressors.distillation.callbacks.hidden_states)

LambdaWrapperCallback (class in compressors.distillation.callbacks.wrappers) LogitsDataset (class in compressors.distillation.data) LotteryTicketCallback (class in compressors.pruning.callbacks)

M

module compressors.distillation.callbacks.hidden_states compressors.distillation.callbacks.logits_diff compressors.distillation.callbacks.wrappers compressors.distillation.data.label_smoothing compressors.distillation.losses compressors.distillation.runners compressors.models.base_distil_model compressors.models.cv compressors.pruning.callbacks compressors.pruning.runners compressors.pruning.utils compressors.runners

mse_loss() (in module compressors.distillation.losses) MSEHiddenStatesCallback (class in compressors.distillation.callbacks) (class in compressors.distillation.callbacks.hidden_states) MSEHiddenStatesLoss (class in compressors.distillation.losses), [1]

O

on_experiment_start() (compressors.pruning.callbacks.PrepareForFinePruningCallback method)

on_stage_start() (compressors.pruning.callbacks.LotteryTicketCallback method)

P

pkt_loss() (in module compressors.distillation.losses) PKTHiddenStatesCallback (class in compressors.distillation.callbacks.hidden_states) predict_batch() (compressors.distillation.runners.EndToEndDistilRunner method) (compressors.pruning.runners.PruneRunner method)

PrepareForFinePruningCallback (class in compressors.pruning.callbacks) probability_shift() (in module compressors.distillation.data.label_smoothing) PruneRunner (class in compressors.pruning.runners)

R

resnet101() (in module compressors.models.cv) resnet152() (in module compressors.models.cv) resnet18() (in module compressors.models.cv) resnet34() (in module compressors.models.cv) resnet50() (in module compressors.models.cv) resnet_cifar_110() (in module compressors.models.cv) resnet_cifar_14() (in module compressors.models.cv) resnet_cifar_20() (in module compressors.models.cv)

resnet_cifar_32() (in module compressors.models.cv) resnet_cifar_32x4() (in module compressors.models.cv) resnet_cifar_44() (in module compressors.models.cv) resnet_cifar_56() (in module compressors.models.cv) resnet_cifar_8() (in module compressors.models.cv) resnet_cifar_8x4() (in module compressors.models.cv) resnext101_32x8d() (in module compressors.models.cv) resnext50_32x4d() (in module compressors.models.cv)

S

stages() (compressors.distillation.runners.EndToEndDistilRunner property) (compressors.pruning.runners.PruneRunner property)

W

wide_resnet101_2() (in module compressors.models.cv)

wide_resnet50_2() (in module compressors.models.cv)