MCPcopy Create free account

hub / github.com/SHI-Labs/Any-Precision-DNNs / functions

Functions66 in github.com/SHI-Labs/Any-Precision-DNNs

⨍ Functions 66◇ Types & classes 22
↓ 18 callersMethodupdate
(self, val, n=1)
utils.py:21
↓ 9 callersMethod__init__
(self, *kargs, **kwargs)
models/quan_ops.py:118
↓ 6 callersFunctionaccuracy
Adapted from https://github.com/pytorch/examples/blob/master/imagenet/main.py
utils.py:28
↓ 4 callersMethod__init__
(self, bit_list, quantize=True)
models/resnet_quan.py:11
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
models/resnet_quan.py:166
↓ 4 callersMethodbackward
(ctx, grad_output)
models/quan_ops.py:75
↓ 4 callersFunctionget_dataset
(name, split='train', transform=None, target_transform=None, download=True)
datasets/data.py:36
↓ 4 callersFunctionget_lr_scheduler
(name, optimizer, lr_decay, last_epoch=-1)
optimizer.py:12
↓ 4 callersFunctionget_transform
(dataset, split)
datasets/data.py:77
↓ 3 callersFunctionconv2d_quantize_fn
(bit_list)
models/quan_ops.py:122
↓ 2 callersMethod_check_integrity
(self)
datasets/cifar10.py:152
↓ 2 callersFunctionforward
(data_loader, model, criterion, criterion_soft, epoch, training=True, optimizer=None, sum_writer=None)
train.py:152
↓ 2 callersFunctionforward
(data_loader, model, criterion, criterion_soft, epoch, training=True, optimizer=None, sum_writer=None)
train_imagenet.py:152
↓ 2 callersFunctionget_optimizer_config
(model, name, lr, weight_decay)
optimizer.py:4
↓ 2 callersFunctionsave_checkpoint
(state, is_best, path, name='model_latest.pth.tar')
utils.py:73
↓ 2 callersFunctionsetup_logging
Setup logging configuration
utils.py:46
↓ 1 callersMethod__init__
(self, bit_list, num_classes, expand=8)
models/svhn_quan.py:35
↓ 1 callersMethod_check_integrity
(self)
datasets/svhn.py:173
↓ 1 callersMethoddownload
(self)
datasets/svhn.py:186
↓ 1 callersMethoddownload
(self)
datasets/cifar10.py:161
↓ 1 callersFunctionmain
()
train.py:43
↓ 1 callersFunctionmain
()
train_imagenet.py:43
↓ 1 callersMethodreset
(self)
utils.py:15
↓ 1 callersFunctionsetup_gpus
Adapted from https://github.com/bamos/setGPU/blob/master/setGPU.py
utils.py:61
Method__getitem__
Args: index (int): Index Returns: tuple: (image, target) where target is index of the target class.
datasets/svhn.py:148
Method__getitem__
Args: index (int): Index Returns: tuple: (image, target) where target is index of the target class.
datasets/cifar10.py:121
Method__init__
(self)
utils.py:12
Method__init__
(self, root, split='train', transform=None, target_transform=None, download=False)
datasets/svhn.py:75
Method__init__
(self, root, split, transform=None, target_transform=None, download=False)
datasets/cifar10.py:53
Method__init__
(self, bit_list, quantize=True)
models/svhn_quan.py:15
Method__init__
(self, bit_list, in_planes, out_planes, stride=1)
models/resnet_quan.py:33
Method__init__
(self, block, num_units, bit_list, num_classes, expand=5)
models/resnet_quan.py:73
Method__init__
(self, bit_list, in_planes, out_planes, stride=1, downsample=None)
models/resnet_quan.py:109
Method__init__
(self, block, layers, bit_list, num_classes)
models/resnet_quan.py:139
Method__init__
(self, num_features, bit_list)
models/quan_ops.py:14
Method__init__
(self, num_features, bit_list=bit_list)
models/quan_ops.py:33
Method__init__
(self, num_features, bit_list)
models/quan_ops.py:42
Method__init__
(self, num_features, bit_list=bit_list)
models/quan_ops.py:61
Method__init__
(self, bit_list)
models/quan_ops.py:81
Method__init__
(self, bit_list)
models/quan_ops.py:103
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1,
models/quan_ops.py:124
Method__init__
(self, *kargs, **kwargs)
models/quan_ops.py:140
Method__init__
(self, in_features, out_features, bias=True)
models/quan_ops.py:146
Method__len__
(self)
datasets/svhn.py:170
Method__len__
(self)
datasets/cifar10.py:146
Functionbatchnorm1d_fn
(bit_list)
models/quan_ops.py:59
Functionbatchnorm2d_fn
(bit_list)
models/quan_ops.py:31
Methodforward
(self, x)
models/svhn_quan.py:24
Methodforward
(self, x)
models/svhn_quan.py:75
Methodforward
(self, output, target)
models/losses.py:8
Methodforward
(self, x)
models/resnet_quan.py:20
Methodforward
(self, x)
models/resnet_quan.py:54
Methodforward
(self, x)
models/resnet_quan.py:96
Methodforward
(self, x)
models/resnet_quan.py:129
Methodforward
(self, x)
models/resnet_quan.py:181
Methodforward
(self, x)
models/quan_ops.py:26
Methodforward
(self, x)
models/quan_ops.py:54
Methodforward
(ctx, input, k)
models/quan_ops.py:69
Methodforward
(self, x)
models/quan_ops.py:87
Methodforward
(self, x)
models/quan_ops.py:109
Methodforward
(self, input, order=None)
models/quan_ops.py:132
Methodforward
(self, input)
models/quan_ops.py:152
Functionlinear_quantize_fn
(bit_list)
models/quan_ops.py:144
Functionresnet20q
(bit_list, num_classes=10)
models/resnet_quan.py:199
Functionresnet50q
(bit_list, num_classes=1000)
models/resnet_quan.py:204
Functionsvhnq
(bit_list, num_classes=10, expand=2)
models/svhn_quan.py:82