MCPcopy Create free account

hub / github.com/HobbitLong/RepDistiller / functions

Functions213 in github.com/HobbitLong/RepDistiller

⨍ Functions 213◇ Types & classes 58
↓ 20 callersMethodcuda
(self)
crd/memory.py:123
↓ 17 callersMethodupdate
(self, val, n=1)
helper/util.py:36
↓ 11 callersMethod__init__
(self, dim_in=1024, dim_out=128)
models/util.py:173
↓ 5 callersMethod_make_layers
(cfg, batch_norm=False, in_channels=3)
models/vgg.py:103
↓ 5 callersMethodpoly_kernel
(self, a, b)
distiller_zoo/NST.py:38
↓ 4 callersMethod__init__
(self, in_channels, out_channels)
models/ShuffleNetv2.py:65
↓ 4 callersMethod_make_layer
(self, block, planes, num_blocks, stride)
models/resnetv2.py:132
↓ 4 callersMethodfactor
(self, f)
distiller_zoo/FT.py:30
↓ 4 callersMethodreset
(self)
helper/util.py:30
↓ 3 callersMethod__init__
(self, dim_in=1024, dim_out=128)
crd/criterion.py:81
↓ 3 callersMethod_make_layer
(self, out_channels, num_blocks)
models/ShuffleNetv2.py:121
↓ 3 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
models/resnet.py:137
↓ 3 callersMethod_make_layer
(self, out_planes, num_blocks, groups)
models/ShuffleNetv1.py:71
↓ 3 callersFunctionaccuracy
Computes the accuracy over the k top predictions for the specified values of k
helper/util.py:43
↓ 3 callersFunctionget_data_folder
return server-dependent path to store the data
dataset/imagenet.py:14
↓ 3 callersFunctioninit
(model_s, model_t, init_modules, criterion, train_loader, logger, opt)
helper/pretrain.py:11
↓ 3 callersMethodremovenan
(x)
distiller_zoo/KDSVD.py:64
↓ 3 callersFunctionvalidate
validation
helper/loops.py:218
↓ 2 callersMethod__init__
(self, block, num_blocks, num_classes=10, zero_init_residual=False)
models/resnetv2.py:76
↓ 2 callersMethod__init__
(self, in_planes, out_planes, stride, dropRate=0.0)
models/wrn.py:14
↓ 2 callersMethod__init__
(self, depth, num_filters, block_name='BasicBlock', num_classes=10)
models/resnet.py:105
↓ 2 callersMethod__init__
(self, in_planes, out_planes, stride, groups, is_last=False)
models/ShuffleNetv1.py:22
↓ 2 callersFunctionadjust_learning_rate
Sets the learning rate to the initial LR decayed by decay rate every steep step
helper/util.py:16
↓ 2 callersMethodat
(self, f)
distiller_zoo/AT.py:28
↓ 2 callersMethodcompute_fsp
(g)
distiller_zoo/FSP.py:30
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
models/resnet.py:18
↓ 2 callersFunctionget_cifar100_dataloaders
cifar 100
dataset/cifar100.py:47
↓ 2 callersFunctionget_data_folder
return server-dependent path to store the data
dataset/cifar100.py:21
↓ 2 callersMethodget_feat_modules
(self)
models/vgg.py:43
↓ 2 callersFunctionget_teacher_name
parse teacher name
train_student.py:125
↓ 2 callersMethodpdist
(e, squared=False, eps=1e-12)
distiller_zoo/RKD.py:48
↓ 2 callersMethodsvd
(self, feat, n=1)
distiller_zoo/KDSVD.py:45
↓ 1 callersFunctionResNet18
(**kwargs)
models/resnetv2.py:165
↓ 1 callersFunctionShuffleV1
(**kwargs)
models/ShuffleNetv1.py:118
↓ 1 callersFunctionShuffleV2
(**kwargs)
models/ShuffleNetv2.py:195
↓ 1 callersMethod__getitem__
Args: index (int): Index Returns: tuple: (image, target) where target is class_index of the target class.
dataset/imagenet.py:82
↓ 1 callersMethod__init__
(self, probs)
crd/memory.py:86
↓ 1 callersMethod__init__
(self, dim_in, n_label=10)
models/classifier.py:12
↓ 1 callersMethod__init__
(self, T, feature_dim, input_size=32, width_mult=1.,
models/mobilenetv2.py:66
↓ 1 callersMethod_build_feature_connector
(t_channel, s_channel)
models/util.py:111
↓ 1 callersMethod_initialize_weights
(self)
models/vgg.py:118
↓ 1 callersMethod_initialize_weights
(self)
models/mobilenetv2.py:162
↓ 1 callersMethod_make_conenctors
(s_shapes, t_shapes)
models/util.py:75
↓ 1 callersMethod_make_conenctors
(self, s_shapes, t_shapes)
models/util.py:102
↓ 1 callersMethod_make_layer
(self, block, in_planes, out_planes, nb_layers, stride, dropRate)
models/wrn.py:46
↓ 1 callersMethodalign_rsv
(a, b)
distiller_zoo/KDSVD.py:69
↓ 1 callersMethodat_loss
(self, f_s, f_t)
distiller_zoo/AT.py:18
↓ 1 callersMethodcompute_loss
(s, t)
distiller_zoo/FSP.py:26
↓ 1 callersFunctionconv_1x1_bn
(inp, oup)
models/mobilenetv2.py:23
↓ 1 callersFunctionconv_bn
(inp, oup, stride)
models/mobilenetv2.py:15
↓ 1 callersMethodcosine_similarity_loss
(output_net, target_net, eps=0.0000001)
distiller_zoo/PKT.py:17
↓ 1 callersMethodcriterion_alternative_l2
(self, source, target)
distiller_zoo/AB.py:26
↓ 1 callersMethoddraw
Draw N samples from multinomial
crd/memory.py:127
↓ 1 callersMethodfactor_loss
(self, f_s, f_t)
distiller_zoo/FT.py:17
↓ 1 callersMethodget_bn_before_relu
(self)
models/resnetv2.py:116
↓ 1 callersMethodget_bn_before_relu
(self)
models/wrn.py:96
↓ 1 callersMethodget_bn_before_relu
(self)
models/vgg.py:57
↓ 1 callersMethodget_bn_before_relu
(self)
models/mobilenetv2.py:119
↓ 1 callersMethodget_bn_before_relu
(self)
models/resnet.py:164
↓ 1 callersFunctionget_cifar100_dataloaders_sample
cifar 100
dataset/cifar100.py:172
↓ 1 callersFunctionload_teacher
(model_path, n_cls)
train_student.py:134
↓ 1 callersFunctionmain
()
train_student.py:143
↓ 1 callersFunctionmain
()
train_teacher.py:85
↓ 1 callersFunctionmobile_half
(num_classes)
models/mobilenetv2.py:183
↓ 1 callersFunctionmobilenetv2_T_w
(T, W, feature_dim=100)
models/mobilenetv2.py:178
↓ 1 callersMethodnst_loss
(self, f_s, f_t)
distiller_zoo/NST.py:16
↓ 1 callersFunctionparse_option
()
train_student.py:35
↓ 1 callersFunctionparse_option
()
train_teacher.py:22
↓ 1 callersFunctionresnet8x4
(**kwargs)
models/resnet.py:233
↓ 1 callersMethodsimilarity_loss
(self, f_s, f_t)
distiller_zoo/SP.py:16
↓ 1 callersFunctionvgg19_bn
VGG 19-layer model (configuration 'E') with batch normalization
models/vgg.py:215
↓ 1 callersFunctionwrn_40_2
(**kwargs)
models/wrn.py:135
FunctionResNet101
(**kwargs)
models/resnetv2.py:177
FunctionResNet152
(**kwargs)
models/resnetv2.py:181
FunctionResNet34
(**kwargs)
models/resnetv2.py:169
FunctionResNet50
(**kwargs)
models/resnetv2.py:173
Method__getitem__
(self, index)
dataset/cifar100.py:42
Method__getitem__
(self, index)
dataset/cifar100.py:138
Method__getitem__
Args: index (int): Index Returns: tuple: (image, target) where target is class_index of the target class.
dataset/imagenet.py:35
Method__init__
(self, root, train=True, transform=None, target_transform=None, download=Fal
dataset/cifar100.py:99
Method__init__
(self, root, transform=None, target_transform=None, is_sample=False, k=4096)
dataset/imagenet.py:49
Method__init__
(self, w_d=25, w_a=50)
distiller_zoo/RKD.py:10
Method__init__
(self, s_shapes, t_shapes)
distiller_zoo/FSP.py:11
Method__init__
(self, k=1)
distiller_zoo/KDSVD.py:13
Method__init__
(self, p1=2, p2=1)
distiller_zoo/FT.py:9
Method__init__
(self)
distiller_zoo/PKT.py:10
Method__init__
(self)
distiller_zoo/NST.py:9
Method__init__
(self)
distiller_zoo/CC.py:11
Method__init__
(self, p=2)
distiller_zoo/AT.py:11
Method__init__
(self, feat_num, margin=1.0)
distiller_zoo/AB.py:11
Method__init__
(self, num_input_channels, num_mid_channel, num_target_chan
distiller_zoo/VID.py:12
Method__init__
(self)
distiller_zoo/FitNet.py:8
Method__init__
(self, T)
distiller_zoo/KD.py:9
Method__init__
(self)
distiller_zoo/SP.py:10
Method__init__
(self, inputSize, outputSize, K, T=0.07, momentum=0.5)
crd/memory.py:10
Method__init__
(self, opt)
crd/criterion.py:23
Method__init__
(self, n_data)
crd/criterion.py:55
Method__init__
(self, power=2)
crd/criterion.py:95
Method__init__
(self)
helper/util.py:27
Method__init__
(self, in_planes, planes, stride=1, is_last=False)
models/resnetv2.py:15
next →1–100 of 213, ranked by callers