Index

A

adaptor() (built-in function) ADAPTOR_KEYS (in module textbrewer.presets) att_ce_loss() (in module textbrewer.losses)

att_ce_mean_loss() (in module textbrewer.losses) att_mse_loss() (in module textbrewer.losses) att_mse_sum_loss() (in module textbrewer.losses)

B

BasicDistiller (class in textbrewer)

BasicTrainer (class in textbrewer)

C

callback() (built-in function)

cos_loss() (in module textbrewer.losses)

D

deleting() (in module textbrewer.data_utils)

display_parameters() (in module textbrewer.utils) DistillationConfig (class in textbrewer)

F

from_dict() (textbrewer.DistillationConfig class method) (textbrewer.TrainingConfig class method)

from_json_file() (textbrewer.DistillationConfig class method) (textbrewer.TrainingConfig class method) fsp_loss() (in module textbrewer.losses)

G

GeneralDistiller (class in textbrewer)

H

hid_mse_loss() (in module textbrewer.losses)

K

KD_LOSS_MAP (in module textbrewer.presets)

L

long_disorder() (in module textbrewer.data_utils)

M

masking() (in module textbrewer.data_utils) MATCH_LOSS_MAP (in module textbrewer.presets)

mmd_loss() (in module textbrewer.losses) MultiTaskDistiller (class in textbrewer) MultiTeacherDistiller (class in textbrewer)

N

n_gram_sampling() (in module textbrewer.data_utils)

P

pkd_loss() (in module textbrewer.losses)

PROJ_MAP (in module textbrewer.presets)

S

short_disorder() (in module textbrewer.data_utils)

T

TEMPERATURE_SCHEDULER (in module textbrewer.presets) train() (textbrewer.BasicDistiller method) (textbrewer.BasicTrainer method) (textbrewer.GeneralDistiller method) (textbrewer.MultiTaskDistiller method) (textbrewer.MultiTeacherDistiller method)

TrainingConfig (class in textbrewer)

W

WEIGHT_SCHEDULER (in module textbrewer.presets)