MCPcopy Create free account

hub / github.com/BayesWatch/pytorch-blockswap / functions

Functions120 in github.com/BayesWatch/pytorch-blockswap

⨍ Functions 120◇ Types & classes 50
↓ 44 callersMethod__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:11
↓ 19 callersMethodupdate
(self, val, n=1)
utils.py:195
↓ 4 callersMethodconcat
(input)
utils.py:247
↓ 3 callersFunctionaccuracy
Computes the precision@k for the specified values of k
utils.py:55
↓ 3 callersFunctionget_no_params
(net, verbose=False)
utils.py:110
↓ 3 callersFunctionvalidate
(net, checkpoint=None)
train.py:210
↓ 2 callersFunctionat
(x)
utils.py:47
↓ 2 callersFunctionblock_function
(blocktype)
models/blocks.py:722
↓ 2 callersFunctionconv_function
(convtype)
models/blocks.py:732
↓ 1 callersMethod_create_mask
This takes an activation to generate the exact mask required. It also records input and output shapes for posterity.
models/blocks.py:997
↓ 1 callersMethod_create_mask
This takes an activation to generate the exact mask required. It also records input and output shapes for posterity.
models/blocks.py:1168
↓ 1 callersMethod_get_fisher
(self, model)
utils.py:213
↓ 1 callersMethod_get_masks
(self, model)
utils.py:232
↓ 1 callersMethod_make_layer
( self, blocks, in_planes, out_planes, nb_layers, stride, dropRate, convs )
models/blocks.py:1438
↓ 1 callersMethod_update_flops
(self, model)
utils.py:241
↓ 1 callersMethodcost
(self)
models/blocks.py:1029
↓ 1 callersFunctionexpand_model
(model, layers=[])
utils.py:71
↓ 1 callersFunctiongenerate_archs
()
model_generator.py:31
↓ 1 callersFunctionget_cifar_loaders
( cifar_loc, batch_size=128, workers=0, cutout=True, n_holes=1, length=16, pin_mem
utils.py:133
↓ 1 callersMethodget_conv_sizes
(self)
models/blocks.py:1018
↓ 1 callersMethodget_conv_sizes
(self)
models/blocks.py:1189
↓ 1 callersFunctionget_param_tot
(convs)
model_generator.py:59
↓ 1 callersMethodgo_fish
(self, model)
utils.py:208
↓ 1 callersFunctionload_network
(loc)
train.py:336
↓ 1 callersFunctionone_shot_fisher
(net, trainloader, n_steps=1)
model_generator.py:79
↓ 1 callersFunctionrank_arch
(candidate)
model_generator.py:118
↓ 1 callersMethodreset
(self)
utils.py:189
↓ 1 callersMethodreset_fisher
(self)
models/blocks.py:1012
↓ 1 callersFunctiontest
()
models/wide_resnet.py:174
↓ 1 callersFunctiontrain_student
(net, teach)
train.py:142
↓ 1 callersFunctiontrain_teacher
(net)
train.py:83
Method__call__
Args: img (Tensor): Tensor image of size (C, H, W). Returns: Tensor: Image with n_holes of dimension length x
utils.py:262
Method__del__
(self)
models/blocks.py:881
Method__del__
(self)
models/blocks.py:970
Method__del__
(self)
models/blocks.py:1251
Method__init__
(self)
utils.py:186
Method__init__
(self, module_name="MaskBlock")
utils.py:203
Method__init__
(self, n_holes, length)
utils.py:258
Method__init__
( self, depth, widen_factor, conv=Conv, block=BasicBlock, num_
models/wide_resnet.py:31
Method__init__
( self, in_planes, out_planes, bottleneck, stride=1, kernel_si
models/blocks.py:29
Method__init__
( self, in_planes, out_planes, bottleneck, group_split, stride
models/blocks.py:65
Method__init__
( self, in_planes, out_planes, bottleneck, groups, stride=1,
models/blocks.py:103
Method__init__
( self, in_planes, out_planes, bottleneck, stride=1, kernel_si
models/blocks.py:141
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:178
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:194
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:210
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:226
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:242
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:258
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:274
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:290
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:306
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:322
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:338
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:354
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:370
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:385
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:400
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:415
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=2, padding=1, bias=False )
models/blocks.py:430
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, paddin
models/blocks.py:450
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:480
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:495
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:510
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:525
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:540
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:555
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:570
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:585
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:600
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:615
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:630
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:645
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, padding=1, bias=False )
models/blocks.py:660
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, paddin
models/blocks.py:688
Method__init__
( self, in_planes, out_planes, stride=1, kernel_size=3, paddin
models/blocks.py:809
Method__init__
(self, in_planes, out_planes, stride, dropRate=0.0, conv=Conv)
models/blocks.py:850
Method__init__
(self)
models/blocks.py:901
Method__init__
( self, in_channels, out_channels, stride=1, dropRate=0.0, conv=Conv, bottle=False )
models/blocks.py:912
Method__init__
(self, in_planes, out_planes, stride, dropRate=0.0, conv=Conv, xy=None)
models/blocks.py:1107
Method__init__
(self, in_planes, out_planes, stride, dropRate=0.0, conv=Conv, xy=None)
models/blocks.py:1225
Method__init__
(self, in_planes, out_planes, stride, dropRate=0.0, conv=Conv)
models/blocks.py:1271
Method__init__
( self, nb_layers, in_planes, out_planes, stride, dropRate=0.0
models/blocks.py:1404
Method__init__
( self, conv, block, layers, num_classes=1000, convs=[],
models/resnet.py:43
Method_fisher
(self, blargh, blergh, grad_output)
models/blocks.py:1004
Method_fisher
(self, blargh, blergh, grad_output)
models/blocks.py:1175
Method_get_convs
(self)
models/wide_resnet.py:144
Functionat_loss
(x, y)
utils.py:51
Methodcompress_weights
(self)
models/blocks.py:1058
Methodcost
(self)
models/blocks.py:1200
Functiondistillation
(y, teacher_scores, labels, T, alpha)
utils.py:41
Methodforward
(self, x)
models/wide_resnet.py:150
Methodforward
(self, x)
models/blocks.py:24
Methodforward
(self, x)
models/blocks.py:57
Methodforward
(self, x)
models/blocks.py:95
Methodforward
(self, x)
models/blocks.py:133
Methodforward
(self, x)
models/blocks.py:170
Methodforward
(self, x)
models/blocks.py:445
Methodforward
(self, x)
models/blocks.py:475
Methodforward
(self, x)
models/blocks.py:679
next →1–100 of 120, ranked by callers