MCPcopy Create free account

hub / github.com/bearpaw/pytorch-classification / functions

Functions98 in github.com/bearpaw/pytorch-classification

↓ 20 callersMethodupdate
(self, val, n=1)
utils/misc.py:72
↓ 14 callersMethodappend
(self, numbers)
utils/logger.py:61
↓ 8 callersFunctiongauss
(x,a,b,c)
utils/visualize.py:18
↓ 8 callersFunctionmake_layers
(cfg, batch_norm=False)
models/cifar/vgg.py:53
↓ 7 callersMethodforward
(self, x)
models/cifar/resnext.py:112
↓ 5 callersFunctionmake_image
(img, mean=(0,0,0), std=(1,1,1))
utils/visualize.py:12
↓ 4 callersMethod_make_layer
Stack n bottleneck modules where n is inferred from the depth of the network. Args: block: block type used to construct ResNext
models/imagenet/resnext.py:111
↓ 4 callersFunctionaccuracy
Computes the precision@k for the specified values of k
utils/eval.py:5
↓ 4 callersMethodplot
(self, names=None)
utils/logger.py:70
↓ 3 callersMethod__init__
(self, depth=22, block=Bottleneck, dropRate=0, num_classes=10, growthRate=12, compressionRate=2)
models/cifar/densenet.py:79
↓ 3 callersMethod_make_denseblock
(self, block, blocks)
models/cifar/densenet.py:113
↓ 3 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
models/cifar/preresnet.py:128
↓ 3 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
models/cifar/resnet.py:129
↓ 3 callersMethodblock
Stack n bottleneck modules where n is inferred from the depth of the network. Args: name: string name of the current block.
models/cifar/resnext.py:92
↓ 3 callersMethodclose
(self)
utils/logger.py:79
↓ 3 callersFunctionsavefig
(fname, dpi=None)
utils/logger.py:11
↓ 2 callersMethod__init__
(self, in_planes, out_planes, stride, dropRate=0.0)
models/cifar/wrn.py:9
↓ 2 callersMethod__init__
(self, depth, num_classes=1000, block_name='BasicBlock')
models/cifar/preresnet.py:95
↓ 2 callersMethod__init__
(self, depth, num_classes=1000, block_name='BasicBlock')
models/cifar/resnet.py:95
↓ 2 callersMethod_make_transition
(self, compressionRate)
models/cifar/densenet.py:122
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
models/cifar/preresnet.py:16
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
models/cifar/resnet.py:16
↓ 2 callersFunctionmkdir_p
make dir if not exist
utils/misc.py:49
↓ 2 callersMethodset_names
(self, names)
utils/logger.py:47
↓ 2 callersFunctiontest
(val_loader, model, criterion, epoch, use_cuda)
imagenet.py:278
↓ 2 callersFunctiontest
(testloader, model, criterion, epoch, use_cuda)
cifar.py:286
↓ 1 callersMethod__init__
Constructor Args: baseWidth: baseWidth for ResNeXt. cardinality: number of convolution groups. layers: co
models/imagenet/resnext.py:75
↓ 1 callersMethod__init__
Constructor Args: cardinality: number of convolution groups. depth: number of layers. num_classes: number
models/cifar/resnext.py:58
↓ 1 callersMethod_initialize_weights
(self)
models/cifar/vgg.py:37
↓ 1 callersMethod_make_layer
(self, block, in_planes, out_planes, nb_layers, stride, dropRate)
models/cifar/wrn.py:38
↓ 1 callersFunctionadjust_learning_rate
(optimizer, epoch)
imagenet.py:336
↓ 1 callersFunctionadjust_learning_rate
(optimizer, epoch)
cifar.py:344
↓ 1 callersFunctionmain
()
imagenet.py:110
↓ 1 callersFunctionmain
()
cifar.py:101
↓ 1 callersFunctionplot_overlap
(logger, names=None)
utils/logger.py:15
↓ 1 callersMethodreset
(self)
utils/misc.py:66
↓ 1 callersFunctionsave_checkpoint
(state, is_best, checkpoint='checkpoint', filename='checkpoint.pth.tar')
imagenet.py:330
↓ 1 callersFunctionsave_checkpoint
(state, is_best, checkpoint='checkpoint', filename='checkpoint.pth.tar')
cifar.py:338
↓ 1 callersFunctiontrain
(train_loader, model, criterion, optimizer, epoch, use_cuda)
imagenet.py:223
↓ 1 callersFunctiontrain
(trainloader, model, criterion, optimizer, epoch, use_cuda)
cifar.py:231
Method__init__
(self)
utils/misc.py:63
Method__init__
(self, fpath, title=None, resume=False)
utils/logger.py:25
Method__init__
paths is a distionary with {name:filepath} pair
utils/logger.py:85
Method__init__
Constructor Args: inplanes: input channel dimensionality planes: output channel dimensionality baseWidth:
models/imagenet/resnext.py:23
Method__init__
Constructor Args: in_channels: input channel dimensionality out_channels: output channel dimensionality s
models/cifar/resnext.py:19
Method__init__
(self, inplanes, expansion=4, growthRate=12, dropRate=0)
models/cifar/densenet.py:13
Method__init__
(self, inplanes, expansion=1, growthRate=12, dropRate=0)
models/cifar/densenet.py:40
Method__init__
(self, inplanes, outplanes)
models/cifar/densenet.py:62
Method__init__
(self, nb_layers, in_planes, out_planes, block, stride, dropRate=0.0)
models/cifar/wrn.py:35
Method__init__
(self, depth, num_classes, widen_factor=1, dropRate=0.0)
models/cifar/wrn.py:47
Method__init__
(self, num_classes=10)
models/cifar/alexnet.py:13
Method__init__
(self, features, num_classes=1000)
models/cifar/vgg.py:25
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
models/cifar/preresnet.py:25
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
models/cifar/preresnet.py:57
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
models/cifar/resnet.py:25
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
models/cifar/resnet.py:57
Functionalexnet
r"""AlexNet model architecture from the `"One weird trick..." <https://arxiv.org/abs/1404.5997>`_ paper.
models/cifar/alexnet.py:39
Functioncolorize
Converts a one-channel grayscale image to a color heatmap image
utils/visualize.py:21
Functiondensenet
Constructs a ResNet model.
models/cifar/densenet.py:145
Methodforward
(self, x)
models/imagenet/resnext.py:47
Methodforward
(self, x)
models/imagenet/resnext.py:136
Methodforward
(self, x)
models/cifar/resnext.py:42
Methodforward
(self, x)
models/cifar/densenet.py:24
Methodforward
(self, x)
models/cifar/densenet.py:49
Methodforward
(self, x)
models/cifar/densenet.py:69
Methodforward
(self, x)
models/cifar/densenet.py:129
Methodforward
(self, x)
models/cifar/wrn.py:23
Methodforward
(self, x)
models/cifar/wrn.py:43
Methodforward
(self, x)
models/cifar/wrn.py:78
Methodforward
(self, x)
models/cifar/alexnet.py:32
Methodforward
(self, x)
models/cifar/vgg.py:31
Methodforward
(self, x)
models/cifar/preresnet.py:35
Methodforward
(self, x)
models/cifar/preresnet.py:70
Methodforward
(self, x)
models/cifar/preresnet.py:144
Methodforward
(self, x)
models/cifar/resnet.py:35
Methodforward
(self, x)
models/cifar/resnet.py:70
Methodforward
(self, x)
models/cifar/resnet.py:146
Functionget_mean_and_std
Compute the mean and std value of dataset.
utils/misc.py:19
Functioninit_params
Init layer parameters.
utils/misc.py:34
Methodplot
(self, names=None)
utils/logger.py:92
Functionpreresnet
Constructs a ResNet model.
models/cifar/preresnet.py:160
Functionresnet
Constructs a ResNet model.
models/cifar/resnet.py:162
Functionresnext
Constructs a ResNeXt.
models/cifar/resnext.py:122
Functionresnext101
Construct ResNeXt-101.
models/imagenet/resnext.py:160
Functionresnext152
Construct ResNeXt-152.
models/imagenet/resnext.py:168
Functionresnext50
Construct ResNeXt-50.
models/imagenet/resnext.py:152
Functionshow_batch
(images, Mean=(2, 2, 2), Std=(0.5,0.5,0.5))
utils/visualize.py:38
Functionshow_mask
(images, masklist, Mean=(2, 2, 2), Std=(0.5,0.5,0.5))
utils/visualize.py:73
Functionshow_mask_single
(images, mask, Mean=(2, 2, 2), Std=(0.5,0.5,0.5))
utils/visualize.py:44
Functionvgg11
VGG 11-layer model (configuration "A") Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
models/cifar/vgg.py:77
Functionvgg11_bn
VGG 11-layer model (configuration "A") with batch normalization
models/cifar/vgg.py:87
Functionvgg13
VGG 13-layer model (configuration "B") Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
models/cifar/vgg.py:93
Functionvgg13_bn
VGG 13-layer model (configuration "B") with batch normalization
models/cifar/vgg.py:103
Functionvgg16
VGG 16-layer model (configuration "D") Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
models/cifar/vgg.py:109
Functionvgg16_bn
VGG 16-layer model (configuration "D") with batch normalization
models/cifar/vgg.py:119
Functionvgg19
VGG 19-layer model (configuration "E") Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
models/cifar/vgg.py:125
Functionvgg19_bn
VGG 19-layer model (configuration 'E') with batch normalization
models/cifar/vgg.py:135
Functionwrn
Constructs a Wide Residual Networks.
models/cifar/wrn.py:88