MCPcopy Create free account

hub / github.com/arunos728/MotionSqueeze / functions

Functions133 in github.com/arunos728/MotionSqueeze

Functioncount_avgpool
(m, x, y)
thop/count_hooks.py:74
Functioncount_bn
(m, x, y)
thop/count_hooks.py:43
Functioncount_convNd
(m, x, y)
thop/count_hooks.py:17
Functioncount_convNd_ver2
(m, x, y)
thop/count_hooks.py:29
Functioncount_correlation
(m, x, y)
thop/count_hooks.py:134
Functioncount_linear
(m, x, y)
thop/count_hooks.py:123
Functioncount_relu
(m, x, y)
thop/count_hooks.py:53
Functioncount_softmax
(m, x, y)
thop/count_hooks.py:61
Functioncount_upsample
(m, x, y)
thop/count_hooks.py:95
Methodcrop_size
(self)
models.py:450
Methodforward
(self, input, temperature)
models.py:333
Methodforward
(self, x)
resnet_TSM.py:56
Methodforward
(self, x)
resnet_TSM.py:88
Methodforward
(self, x)
resnet_TSM.py:123
Methodforward
(self, feature1, feature2)
resnet_TSM.py:159
Methodforward
(self, feature1, feature2)
resnet_TSM.py:212
Methodforward
(self, x, res, match_v)
resnet_TSM.py:271
Methodforward
(self, x, temperature)
resnet_TSM.py:475
Methodforward
(self, input)
ops/basic_ops.py:6
Methodforward
(self, input_tensor)
ops/basic_ops.py:17
Methodforward
(self, input)
ops/basic_ops.py:46
Functionget_grad_hook
(name)
ops/utils.py:5
Methodget_optim_policies_BN2to1D
(self)
models.py:264
Functionhook
(m, grad_in, grad_out)
ops/utils.py:6
Methodlabel
(self)
dataset.py:22
Functionlog_add
(log_a, log_b)
ops/utils.py:22
Methodnum_frames
(self)
dataset.py:18
Methodpath
(self)
dataset.py:14
Functionprofile
(model, inputs, custom_ops=custom_ops, verbose=True)
thop/profile.py:66
Functionresnet101
Constructs a ResNet-101 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
resnet_TSM.py:557
Functionresnet34
Constructs a ResNet-34 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
resnet_TSM.py:519
Methodscale_size
(self)
models.py:454
Functionsoftmax
(scores)
ops/utils.py:17
← previous101–133 of 133, ranked by callers