MCPcopy Create free account

hub / github.com/AberHu/TF-NAS / functions

Functions144 in github.com/AberHu/TF-NAS

⨍ Functions 144◇ Types & classes 17
↓ 32 callersMethodupdate
(self, val, n=1)
tools/utils.py:54
↓ 19 callersFunctionmeasure_latency_in_ms
(model, input_shape, is_cuda)
tools/utils.py:12
↓ 8 callersFunctionaccuracy
Computes the precision@k for the specified values of k
tools/utils.py:61
↓ 8 callersMethodreset_switches
(self)
models/model_search.py:118
↓ 7 callersFunctionget_mc_num_dddict
(mc_mask_dddict, is_max=False)
parsing_model.py:76
↓ 7 callersMethodweight_parameters
(self)
models/model_search.py:311
↓ 6 callersMethod__init__
(self, inplace=False)
models/layers.py:27
↓ 6 callersMethod_make_stage
(self, stage_name, ics, ocs, ss, affs, acts)
models/model_eval.py:95
↓ 6 callersMethod_make_stage
(self, stage_name)
models/model_eval.py:284
↓ 6 callersMethodarch_parameters
(self)
models/model_search.py:320
↓ 6 callersFunctionreduce_tensor
(tensor)
train_eval_amp.py:399
↓ 5 callersFunctionget_op_and_depth_weights
(model_or_path)
parsing_model.py:23
↓ 5 callersFunctionparse_architecture
(op_weights, depth_weights)
parsing_model.py:44
↓ 5 callersFunctionset_layer_from_config
(layer_config)
models/layers.py:10
↓ 4 callersFunctioncount_parameters_in_MB
(model)
tools/utils.py:114
↓ 4 callersMethodfink_ori_idx
(self, idx)
models/model_search.py:49
↓ 4 callersFunctionfit_mc_num_by_latency
(parsed_arch, mc_num_dddict, mc_maxnum_dddict, lat_lookup_key_dddict, lat_lookup, target_lat, stages, sign)
train_search.py:478
↓ 3 callersFunctionchannel_shuffle
(x, groups)
tools/utils.py:89
↓ 3 callersFunctioncreate_exp_dir
(path, scripts_to_save=None)
tools/utils.py:126
↓ 3 callersFunctionget_lookup_latency
(parsed_arch, mc_num_dddict, lat_lookup_key_dddict, lat_lookup)
train_search.py:465
↓ 2 callersMethod__init__
(self, in_channels, out_channels, stride, affine, act_func, num_ops, mc_num_dict, lat_lookup)
models/model_search.py:33
↓ 2 callersFunctionget_latency_lookup
(is_cuda)
latency_pkl/make_lat_lut_example.py:44
↓ 2 callersFunctionget_same_padding
(kernel_size)
tools/utils.py:103
↓ 2 callersFunctionsave_checkpoint
(state, is_best, save)
tools/utils.py:118
↓ 2 callersFunctionset_seed
(seed)
train_eval_amp.py:108
↓ 2 callersMethodset_temperature
(self, T)
models/model_search.py:121
↓ 1 callersMethod__init__
(self, num_classes, parsed_arch, mc_num_dddict, lat_lookup=None, dropout_rate=0.0, drop_connect_rate=0.0)
models/model_eval.py:32
↓ 1 callersMethod_get_block_count
(self)
models/model_eval.py:88
↓ 1 callersMethod_get_block_count
(self)
models/model_eval.py:276
↓ 1 callersMethod_initialization
(self)
models/model_eval.py:232
↓ 1 callersMethod_initialization
(self)
models/model_eval.py:418
↓ 1 callersMethod_initialization
(self)
models/model_search.py:352
↓ 1 callersMethod_initialize_betas
(self)
models/model_search.py:208
↓ 1 callersMethod_initialize_log_alphas
(self)
models/model_search.py:113
↓ 1 callersFunctionadd_batch_counter_hook_function
(module)
tools/flops_benchmark.py:105
↓ 1 callersFunctionadd_flops_counting_methods
( model )
tools/flops_benchmark.py:26
↓ 1 callersMethodbetas_parameters
(self)
models/model_search.py:338
↓ 1 callersFunctionbound_clip
(mc_num, max_mc_num)
train_search.py:519
↓ 1 callersMethodbuild_from_config
(config)
models/layers.py:68
↓ 1 callersFunctioncalculate_FLOPs_in_M
(model, shape)
tools/flops_benchmark.py:9
↓ 1 callersFunctioncompute_average_flops_cost
A method that will be available after add_flops_counting_methods() is called on a desired net object. Returns current mean flops consumption pe
tools/flops_benchmark.py:34
↓ 1 callersFunctiondrop_connect
Apply drop connect.
tools/utils.py:77
↓ 1 callersMethodget_lookup_latency
(self, x)
models/model_eval.py:134
↓ 1 callersMethodget_lookup_latency
(self, size)
models/model_search.py:93
↓ 1 callersMethodlog_alphas_parameters
(self)
models/model_search.py:329
↓ 1 callersFunctionmain
()
test.py:47
↓ 1 callersFunctionmain
()
train_eval.py:93
↓ 1 callersFunctionmain
()
train_search.py:79
↓ 1 callersFunctionmain
()
train_eval_amp.py:114
↓ 1 callersFunctionset_seed
(seed)
train_eval.py:87
↓ 1 callersFunctiontrain
(train_queue, model, criterion, optimizer)
train_eval.py:228
↓ 1 callersFunctiontrain
(train_loader, model, criterion, optimizer)
train_eval_amp.py:310
↓ 1 callersFunctiontrain_w_arch
(train_queue, val_queue, model, criterion, optimizer_w, optimizer_a)
train_search.py:357
↓ 1 callersFunctiontrain_wo_arch
(train_queue, model, criterion, optimizer_w)
train_search.py:318
↓ 1 callersFunctionvalidate
(val_queue, model)
test.py:95
↓ 1 callersFunctionvalidate
(val_queue, model, criterion)
train_eval.py:271
↓ 1 callersFunctionvalidate
(val_queue, model, criterion)
train_search.py:435
↓ 1 callersFunctionvalidate
(val_loader, model, criterion)
train_eval_amp.py:366
↓ 1 callersMethodweight_call
(self, x)
models/layers.py:143
Method__getitem__
(self, index)
dataset/dataset.py:39
Method__init__
(self, num_classes, epsilon)
train_eval.py:73
Method__init__
(self, num_classes, epsilon)
train_eval_amp.py:94
Method__init__
(self)
tools/utils.py:42
Method__init__
(self, root, list_path, transform=None, list_reader=default_list_reader, loader=default_loader)
dataset/dataset.py:33
Method__init__
(self, inplace=False)
models/layers.py:39
Method__init__
( self, in_channels, out_channels, use_bn=True, affine = True, act_func='relu6', ops_orde
models/layers.py:80
Method__init__
( self, in_channels, out_channels, kernel_size=3, stride=1, groups=1, has_shuffle=False,
models/layers.py:192
Method__init__
( self, in_channels, out_channels, use_bn=False, affine=False, act_func=None, ops_order='
models/layers.py:276
Method__init__
( self, in_features, out_features, bias=True, use_bn=False, affine=False, act_func=None,
models/layers.py:324
Method__init__
( self, in_channels, mid_channels, se_channels, out_channels, kernel_size=3, stride=1,
models/layers.py:433
Method__init__
(self, num_classes, model_config, lat_lookup=None, dropout_rate=0.0, drop_connect_rate=0.0)
models/model_eval.py:248
Method__init__
(self, ics, ocs, ss, affs, acts, mc_num_ddict, lat_lookup, stage_type)
models/model_search.py:126
Method__init__
(self, num_classes, mc_num_dddict, lat_lookup)
models/model_search.py:214
Method__len__
(self)
dataset/dataset.py:48
Functionadap_pool_flops_counter_hook
(pool_module, inputs, output)
tools/flops_benchmark.py:59
Functionadd_flops_counter_hook_function
(module)
tools/flops_benchmark.py:118
Functionadd_flops_counter_variable_or_reset
(module)
tools/flops_benchmark.py:111
Functionbatch_counter_hook
(module, inputs, output)
tools/flops_benchmark.py:98
Methodbn_before_weight
(self)
models/layers.py:135
Methodbn_before_weight
(self)
models/layers.py:377
Methodbuild_from_config
(config)
models/layers.py:180
Methodbuild_from_config
(config)
models/layers.py:264
Methodbuild_from_config
(config)
models/layers.py:312
Methodbuild_from_config
(config)
models/layers.py:421
Methodbuild_from_config
(config)
models/layers.py:599
Methodconfig
(self)
models/layers.py:64
Methodconfig
(self)
models/layers.py:169
Methodconfig
(self)
models/layers.py:251
Methodconfig
(self)
models/layers.py:304
Methodconfig
(self)
models/layers.py:408
Methodconfig
(self)
models/layers.py:581
Methodconfig
(self)
models/model_eval.py:218
Methodconfig
(self)
models/model_eval.py:404
Functionconv_flops_counter_hook
(conv_module, inputs, output)
tools/flops_benchmark.py:80
Functioncv2_loader
(path)
dataset/dataset.py:13
Functiondefault_list_reader
(list_path)
dataset/dataset.py:22
Functionfc_flops_counter_hook
(fc_module, inputs, output)
tools/flops_benchmark.py:70
Methodforward
(self, xs, targets)
train_eval.py:79
Methodforward
(self, xs, targets)
train_eval_amp.py:100
Methodforward
(self, x)
models/layers.py:31
next →1–100 of 144, ranked by callers