Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/BayesWatch/pytorch-prunes
/ functions
Functions
87 in github.com/BayesWatch/pytorch-prunes
⨍
Functions
87
◇
Types & classes
20
↓ 20 callers
Method
update
(self, val, n=1)
funcs.py:192
↓ 9 callers
Method
__init__
(self)
models/wideresnet.py:16
↓ 7 callers
Method
__init__
(self)
models/densenet.py:28
↓ 7 callers
Function
get_layer_param
(model)
funcs.py:250
↓ 5 callers
Method
concat
(input)
funcs.py:147
↓ 4 callers
Function
get_error
Computes the error@k for the specified values of k
funcs.py:164
↓ 4 callers
Method
prune
(self, model, feat_index)
funcs.py:79
↓ 3 callers
Method
_make_dense
(self, nChannels, growthRate, nDenseBlocks, bottleneck, mask=False, width=1)
models/densenet.py:245
↓ 2 callers
Method
_get_fisher
(self, model)
funcs.py:107
↓ 2 callers
Method
cost
(self)
models/densenet.py:108
↓ 2 callers
Function
is_leaf
(model)
funcs.py:240
↓ 2 callers
Function
save_checkpoint
(state, filename='checkpoint.pth.tar')
funcs.py:160
↓ 2 callers
Function
validate
()
train.py:187
↓ 1 callers
Method
_create_mask
This takes an activation to generate the exact mask required. It also records input and output shapes for posterity.
models/densenet.py:86
↓ 1 callers
Method
_create_mask
(self, x, out)
models/wideresnet.py:153
↓ 1 callers
Method
_get_masks
(self, model)
funcs.py:124
↓ 1 callers
Method
_make_layer
(self, block, in_channels, out_channels, nb_layers, stride, dropRate)
models/wideresnet.py:231
↓ 1 callers
Method
_make_layer
(self, block, in_channels, out_channels, mid_channels, nb_layers, stride, dropRate)
models/wideresnet.py:246
↓ 1 callers
Method
_update_cost
(self, model)
funcs.py:133
↓ 1 callers
Method
compress
(self, model)
funcs.py:102
↓ 1 callers
Method
compress_weights
(self)
models/densenet.py:123
↓ 1 callers
Function
finetune
()
prune.py:114
↓ 1 callers
Method
fisher_prune
(self, model, prune_every)
funcs.py:24
↓ 1 callers
Method
fixed_prune
(self, model, ID)
funcs.py:33
↓ 1 callers
Method
forward
(self, x)
models/densenet.py:31
↓ 1 callers
Method
get_cost
(self, model)
funcs.py:138
↓ 1 callers
Function
get_inf_params
(net, verbose=True, sd=False)
funcs.py:199
↓ 1 callers
Function
get_layer_info
(layer)
funcs.py:244
↓ 1 callers
Function
get_num_gen
(gen)
funcs.py:228
↓ 1 callers
Function
is_pruned
(layer)
funcs.py:232
↓ 1 callers
Method
l1_prune
(self, model, prune_every)
funcs.py:52
↓ 1 callers
Function
measure_layer
(layer, x)
funcs.py:255
↓ 1 callers
Function
modify_forward
(model)
funcs.py:334
↓ 1 callers
Function
new_forward
(m)
funcs.py:337
↓ 1 callers
Function
prune
()
prune.py:175
↓ 1 callers
Method
random_prune
(self, model)
funcs.py:37
↓ 1 callers
Method
reset
(self)
funcs.py:186
↓ 1 callers
Method
reset_fisher
(self)
models/densenet.py:101
↓ 1 callers
Function
restore_forward
(model)
funcs.py:349
↓ 1 callers
Function
should_measure
(x)
funcs.py:331
↓ 1 callers
Function
train
()
train.py:134
↓ 1 callers
Function
validate
()
prune.py:189
Method
__init__
(self, module_name='MaskBlock')
funcs.py:18
Method
__init__
(self)
funcs.py:183
Method
__init__
(self)
models/densenet.py:20
Method
__init__
(self, channels, spatial)
models/densenet.py:36
Method
__init__
(self, nChannels, growthRate)
models/densenet.py:47
Method
__init__
(self, nChannels, growthRate, width=1)
models/densenet.py:156
Method
__init__
(self, nChannels, growthRate)
models/densenet.py:174
Method
__init__
(self, nChannels, nOutChannels)
models/densenet.py:187
Method
__init__
(self, growthRate, depth, reduction, nClasses, bottleneck, mask=False, width=1.)
models/densenet.py:200
Method
__init__
(self)
models/wideresnet.py:8
Method
__init__
(self, channels, spatial)
models/wideresnet.py:24
Method
__init__
(self, in_channels, out_channels, stride, dropRate=0.0)
models/wideresnet.py:35
Method
__init__
(self, in_channels, out_channels, mid_channels, stride, dropRate=0.0)
models/wideresnet.py:64
Method
__init__
(self, in_channels, out_channels, stride=1, dropRate=0.0)
models/wideresnet.py:95
Method
__init__
(self, nb_layers, in_channels, out_channels, block, stride, dropRate=0.0)
models/wideresnet.py:227
Method
__init__
(self, nb_layers, in_channels, out_channels, mid_channels, block, stride, dropRate=0.0)
models/wideresnet.py:242
Method
__init__
(self, depth, widen_factor, num_classes=10, dropRate=0.0, mask=False)
models/wideresnet.py:259
Method
__init__
(self, depth, widen_factor, num_classes=10, dropRate=0.0, bottle_mult=0.5)
models/wideresnet.py:314
Method
_fisher
(self, _, __, grad_output)
models/densenet.py:93
Method
_fisher
(self, notused1, notused2, grad_output)
models/wideresnet.py:159
Method
compress_weights
(self)
models/wideresnet.py:192
Function
concat
(input)
funcs.py:156
Method
cost
(self)
models/wideresnet.py:170
Function
find
(input)
funcs.py:151
Method
forward
(self, x)
models/densenet.py:23
Method
forward
(self, x)
models/densenet.py:41
Method
forward
(self, x)
models/densenet.py:72
Method
forward
(self, x)
models/densenet.py:166
Method
forward
(self, x)
models/densenet.py:180
Method
forward
(self, x)
models/densenet.py:193
Method
forward
(self, x)
models/densenet.py:257
Method
forward
(self, x)
models/wideresnet.py:11
Method
forward
(self, x)
models/wideresnet.py:19
Method
forward
(self, x)
models/wideresnet.py:29
Method
forward
(self, x)
models/wideresnet.py:50
Method
forward
(self, x)
models/wideresnet.py:79
Method
forward
(self, x)
models/wideresnet.py:126
Method
forward
(self, x)
models/wideresnet.py:237
Method
forward
(self, x)
models/wideresnet.py:254
Method
forward
(self, x)
models/wideresnet.py:302
Method
forward
(self, x)
models/wideresnet.py:352
Function
lambda_forward
(x)
funcs.py:338
Function
measure_model
(model, H, W)
funcs.py:325
Method
reset_fisher
(self)
models/wideresnet.py:167
Method
update
(self, previous_mask)
models/densenet.py:104