Functions — github.com/Alibaba-MIIL/TResNet

CodeHub MCPcopy Create free account

hub / github.com/Alibaba-MIIL/TResNet / functions

Functions73 in github.com/Alibaba-MIIL/TResNet

⨍ Functions 73 ◇ Types & classes 29

↓ 11 callersFunctionconv2d_ABN

(ni, nf, stride, activation="leaky_relu", kernel_size=3, activation_param=1e-2, groups=1)

src/models/tresnet_v2/tresnet_v2.py:51

↓ 11 callersFunctionconv2d_ABN

(ni, nf, stride, activation="leaky_relu", kernel_size=3, activation_param=1e-2, groups=1)

src/models/tresnet/tresnet.py:21

↓ 4 callersMethod_make_layer

(self, block, planes, blocks, stride=1, use_se=True, anti_alias_layer=None)

src/models/tresnet_v2/tresnet_v2.py:201

↓ 4 callersMethod_make_layer

(self, block, planes, blocks, stride=1, use_se=True, anti_alias_layer=None)

src/models/tresnet/tresnet.py:169

↓ 2 callersMethod__init__

(self, layers, in_chans=3, num_classes=1000, width_factor=1.0, remove_model_jit=False)

src/models/tresnet_v2/tresnet_v2.py:148

↓ 2 callersMethod__init__

(self, block_size=4)

src/models/tresnet_v2/layers/space_to_depth.py:6

↓ 2 callersMethod__init__

(self, layers, in_chans=3, num_classes=1000, width_factor=1.0, remove_aa_jit=False)

src/models/tresnet/tresnet.py:116

↓ 2 callersMethod__init__

(self, block_size=4)

src/models/tresnet/layers/space_to_depth.py:6

↓ 1 callersFunctionInplacABN_to_ABN

(module: nn.Module)

src/models/tresnet_v2/tresnet_v2.py:18

↓ 1 callersFunctionTResnetL

Constructs a large TResnet model.

src/models/tresnet/tresnet.py:206

↓ 1 callersFunctionTResnetL_V2

Constructs a large TResnet model.

src/models/tresnet_v2/tresnet_v2.py:229

↓ 1 callersFunctionTResnetM

Constructs a medium TResnet model.

src/models/tresnet/tresnet.py:195

↓ 1 callersFunctionTResnetXL

Constructs an extra-large TResnet model.

src/models/tresnet/tresnet.py:217

↓ 1 callersMethod__init__

(self, filt_size=3, stride=2, channels=None)

src/models/tresnet_v2/layers/anti_aliasing.py:44

↓ 1 callersMethod__init__

(self, channels, reduction_channels, inplace=True)

src/models/tresnet_v2/layers/squeeze_and_excite.py:13

↓ 1 callersMethod__init__

(self, filt_size=3, stride=2, channels=None)

src/models/tresnet/layers/anti_aliasing.py:44

↓ 1 callersMethod__init__

(self, channels, reduction_channels, inplace=True)

src/models/tresnet/layers/squeeze_and_excite.py:13

↓ 1 callersFunctionaccuracy

Computes the precision@k for the specified values of k

src/helper_functions/helper_functions.py:24

↓ 1 callersFunctioncreate_dataloader

src/helper_functions/helper_functions.py:7

↓ 1 callersFunctioncreate_model

src/models/utils/factory.py:10

↓ 1 callersFunctionmain

↓ 1 callersMethodreset

src/helper_functions/helper_functions.py:45

↓ 1 callersMethodupdate

(self, val, n=1)

src/helper_functions/helper_functions.py:47

↓ 1 callersFunctionvalidate

(model, val_loader)

src/helper_functions/helper_functions.py:54

FunctionIABN2Float

If `module` is IABN don't use half precision.

src/models/tresnet/tresnet.py:13

(self, input: torch.Tensor)

src/models/tresnet_v2/layers/anti_aliasing.py:36

(self, x: torch.Tensor)

src/models/tresnet_v2/layers/space_to_depth.py:21

