Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/aaron-xichen/pytorch-playground
/ functions
Functions
136 in github.com/aaron-xichen/pytorch-playground
⨍
Functions
136
◇
Types & classes
25
↓ 8 callers
Function
make_layers
(cfg, batch_norm=False)
imagenet/vgg.py:58
↓ 7 callers
Method
__init__
(self, num_classes=1000, aux_logits=True, transform_input=False)
imagenet/inception.py:29
↓ 6 callers
Function
expand_user
(path)
utee/misc.py:94
↓ 5 callers
Method
init
(self, logdir, name='log')
utee/misc.py:15
↓ 4 callers
Method
_make_layer
(self, block, planes, blocks, stride=1)
imagenet/resnet.py:115
↓ 2 callers
Method
__init__
(self, name, bits, sf=None, overflow_rate=0.0, counter=10)
utee/quant.py:114
↓ 2 callers
Method
__init__
(self, block, layers, num_classes=1000)
imagenet/resnet.py:83
↓ 2 callers
Function
compute_integral_part
(input, overflow_rate)
utee/quant.py:8
↓ 2 callers
Function
conv3x3
(in_planes, out_planes, stride=1)
imagenet/resnet.py:19
↓ 2 callers
Function
linear_quantize
(input, sf, bits)
utee/quant.py:18
↓ 2 callers
Function
make_layers
(cfg, batch_norm=False)
cifar/model.py:31
↓ 1 callers
Method
__init__
(self)
utee/misc.py:12
↓ 1 callers
Method
__init__
(self, inplanes, squeeze_planes, expand1x1_planes, expand3x3_planes)
imagenet/squeezenet.py:19
↓ 1 callers
Method
_initialize_weights
(self)
imagenet/vgg.py:42
↓ 1 callers
Function
alexnet
(cuda=True, model_root=None)
utee/selector.py:58
↓ 1 callers
Function
cifar10
(n_channel, pretrained=None)
cifar/model.py:48
↓ 1 callers
Function
get
(batch_size, data_root='/tmp/public_dataset/pytorch', train=False, val=True, **kwargs)
imagenet/dataset.py:8
↓ 1 callers
Function
get
(batch_size, data_root='/mnt/local0/public_dataset/pytorch/', train=True, val=True, **kwargs)
stl10/dataset.py:7
↓ 1 callers
Function
log_linear_quantize
(input, sf, bits)
utee/quant.py:42
↓ 1 callers
Function
main
()
quantize.py:8
↓ 1 callers
Function
make_layers
(cfg, batch_norm=False)
svhn/model.py:30
↓ 1 callers
Function
make_layers
(cfg, batch_norm=False)
stl10/model.py:30
↓ 1 callers
Function
min_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
Function
alexnet
(pretrained=False, model_root=None, **kwargs)
imagenet/alexnet.py:49
Function
auto_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
Function
cifar10
(cuda=True, model_root=None)
utee/selector.py:34
Function
cifar100
(cuda=True, model_root=None)
utee/selector.py:42
Function
cifar100
(n_channel, pretrained=None)
cifar/model.py:59
Method
counter
(self)
utee/quant.py:96
Method
counter
(self)
utee/quant.py:124
Method
counter
(self)
utee/quant.py:150
Function
dump_pickle
(obj, path)
utee/misc.py:52
Function
duplicate_model_with_quant
assume that original model has at least a nn.Sequential
utee/quant.py:160
Function
ensure_dir
(path, erase=False)
utee/misc.py:36
Function
eval_model
(model, ds, n_sample=None, ngpu=1, is_imagenet=False)
utee/misc.py:152
Method
forward
(self, input)
utee/misc.py:165
Method
forward
(self, input)
utee/quant.py:99
Method
forward
(self, input)
utee/quant.py:127
Method
forward
(self, input)
utee/quant.py:153
Method
forward
(self, input)
mnist/model.py:33
Method
forward
(self, x)
imagenet/alexnet.py:42
Method
forward
(self, x)
imagenet/squeezenet.py:45
Method
forward
(self, x)
imagenet/squeezenet.py:111
Method
forward
(self, x)
imagenet/vgg.py:36
Method
forward
(self, x)
imagenet/inception.py:68
Method
forward
(self, x)
imagenet/inception.py:142
Method
forward
(self, x)
imagenet/inception.py:169
Method
forward
(self, x)
imagenet/inception.py:201
Method
forward
(self, x)
imagenet/inception.py:233
Method
forward
(self, x)
imagenet/inception.py:264
Method
forward
(self, x)
imagenet/inception.py:306
Method
forward
(self, x)
imagenet/inception.py:332
Method
forward
(self, x)
imagenet/resnet.py:39
Method
forward
(self, x)
imagenet/resnet.py:70
Method
forward
(self, x)
imagenet/resnet.py:131
Method
forward
(self, x)
svhn/model.py:24
Method
forward
(self, x)
cifar/model.py:25
Method
forward
(self, x)
stl10/model.py:24
Function
get
(batch_size, data_root='/tmp/public_dataset/pytorch', train=True, val=True, **kwargs)
mnist/dataset.py:6
Function
get
(batch_size, data_root='/tmp/public_dataset/pytorch', train=True, val=True, **kwargs)
svhn/dataset.py:6
Function
get10
(batch_size, data_root='/tmp/public_dataset/pytorch', train=True, val=True, **kwargs)
cifar/dataset.py:6
Function
get100
(batch_size, data_root='/tmp/public_dataset/pytorch', train=True, val=True, **kwargs)
cifar/dataset.py:38
Function
img2str
(img)
utee/misc.py:144
Function
inception_v3
(cuda=True, model_root=None)
utee/selector.py:98
Function
inception_v3
(pretrained=False, model_root=None, **kwargs)
imagenet/inception.py:16
Method
info
(self, str_info)
utee/misc.py:30
Function
load_lmdb
(lmdb_file, n_records=None)
utee/misc.py:117
Function
load_pickle
(path)
utee/misc.py:44
Function
load_state_dict
(model, model_urls, model_root)
utee/misc.py:201
Function
log_minmax_quantize
(input, bits)
utee/quant.py:31
Function
md5
(s)
utee/misc.py:147
Function
mnist
(cuda=True, model_root=None)
utee/selector.py:18
Function
mnist
(input_dims=784, n_hiddens=[256, 256], n_class=10, pretrained=None)
mnist/model.py:38
Function
model_snapshot
(model, new_file, old_file=None, verbose=False)
utee/misc.py:97
next →
1–100 of 136, ranked by callers