MCPcopy Create free account

hub / github.com/aaron-xichen/pytorch-playground / functions

Functions136 in github.com/aaron-xichen/pytorch-playground

↓ 8 callersFunctionmake_layers
(cfg, batch_norm=False)
imagenet/vgg.py:58
↓ 7 callersMethod__init__
(self, num_classes=1000, aux_logits=True, transform_input=False)
imagenet/inception.py:29
↓ 6 callersFunctionexpand_user
(path)
utee/misc.py:94
↓ 5 callersMethodinit
(self, logdir, name='log')
utee/misc.py:15
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
imagenet/resnet.py:115
↓ 2 callersMethod__init__
(self, name, bits, sf=None, overflow_rate=0.0, counter=10)
utee/quant.py:114
↓ 2 callersMethod__init__
(self, block, layers, num_classes=1000)
imagenet/resnet.py:83
↓ 2 callersFunctioncompute_integral_part
(input, overflow_rate)
utee/quant.py:8
↓ 2 callersFunctionconv3x3
(in_planes, out_planes, stride=1)
imagenet/resnet.py:19
↓ 2 callersFunctionlinear_quantize
(input, sf, bits)
utee/quant.py:18
↓ 2 callersFunctionmake_layers
(cfg, batch_norm=False)
cifar/model.py:31
↓ 1 callersMethod__init__
(self)
utee/misc.py:12
↓ 1 callersMethod__init__
(self, inplanes, squeeze_planes, expand1x1_planes, expand3x3_planes)
imagenet/squeezenet.py:19
↓ 1 callersMethod_initialize_weights
(self)
imagenet/vgg.py:42
↓ 1 callersFunctionalexnet
(cuda=True, model_root=None)
utee/selector.py:58
↓ 1 callersFunctioncifar10
(n_channel, pretrained=None)
cifar/model.py:48
↓ 1 callersFunctionget
(batch_size, data_root='/tmp/public_dataset/pytorch', train=False, val=True, **kwargs)
imagenet/dataset.py:8
↓ 1 callersFunctionget
(batch_size, data_root='/mnt/local0/public_dataset/pytorch/', train=True, val=True, **kwargs)
stl10/dataset.py:7
↓ 1 callersFunctionlog_linear_quantize
(input, sf, bits)
utee/quant.py:42
↓ 1 callersFunctionmain
()
quantize.py:8
↓ 1 callersFunctionmake_layers
(cfg, batch_norm=False)
svhn/model.py:30
↓ 1 callersFunctionmake_layers
(cfg, batch_norm=False)
stl10/model.py:30
↓ 1 callersFunctionmin_max_quantize
(input, bits)
utee/quant.py:53
Method__init__
(self, model)
utee/misc.py:159
Method__init__
(self, name, bits, sf=None, overflow_rate=0.0, counter=10)
utee/quant.py:86
Method__init__
(self, name, bits, quant_func)
utee/quant.py:143
Method__init__
(self, input_dims, n_hiddens, n_class)
mnist/model.py:12
Method__init__
(self, num_classes=1000)
imagenet/alexnet.py:15
Method__init__
(self, version=1.0, num_classes=1000)
imagenet/squeezenet.py:52
Method__init__
(self, root, batch_size, train=False, input_size=224, **kwargs)
imagenet/dataset.py:21
Method__init__
(self, features, num_classes=1000)
imagenet/vgg.py:22
Method__init__
(self, in_channels, pool_features)
imagenet/inception.py:129
Method__init__
(self, in_channels)
imagenet/inception.py:161
Method__init__
(self, in_channels, channels_7x7)
imagenet/inception.py:184
Method__init__
(self, in_channels)
imagenet/inception.py:223
Method__init__
(self, in_channels)
imagenet/inception.py:249
Method__init__
(self, in_channels, num_classes)
imagenet/inception.py:291
Method__init__
(self, in_channels, out_channels, **kwargs)
imagenet/inception.py:323
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
imagenet/resnet.py:26
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
imagenet/resnet.py:54
Method__init__
(self, features, n_channel, num_classes)
svhn/model.py:14
Method__init__
(self, features, n_channel, num_classes)
cifar/model.py:15
Method__init__
(self, features, n_channel, num_classes)
stl10/model.py:14
Method__iter__
(self)
imagenet/dataset.py:46
Method__len__
(self)
imagenet/dataset.py:43
Method__next__
(self)
imagenet/dataset.py:49
Method__repr__
(self)
utee/quant.py:109
Method__repr__
(self)
utee/quant.py:138
Method__repr__
(self)
utee/quant.py:157
Functionalexnet
(pretrained=False, model_root=None, **kwargs)
imagenet/alexnet.py:49
Functionauto_select_gpu
(mem_bound=500, utility_bound=0, gpus=(0, 1, 2, 3, 4, 5, 6, 7), num_gpu=1, selected_gpus=None)
utee/misc.py:57
Functioncifar10
(cuda=True, model_root=None)
utee/selector.py:34
Functioncifar100
(cuda=True, model_root=None)
utee/selector.py:42
Functioncifar100
(n_channel, pretrained=None)
cifar/model.py:59
Methodcounter
(self)
utee/quant.py:96
Methodcounter
(self)
utee/quant.py:124
Methodcounter
(self)
utee/quant.py:150
Functiondump_pickle
(obj, path)
utee/misc.py:52
Functionduplicate_model_with_quant
assume that original model has at least a nn.Sequential
utee/quant.py:160
Functionensure_dir
(path, erase=False)
utee/misc.py:36
Functioneval_model
(model, ds, n_sample=None, ngpu=1, is_imagenet=False)
utee/misc.py:152
Methodforward
(self, input)
utee/misc.py:165
Methodforward
(self, input)
utee/quant.py:99
Methodforward
(self, input)
utee/quant.py:127
Methodforward
(self, input)
utee/quant.py:153
Methodforward
(self, input)
mnist/model.py:33
Methodforward
(self, x)
imagenet/alexnet.py:42
Methodforward
(self, x)
imagenet/squeezenet.py:45
Methodforward
(self, x)
imagenet/squeezenet.py:111
Methodforward
(self, x)
imagenet/vgg.py:36
Methodforward
(self, x)
imagenet/inception.py:68
Methodforward
(self, x)
imagenet/inception.py:142
Methodforward
(self, x)
imagenet/inception.py:169
Methodforward
(self, x)
imagenet/inception.py:201
Methodforward
(self, x)
imagenet/inception.py:233
Methodforward
(self, x)
imagenet/inception.py:264
Methodforward
(self, x)
imagenet/inception.py:306
Methodforward
(self, x)
imagenet/inception.py:332
Methodforward
(self, x)
imagenet/resnet.py:39
Methodforward
(self, x)
imagenet/resnet.py:70
Methodforward
(self, x)
imagenet/resnet.py:131
Methodforward
(self, x)
svhn/model.py:24
Methodforward
(self, x)
cifar/model.py:25
Methodforward
(self, x)
stl10/model.py:24
Functionget
(batch_size, data_root='/tmp/public_dataset/pytorch', train=True, val=True, **kwargs)
mnist/dataset.py:6
Functionget
(batch_size, data_root='/tmp/public_dataset/pytorch', train=True, val=True, **kwargs)
svhn/dataset.py:6
Functionget10
(batch_size, data_root='/tmp/public_dataset/pytorch', train=True, val=True, **kwargs)
cifar/dataset.py:6
Functionget100
(batch_size, data_root='/tmp/public_dataset/pytorch', train=True, val=True, **kwargs)
cifar/dataset.py:38
Functionimg2str
(img)
utee/misc.py:144
Functioninception_v3
(cuda=True, model_root=None)
utee/selector.py:98
Functioninception_v3
(pretrained=False, model_root=None, **kwargs)
imagenet/inception.py:16
Methodinfo
(self, str_info)
utee/misc.py:30
Functionload_lmdb
(lmdb_file, n_records=None)
utee/misc.py:117
Functionload_pickle
(path)
utee/misc.py:44
Functionload_state_dict
(model, model_urls, model_root)
utee/misc.py:201
Functionlog_minmax_quantize
(input, bits)
utee/quant.py:31
Functionmd5
(s)
utee/misc.py:147
Functionmnist
(cuda=True, model_root=None)
utee/selector.py:18
Functionmnist
(input_dims=784, n_hiddens=[256, 256], n_class=10, pretrained=None)
mnist/model.py:38
Functionmodel_snapshot
(model, new_file, old_file=None, verbose=False)
utee/misc.py:97
next →1–100 of 136, ranked by callers