KD_Lib
stable
Contents:
KD-Lib
Installation
Tutorials
API:
Knowledge Distillation
Pruning
Quantization
Models
KD_Lib
»
Index
Edit on GitHub
Index
B
|
C
|
D
|
E
|
F
|
G
|
K
|
L
|
M
|
N
|
P
|
Q
|
R
|
S
|
T
B
BaseClass (class in KD_Lib.KD.common.base_class)
BasicBlock (class in KD_Lib.models.resnet)
BERT2LSTM (class in KD_Lib.KD.text.BERT2LSTM.bert2lstm)
Bottleneck (class in KD_Lib.models.resnet)
C
calculate_kd_loss() (KD_Lib.KD.common.base_class.BaseClass method)
(KD_Lib.KD.text.BERT2LSTM.bert2lstm.BERT2LSTM method)
D
df_to_bert_dataset() (in module KD_Lib.KD.text.utils.bert)
df_to_bert_format() (in module KD_Lib.KD.text.utils.bert)
Dynamic_Quantizer (class in KD_Lib.Quantization.dynamic.dynamic_quantization)
E
evaluate() (KD_Lib.KD.common.base_class.BaseClass method)
evaluate_student() (KD_Lib.KD.text.BERT2LSTM.bert2lstm.BERT2LSTM method)
evaluate_teacher() (KD_Lib.KD.text.BERT2LSTM.bert2lstm.BERT2LSTM method)
expansion (KD_Lib.models.resnet.BasicBlock attribute)
(KD_Lib.models.resnet.Bottleneck attribute)
F
forward() (KD_Lib.models.lenet.LeNet method)
(KD_Lib.models.lenet.ModLeNet method)
(KD_Lib.models.lstm.LSTMNet method)
(KD_Lib.models.nin.NetworkInNetwork method)
(KD_Lib.models.resnet.BasicBlock method)
(KD_Lib.models.resnet.Bottleneck method)
(KD_Lib.models.resnet.MeanResnet method)
(KD_Lib.models.resnet.ResNet method)
(KD_Lib.models.resnet.ResnetWithAT method)
(KD_Lib.models.shallow.Shallow method)
G
get_bert_dataloader() (in module KD_Lib.KD.text.utils.bert)
get_model_sizes() (KD_Lib.Quantization.common.base_class.Quantizer method)
get_parameters() (KD_Lib.KD.common.base_class.BaseClass method)
get_performance_statistics() (KD_Lib.Quantization.common.base_class.Quantizer method)
K
KD_Lib.KD.common.base_class (module)
KD_Lib.KD.text.BERT2LSTM.bert2lstm (module)
KD_Lib.KD.text.utils (module)
KD_Lib.KD.text.utils.bert (module)
KD_Lib.models.lenet (module)
KD_Lib.models.lstm (module)
KD_Lib.models.nin (module)
KD_Lib.models.resnet (module)
KD_Lib.models.shallow (module)
KD_Lib.Pruning.lottery_tickets.lottery_tickets (module)
KD_Lib.Quantization.common.base_class (module)
KD_Lib.Quantization.dynamic.dynamic_quantization (module)
KD_Lib.Quantization.qat.qat (module)
KD_Lib.Quantization.static.static_quantization (module)
L
LeNet (class in KD_Lib.models.lenet)
LotteryTicketsPruner (class in KD_Lib.Pruning.lottery_tickets.lottery_tickets)
LSTMNet (class in KD_Lib.models.lstm)
M
MeanResnet (class in KD_Lib.models.resnet)
ModLeNet (class in KD_Lib.models.lenet)
N
NetworkInNetwork (class in KD_Lib.models.nin)
P
post_epoch_call() (KD_Lib.KD.common.base_class.BaseClass method)
prune_model() (KD_Lib.Pruning.lottery_tickets.lottery_tickets.LotteryTicketsPruner method)
Q
QAT_Quantizer (class in KD_Lib.Quantization.qat.qat)
quantize() (KD_Lib.Quantization.common.base_class.Quantizer method)
(KD_Lib.Quantization.dynamic.dynamic_quantization.Dynamic_Quantizer method)
(KD_Lib.Quantization.qat.qat.QAT_Quantizer method)
(KD_Lib.Quantization.static.static_quantization.Static_Quantizer method)
Quantizer (class in KD_Lib.Quantization.common.base_class)
R
ResNet (class in KD_Lib.models.resnet)
ResNet101() (in module KD_Lib.models.resnet)
ResNet152() (in module KD_Lib.models.resnet)
ResNet18() (in module KD_Lib.models.resnet)
ResNet34() (in module KD_Lib.models.resnet)
ResNet50() (in module KD_Lib.models.resnet)
ResnetWithAT (class in KD_Lib.models.resnet)
S
set_seed() (KD_Lib.KD.text.BERT2LSTM.bert2lstm.BERT2LSTM method)
Shallow (class in KD_Lib.models.shallow)
Static_Quantizer (class in KD_Lib.Quantization.static.static_quantization)
T
train_student() (KD_Lib.KD.common.base_class.BaseClass method)
(KD_Lib.KD.text.BERT2LSTM.bert2lstm.BERT2LSTM method)
train_teacher() (KD_Lib.KD.common.base_class.BaseClass method)
(KD_Lib.KD.text.BERT2LSTM.bert2lstm.BERT2LSTM method)
Read the Docs
v: stable
Versions
latest
stable
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds