Functions — github.com/HRNet/Lite-HRNet

hub / github.com/HRNet/Lite-HRNet / functions

Functions104 in github.com/HRNet/Lite-HRNet

⨍ Functions 104◇ Types & classes 13 ↳ Endpoints 2

↓ 7 callersMethod__init__

(self, in_channels, stem_channels, out_channels,

models/backbones/litehrnet.py:164

↓ 5 callersFunctionbuild

Build a module. Args: cfg (dict, list[dict]): The config of modules, it is either a dict or a list of configs. regist

models/builder.py:7

↓ 3 callersFunctionbuild_posenet

Build posenet.

models/builder.py:50

↓ 3 callersFunctioncompute_Conv2d_flops

(module, inp, out)

tools/torchstat/compute_flops.py:45

↓ 3 callersFunctionnum_params

(module)

tools/torchstat/compute_memory.py:25

↓ 3 callersFunctionround_value

(value, binary=False)

tools/torchstat/reporter.py:8

↓ 2 callersFunction_flatten

Flattens the tree of tensors to flattened sequence of tensors

tools/torchstat/analyzer.py:91

↓ 2 callersFunction_round_value

(value, binary=False)

tools/torchstat_utils.py:51

↓ 2 callersMethodfind_child_index

(self, child_name)

tools/torchstat/stat_tree.py:181

↓ 1 callersMethod_check_branches

Check input to avoid ValueError.

models/backbones/litehrnet.py:478

↓ 1 callersFunction_convert_leaf_modules_to_stat_tree

(leaf_modules)

tools/torchstat/analyzer.py:152

↓ 1 callersFunction_for_leaf

(model, fn, *args)

tools/torchstat/analyzer.py:66

↓ 1 callersMethod_make_fuse_layers

Make fuse layer.

models/backbones/litehrnet.py:533

↓ 1 callersMethod_make_naive_branches

Make branches.

models/backbones/litehrnet.py:524

↓ 1 callersMethod_make_one_branch

Make one branch.

models/backbones/litehrnet.py:499

↓ 1 callersMethod_make_stage

(self, stages_spec, stage_index, in_channels,

models/backbones/litehrnet.py:823

↓ 1 callersMethod_make_transition_layer

Make transition layer.

models/backbones/litehrnet.py:755

↓ 1 callersMethod_make_weighting_blocks

(self, num_blocks, reduce_ratio, stride=1)

models/backbones/litehrnet.py:485

↓ 1 callersMethodadd_child

(self, node)

tools/torchstat/stat_tree.py:190

↓ 1 callersFunctionanalyze

(model: nn.Module, input_size, query_granularity: int)

tools/torchstat/analyzer.py:40

↓ 1 callersFunctioncompute_AvgPool2d_madd

(module, inp, out)

tools/torchstat/compute_madd.py:77

↓ 1 callersFunctioncompute_BatchNorm2d_flops

(module, inp, out)

tools/torchstat/compute_flops.py:120

↓ 1 callersFunctioncompute_BatchNorm2d_madd

(module, inp, out)

tools/torchstat/compute_madd.py:51

↓ 1 callersFunctioncompute_BatchNorm2d_memory

(module, inp, out)

tools/torchstat/compute_memory.py:62

↓ 1 callersFunctioncompute_Bilinear_madd

(module, inp1, inp2, out)

tools/torchstat/compute_madd.py:127

↓ 1 callersFunctioncompute_Conv2d_madd

(module, inp, out)

tools/torchstat/compute_madd.py:8

↓ 1 callersFunctioncompute_Conv2d_memory

(module, inp, out)

tools/torchstat/compute_memory.py:48

↓ 1 callersFunctioncompute_ConvTranspose2d_flops

(module, inp, out)

tools/torchstat/compute_flops.py:70

↓ 1 callersFunctioncompute_ConvTranspose2d_madd

(module, inp, out)

tools/torchstat/compute_madd.py:30

↓ 1 callersFunctioncompute_Linear_flops

(module, inp, out)

tools/torchstat/compute_flops.py:165

↓ 1 callersFunctioncompute_Linear_madd

(module, inp, out)

tools/torchstat/compute_madd.py:115

↓ 1 callersFunctioncompute_Linear_memory

(module, inp, out)

tools/torchstat/compute_memory.py:73

↓ 1 callersFunctioncompute_MaxPool2d_madd

(module, inp, out)

tools/torchstat/compute_madd.py:64

↓ 1 callersFunctioncompute_PReLU_memory

(module, inp, out)

tools/torchstat/compute_memory.py:38

↓ 1 callersFunctioncompute_Pool2d_flops

(module, input, out)

tools/torchstat/compute_flops.py:142

↓ 1 callersFunctioncompute_Pool2d_memory

(module, inp, out)

tools/torchstat/compute_memory.py:86

↓ 1 callersFunctioncompute_ReLU_flops

(module, inp, out)

tools/torchstat/compute_flops.py:130

↓ 1 callersFunctioncompute_ReLU_madd

(module, inp, out)

tools/torchstat/compute_madd.py:93

↓ 1 callersFunctioncompute_ReLU_memory

(module, inp, out)

tools/torchstat/compute_memory.py:29

↓ 1 callersFunctioncompute_Softmax_madd

(module, inp, out)

tools/torchstat/compute_madd.py:102

↓ 1 callersFunctioncompute_Upsample_flops

(module, inp, out)

tools/torchstat/compute_flops.py:172

↓ 1 callersFunctioncompute_adaptivepool_flops

(module, input, output)

tools/torchstat/compute_flops.py:99

↓ 1 callersFunctioncompute_flops

(module, inp, out)

tools/torchstat/compute_flops.py:7

↓ 1 callersFunctioncompute_madd

(module, inp, out)

tools/torchstat/compute_madd.py:141

↓ 1 callersFunctioncompute_matmul_flops

(moudle, inp, out)

tools/torchstat/compute_flops.py:37

↓ 1 callersFunctioncompute_memory

(module, inp, out)

tools/torchstat/compute_memory.py:6

↓ 1 callersMethodget_collected_stat_nodes

(self, query_granularity)

tools/torchstat/stat_tree.py:27

↓ 1 callersFunctionget_parent_node

(root_node, stat_node_name)

tools/torchstat/analyzer.py:139

↓ 1 callersMethodget_same_level_max_node_depth

(self, query_node)

tools/torchstat/stat_tree.py:11

↓ 1 callersFunctionmain

()

tools/train.py:72

↓ 1 callersFunctionmain

()

tools/test.py:70

↓ 1 callersFunctionmain

()

tools/summary_network.py:28

↓ 1 callersFunctionmerge_configs

(cfg1, cfg2)

tools/test.py:59

↓ 1 callersFunctionmodel_stats

(model, input_shape)

tools/torchstat_utils.py:46

↓ 1 callersFunctionmodel_stats2df

(model_stats: ModelStats)

tools/torchstat_utils.py:65

↓ 1 callersFunctionparse_args

()

tools/train.py:20

↓ 1 callersFunctionparse_args

()

tools/test.py:18

↓ 1 callersFunctionparse_args

()

tools/summary_network.py:13

↓ 1 callersFunctionreport_format

(collected_nodes)

tools/torchstat/reporter.py:22

↓ 1 callersMethodupdate_stat_nodes_granularity

(self)

tools/torchstat/stat_tree.py:18

MethodFlops

(self)

tools/torchstat/stat_tree.py:146

MethodMAdd

(self)

tools/torchstat/stat_tree.py:135

MethodMemory

(self)

tools/torchstat/stat_tree.py:157

Method__init__

(self, node)

tools/torchstat_utils.py:11

Method__init__

(self, model, input_shape, clone_model=False)

tools/torchstat_utils.py:25

Method__init__

(self, name)

tools/torchstat/analyzer.py:20

Method__init__

(self, root_node)

tools/torchstat/stat_tree.py:6

Method__init__

(self, name=str(), parent=None)

tools/torchstat/stat_tree.py:46

Method__init__

(self, channels, ratio=16, conv_cfg=None,

models/backbones/litehrnet.py:19

Method__init__

(self, channels, ratio=16, conv_cfg=None,

models/backbones/litehrnet.py:54

Method__init__

(self, in_channels, stride, reduce_ratio,

models/backbones/litehrnet.py:100

Method__init__

(self, in_channels, conv_cfg=None, norm_cfg=dict(type='BN'))

models/backbones/litehrnet.py:274

Method__init__

(self, in_channels, out_channels, stride=1,

models/backbones/litehrnet.py:344

Method__init__

( self, num_branches, num_blocks, in_channels, red

models/backbones/litehrnet.py:444

Method__init__

(self, extra, in_channels=3, conv_cfg=None,

models/backbones/litehrnet.py:706

Function_forward_post_hook

(module_stats: ModuleStats, module: nn.Module, input, output)

tools/torchstat/analyzer.py:108

Function_forward_pre_hook

(module_stats: ModuleStats, module: nn.Module, input)

tools/torchstat/analyzer.py:103

Method_inner_forward

(x)

models/backbones/litehrnet.py:140

Method_inner_forward

(x)

models/backbones/litehrnet.py:250

Method_inner_forward

(x)

models/backbones/litehrnet.py:423

Function_register_hooks

(name: str, module: nn.Module, pre_hooks, post_hooks, stats)

tools/torchstat/analyzer.py:72

Functionbuild_backbone

tools/torchstat/stat_tree.py:80

Methodduration

(self)

tools/torchstat/stat_tree.py:171

Methodforward

(self, x)

models/backbones/litehrnet.py:45

Methodforward

(self, x)

models/backbones/litehrnet.py:84

Methodforward

(self, x)

models/backbones/litehrnet.py:138

Methodforward

(self, x)

models/backbones/litehrnet.py:248

Methodforward

(self, x)

models/backbones/litehrnet.py:307

Methodforward

(self, x)

models/backbones/litehrnet.py:421

Methodforward

Forward function.

models/backbones/litehrnet.py:618

Methodforward

Forward function.

models/backbones/litehrnet.py:885

Methodgranularity

(self)

tools/torchstat/stat_tree.py:72

Methodinference_memory

(self)

tools/torchstat/stat_tree.py:124

Methodinit_weights

Initialize the weights in backbone. Args: pretrained (str, optional): Path to pre-trained weights. Defaults to No

models/backbones/litehrnet.py:859

Methodinput_shape

(self)

tools/torchstat/stat_tree.py:87

Methodname

(self)

tools/torchstat/stat_tree.py:64

next →1–100 of 104, ranked by callers