Types & classes — github.com/Eric-mingjie/network-slimming

CodeHub MCPcopy Create free account

hub / github.com/Eric-mingjie/network-slimming / types & classes

Types & classes13 in github.com/Eric-mingjie/network-slimming

⨍ Functions 49 ◇ Types & classes 13

↓ 5 callersClasschannel_selection

Select channels from the output of BatchNorm2d layer. It should be put directly after BatchNorm2d layer. The output shape of this layer is de

models/channel_selection.py:6

↓ 3 callersClassvgg

models/vgg.py:16

↓ 2 callersClassdensenet

models/densenet.py:55

↓ 2 callersClassresnet

models/preresnet.py:53

↓ 1 callersClassTransition

mask-impl/models/densenet.py:35

↓ 1 callersClassTransition

models/densenet.py:38

↓ 1 callersClassvgg

mask-impl/models/vgg.py:16

ClassBasicBlock

mask-impl/models/densenet.py:14

ClassBasicBlock

models/densenet.py:15

ClassBottleneck

mask-impl/models/preresnet.py:12

ClassBottleneck

models/preresnet.py:13

mask-impl/models/densenet.py:50

mask-impl/models/preresnet.py:50