MCPcopy Create free account

hub / github.com/Yang-YiFan/DiracDeltaNet / functions

Functions52 in github.com/Yang-YiFan/DiracDeltaNet

↓ 6 callersMethod__init__
(self, planes)
models/diracdeltanet.py:136
↓ 6 callersMethod_make_layer
(self, block, planes, num_blocks, stride, weight_bit, act_bit, first_block=False, extern_init=False, init_mode
models/diracdeltanet.py:331
↓ 6 callersMethod_make_layer
(self, block, planes, num_blocks, stride, weight_bit, act_bit, extern_init=False, init_model=nn.Sequential())
models/shufflenetv2.py:201
↓ 4 callersMethod_make_layer
(self, planes, num_layers, stride=1)
models/resnet50.py:22
↓ 3 callersFunctionprogress_bar
(current, total, msg=None)
extensions/utils.py:54
↓ 2 callersFunctionDiracDeltaNet_wrapper
(expansion=1, base_channelsize=48, num_classes=1000, weight_bit=32, act_bit=32, first_weight_bit=32, first_act
models/diracdeltanet.py:369
↓ 2 callersMethod__init__
(self, in_planes, out_planes, stride=1, expansion=1, weight_bit=3, act_bit=4, extern_init=False, init_model=nn
models/shufflenetv2.py:58
↓ 2 callersFunctionaccuracy
Computes the precision@k for the specified values of k
train.py:140
↓ 2 callersFunctionformat_time
(seconds)
extensions/utils.py:97
↓ 1 callersMethod__init__
(self)
models/resnet50.py:10
↓ 1 callersFunctionaccuracy
Computes the precision@k for the specified values of k
test.py:75
↓ 1 callersFunctionadjust_learning_rate
(optimizer, iteration, lr)
train.py:157
↓ 1 callersMethodbackward
(ctx, grad_output)
models/shufflenetv2.py:19
↓ 1 callersMethodclassifier
(self, x)
models/resnet50.py:43
↓ 1 callersMethodfeats
(self, x)
models/resnet50.py:46
↓ 1 callersFunctiontest
(epoch)
train.py:217
↓ 1 callersFunctiontest
()
test.py:92
↓ 1 callersFunctiontrain
(epoch)
train.py:180
MethodLR_REGIME
(self)
extensions/model_refinery_wrapper.py:20
FunctionShuffleNetv2_wrapper
(expansion=1, base_channelsize=48, num_classes=1000, weight_bit=32, act_bit=32, extern_init=False, init_model=
models/shufflenetv2.py:230
Method__call__
Args: img (Tensor): Tensor image of size (C, H, W). Returns: Tensor: Image with n_holes of dimension leng
extensions/utils.py:141
Method__init__
(self, n_holes, length)
extensions/utils.py:137
Method__init__
(self, model, label_refinery)
extensions/model_refinery_wrapper.py:8
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
models/resnet50.py:69
Method__init__
(self, act_bit=4, scale_coef=10.0, extern_init=False, init_model=nn.Sequential())
models/diracdeltanet.py:24
Method__init__
(self, act_bit=4, scale_coef=10.0, fixed_rescale=6.0, extern_init=False, init_model=nn.Sequential())
models/diracdeltanet.py:45
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bi
models/diracdeltanet.py:68
Method__init__
(self, in_features, out_features, bias=True, bit=32, extern_init=False, init_model=nn.Sequential())
models/diracdeltanet.py:104
Method__init__
(self, in_planes, out_planes, stride=1, expansion=1, weight_bit=3, act_bit=4, first_block=False, extern_init=F
models/diracdeltanet.py:170
Method__init__
(self, block, num_blocks, base_channelsize=48, num_classes=1000, weight_bit=4, act_bit=4, first_weight_bit=32,
models/diracdeltanet.py:277
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bi
models/shufflenetv2.py:24
Method__init__
(self, block, num_blocks, base_channelsize=48, num_classes=1000, weight_bit=4, act_bit=4, extern_init=False, i
models/shufflenetv2.py:167
Methodbackward
(ctx, grad_output)
models/diracdeltanet.py:20
Methodforward
(self, input)
extensions/model_refinery_wrapper.py:28
Methodforward
(self, output, target)
extensions/refinery_loss.py:15
Methodforward
(self, x)
models/resnet50.py:61
Methodforward
(self, x)
models/resnet50.py:80
Methodforward
(ctx, input, pwr_coef)
models/diracdeltanet.py:15
Methodforward
(self, x)
models/diracdeltanet.py:36
Methodforward
(self, x)
models/diracdeltanet.py:58
Methodforward
(self, x)
models/diracdeltanet.py:84
Methodforward
(self, x)
models/diracdeltanet.py:116
Methodforward
(self, input)
models/diracdeltanet.py:157
Methodforward
(self, x)
models/diracdeltanet.py:240
Methodforward
(self, x)
models/diracdeltanet.py:349
Methodforward
(ctx, input, pwr_coef)
models/shufflenetv2.py:14
Methodforward
(self, x)
models/shufflenetv2.py:137
Methodforward
(self, x)
models/shufflenetv2.py:217
Methodfoward
(self, x)
models/shufflenetv2.py:40
Functionget_mean_and_std
Compute the mean and std value of dataset.
extensions/utils.py:18
Functioninit_params
Init layer parameters.
extensions/utils.py:32
Methodstate_dict
(self)
extensions/model_refinery_wrapper.py:25