(self, input: torch.Tensor)

src/models/tresnet/layers/anti_aliasing.py:36

(self, x: torch.Tensor)

src/models/tresnet/layers/space_to_depth.py:21

src/helper_functions/helper_functions.py:43

(self, inplanes, planes, stride=1, downsample=None, use_se=True, anti_alias_layer=None)

src/models/tresnet_v2/tresnet_v2.py:63

(self, inplanes, planes, stride=1, downsample=None, use_se=True, anti_alias_layer=None)

src/models/tresnet_v2/tresnet_v2.py:103

(self, remove_aa_jit: bool = False, filt_size: int = 3, stride: int = 2, channels: int = 0)

src/models/tresnet_v2/layers/anti_aliasing.py:9

(self, filt_size: int = 3, stride: int = 2, channels: int = 0)

src/models/tresnet_v2/layers/anti_aliasing.py:23

(self, flatten=False)

src/models/tresnet_v2/layers/avg_pool.py:8

(self, inplace=True)

src/models/tresnet_v2/layers/squeeze_and_excite.py:31

(self, remove_model_jit=False)

src/models/tresnet_v2/layers/space_to_depth.py:31

(self, block_size)

src/models/tresnet_v2/layers/space_to_depth.py:44

(self, inplanes, planes, stride=1, downsample=None, use_se=True, anti_alias_layer=None)

src/models/tresnet/tresnet.py:32

(self, inplanes, planes, stride=1, downsample=None, use_se=True, anti_alias_layer=None)

src/models/tresnet/tresnet.py:71

(self, remove_aa_jit: bool = False, filt_size: int = 3, stride: int = 2, channels: int = 0)

src/models/tresnet/layers/anti_aliasing.py:9

(self, filt_size: int = 3, stride: int = 2, channels: int = 0)

src/models/tresnet/layers/anti_aliasing.py:23

(self, flatten=False)

src/models/tresnet/layers/avg_pool.py:8

(self, inplace=True)

src/models/tresnet/layers/squeeze_and_excite.py:31

(self, remove_model_jit=False)

src/models/tresnet/layers/space_to_depth.py:31

(self, block_size)

src/models/tresnet/layers/space_to_depth.py:44

(ni, nf, stride)

src/models/tresnet_v2/tresnet_v2.py:43

Functionconv3x3

(in_planes, out_planes, stride=1)

src/models/tresnet_v2/tresnet_v2.py:35

Functionconv3x3_depth

(planes, stride=1)

src/models/tresnet_v2/tresnet_v2.py:39

src/models/tresnet_v2/tresnet_v2.py:82

src/models/tresnet_v2/tresnet_v2.py:129

src/models/tresnet_v2/tresnet_v2.py:222

src/models/tresnet_v2/layers/anti_aliasing.py:17

src/models/tresnet_v2/layers/anti_aliasing.py:60

src/models/tresnet_v2/layers/avg_pool.py:12

src/models/tresnet_v2/layers/squeeze_and_excite.py:7

src/models/tresnet_v2/layers/squeeze_and_excite.py:22

src/models/tresnet_v2/layers/squeeze_and_excite.py:35

src/models/tresnet_v2/layers/space_to_depth.py:11

src/models/tresnet_v2/layers/space_to_depth.py:38

src/models/tresnet_v2/layers/space_to_depth.py:48

src/models/tresnet/tresnet.py:50

src/models/tresnet/tresnet.py:97

src/models/tresnet/tresnet.py:188

src/models/tresnet/layers/anti_aliasing.py:17

src/models/tresnet/layers/anti_aliasing.py:60

src/models/tresnet/layers/avg_pool.py:12

src/models/tresnet/layers/squeeze_and_excite.py:7

src/models/tresnet/layers/squeeze_and_excite.py:22

src/models/tresnet/layers/squeeze_and_excite.py:35

src/models/tresnet/layers/space_to_depth.py:11

src/models/tresnet/layers/space_to_depth.py:38

src/models/tresnet/layers/space_to_depth.py:48