MCPcopy Create free account

hub / github.com/HobbitLong/RepDistiller / types & classes

Types & classes58 in github.com/HobbitLong/RepDistiller

↓ 17 callersClassAverageMeter
Computes and stores the average and current value
helper/util.py:25
↓ 10 callersClassVGG
models/vgg.py:23
↓ 9 callersClassResNet
models/resnet.py:103
↓ 5 callersClassResNet
models/resnetv2.py:75
↓ 5 callersClassWideResNet
models/wrn.py:56
↓ 3 callersClassNetworkBlock
models/wrn.py:41
↓ 3 callersClassNormalize
normalization layer
models/util.py:213
↓ 2 callersClassContrastLoss
contrastive loss, corresponding to Eq (18)
crd/criterion.py:51
↓ 2 callersClassConvReg
Convolutional regression for FitNet
models/util.py:131
↓ 2 callersClassDistillKL
Distilling the Knowledge in a Neural Network
distiller_zoo/KD.py:7
↓ 2 callersClassEmbed
Embedding module
crd/criterion.py:79
↓ 2 callersClassLinearEmbed
Linear Embedding
models/util.py:185
↓ 2 callersClassShuffleBlock
models/ShuffleNetv2.py:9
↓ 1 callersClassABLoss
Knowledge Transfer via Distillation of Activation Boundaries Formed by Hidden Neurons code: https://github.com/bhheo/AB_distillation
distiller_zoo/AB.py:7
↓ 1 callersClassAliasMethod
From: https://hips.seas.harvard.edu/blog/2013/03/03/the-alias-method-efficient-sampling-with-many-discrete-outcomes/
crd/memory.py:82
↓ 1 callersClassAttention
Paying More Attention to Attention: Improving the Performance of Convolutional Neural Networks via Attention Transfer code: https://github.com
distiller_zoo/AT.py:7
↓ 1 callersClassBasicBlock
models/ShuffleNetv2.py:31
↓ 1 callersClassBottleneck
models/ShuffleNetv1.py:21
↓ 1 callersClassCIFAR100Instance
CIFAR100Instance Dataset.
dataset/cifar100.py:39
↓ 1 callersClassCIFAR100InstanceSample
CIFAR100Instance+Sample Dataset
dataset/cifar100.py:95
↓ 1 callersClassCRDLoss
CRD Loss function includes two symmetric parts: (a) using teacher as anchor, choose positive and negatives over the student side (b) using
crd/criterion.py:8
↓ 1 callersClassConnector
Connect for Knowledge Transfer via Distillation of Activation Boundaries Formed by Hidden Neurons
models/util.py:65
↓ 1 callersClassConnectorV2
A Comprehensive Overhaul of Feature Distillation (ICCV 2019)
models/util.py:93
↓ 1 callersClassContrastMemory
memory buffer that supplies large amount of negative samples.
crd/memory.py:6
↓ 1 callersClassCorrelation
Correlation Congruence for Knowledge Distillation, ICCV 2019. The authors nicely shared the code with me. I restructured their code to be com
distiller_zoo/CC.py:7
↓ 1 callersClassDownBlock
models/ShuffleNetv2.py:64
↓ 1 callersClassFSP
A Gift from Knowledge Distillation: Fast Optimization, Network Minimization and Transfer Learning
distiller_zoo/FSP.py:8
↓ 1 callersClassFactorTransfer
Paraphrasing Complex Network: Network Compression via Factor Transfer, NeurIPS 2018
distiller_zoo/FT.py:7
↓ 1 callersClassFlatten
flatten module
models/util.py:225
↓ 1 callersClassHintLoss
Fitnets: hints for thin deep nets, ICLR 2015
distiller_zoo/FitNet.py:6
↓ 1 callersClassImageFolderInstance
: Folder datasets which returns the index of the image as well::
dataset/imagenet.py:32
↓ 1 callersClassImageFolderSample
: Folder datasets which returns (img, label, index, contrast_index):
dataset/imagenet.py:46
↓ 1 callersClassInvertedResidual
models/mobilenetv2.py:31
↓ 1 callersClassKDSVD
Self-supervised Knowledge Distillation using Singular Value Decomposition original Tensorflow code: https://github.com/sseung0703/SSKD_SVD
distiller_zoo/KDSVD.py:8
↓ 1 callersClassMobileNetV2
mobilenetV2
models/mobilenetv2.py:64
↓ 1 callersClassNSTLoss
like what you like: knowledge distill via neuron selectivity transfer
distiller_zoo/NST.py:7
↓ 1 callersClassNormalize
normalization layer
crd/criterion.py:93
↓ 1 callersClassPKT
Probabilistic Knowledge Transfer for deep representation learning Code from author: https://github.com/passalis/probabilistic_kt
distiller_zoo/PKT.py:7
↓ 1 callersClassParaphraser
Paraphrasing Complex Network: Network Compression via Factor Transfer
models/util.py:7
↓ 1 callersClassRKDLoss
Relational Knowledge Disitllation, CVPR2019
distiller_zoo/RKD.py:8
↓ 1 callersClassShuffleBlock
models/ShuffleNetv1.py:9
↓ 1 callersClassShuffleNet
models/ShuffleNetv1.py:56
↓ 1 callersClassShuffleNetV2
models/ShuffleNetv2.py:102
↓ 1 callersClassSimilarity
Similarity-Preserving Knowledge Distillation, ICCV2019, verified by original author
distiller_zoo/SP.py:8
↓ 1 callersClassSplitBlock
models/ShuffleNetv2.py:21
↓ 1 callersClassTranslator
models/util.py:44
↓ 1 callersClassVIDLoss
Variational Information Distillation for Knowledge Transfer (CVPR 2019), code from author: https://github.com/ssahn0215/variational-information-di
distiller_zoo/VID.py:9
ClassBasicBlock
models/resnetv2.py:12
ClassBasicBlock
models/wrn.py:13
ClassBasicBlock
models/resnet.py:24
ClassBottleneck
models/resnetv2.py:42
ClassBottleneck
models/resnet.py:60
ClassEmbed
Embedding module
models/util.py:171
ClassLinearClassifier
models/classifier.py:10
ClassMLPEmbed
non-linear embed by MLP
models/util.py:197
ClassNonLinearClassifier
models/classifier.py:21
ClassPoolEmbed
pool and embed
models/util.py:234
ClassRegress
Simple Linear Regression for hints
models/util.py:157