MCPcopy Create free account

hub / github.com/DingXiaoH/GSM-SGD / functions

Functions233 in github.com/DingXiaoH/GSM-SGD

⨍ Functions 233◇ Types & classes 28
↓ 30 callersMethodConv2dBNReLU
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, padding_mode='zeros'
builder.py:65
↓ 16 callersMethodLinear
(self, in_features, out_features, bias=True)
builder.py:81
↓ 12 callersMethodupdate
(self, mean_var, count=1)
utils/misc.py:106
↓ 11 callersMethodReLU
(self)
builder.py:51
↓ 10 callersMethodMaxpool2d
(self, kernel_size, stride=None)
builder.py:93
↓ 10 callersMethodSequential
(self, *args)
builder.py:48
↓ 9 callersMethodflatten
(self, in_features)
builder.py:119
↓ 8 callersMethodlog
(self, msg)
utils/engine.py:255
↓ 7 callersMethodConv2d
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_m
builder.py:24
↓ 7 callersMethodConv2dBN
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, padding_mode='zeros'
builder.py:54
↓ 7 callersMethodavg_pool2d
(self, in_features, kernel_size, stride, padding)
builder.py:116
↓ 6 callersFunctionget_baseconfig_by_epoch
(network_type, dataset_name, dataset_subset, global_batch_size, num_node, weight_decay, op
base_config.py:16
↓ 6 callersFunctionrc_pacesetter_idxes
(n)
constants.py:149
↓ 6 callersMethodrelu
(self, in_features)
builder.py:110
↓ 5 callersFunctioncreate_dataset
(dataset_name, subset, batch_size)
dataset.py:43
↓ 5 callersFunctionensure_dir
create directories if *path* does not exist
utils/pyt_utils.py:66
↓ 5 callersFunctionread_hdf5
(file_path)
utils/misc.py:33
↓ 4 callersMethodFlatten
(self)
builder.py:99
↓ 4 callersMethodResIdentity
(self, num_channels)
builder.py:87
↓ 4 callersMethod__init__
(self, block_counts, num_classes, builder:ConvBuilder, deps)
base_model/resnet.py:94
↓ 4 callersMethod_make_stage
(self, block, planes, num_blocks, stride)
base_model/resnet.py:165
↓ 4 callersFunction_resnet_bottlenck_first_internal_layer_idx_of_stage
(num_blocks)
constants.py:280
↓ 4 callersMethodsave_hdf5
(self, path)
utils/engine.py:100
↓ 4 callersFunctiontorch_accuracy
param output, target: should be torch Variable
utils/misc.py:66
↓ 3 callersMethodBNReLUConv2d
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, padding_mode='zeros'
builder.py:71
↓ 3 callersMethodBatchNorm2d
(self, num_features, eps=None, momentum=None, affine=None, track_running_stats=None)
builder.py:31
↓ 3 callersMethodDropout
(self, keep_prob)
builder.py:90
↓ 3 callersMethod__init__
(self, BasicLoss, weights:List[float])
utils/loss.py:65
↓ 3 callersMethod_build_stage
(self, stage_in_channels, stage_channels, num_blocks, stride)
base_model/resnet.py:109
↓ 3 callersMethod_build_wrn_stage
(self, num_blocks, stage_input_channels, stage_deps, downsample, use_dropout)
base_model/wrn.py:52
↓ 3 callersFunctionding_train
(cfg:BaseConfigByEpoch, net=None, train_dataloader=None, val_dataloader=None, show_variables=False, convbuilde
ding_train.py:75
↓ 3 callersFunctionget_mask_by_magnitude
(weights_path, nonzero_ratio)
gsm/gsm_prune_pipeline.py:9
↓ 3 callersFunctionget_model_fn
(dataset_name, model_name)
model_map.py:80
↓ 3 callersFunctionget_value_by_pattern
(pattern, line)
show_log.py:10
↓ 3 callersMethodload_checkpoint
(self, weights, just_weights=False)
utils/engine.py:178
↓ 3 callersMethodload_hdf5
(self, path)
utils/engine.py:120
↓ 3 callersFunctionrc_internal_layers
(n)
constants.py:155
↓ 3 callersFunctionrc_origin_deps_flattened
(n)
constants.py:126
↓ 3 callersMethodregister_state
(self, **kwargs)
utils/engine.py:88
↓ 3 callersFunctionresnet_bottleneck_pacesetter_indices
(res_n)
constants.py:330
↓ 3 callersMethodsave_checkpoint
(self, path)
utils/engine.py:140
↓ 3 callersMethodsetup_log
(self, name='train', log_dir=None, file_name=None)
utils/engine.py:67
↓ 3 callersMethodshow_variables
(self)
utils/engine.py:94
↓ 2 callersMethodIdentity
(self)
builder.py:84
↓ 2 callersMethod__iter__
(self)
dataset.py:30
↓ 2 callersFunctioncur_time
()
utils/misc.py:11
↓ 2 callersFunctiongeneral_test
(network_type, weights, builder=None)
ding_test.py:123
↓ 2 callersFunctionget_logger
(name='', save_dir=None, distributed_rank=0, filename="log.txt")
utils/logger.py:6
↓ 2 callersMethodget_lr
(self)
utils/lr_scheduler.py:79
↓ 2 callersFunctionget_world_size
()
utils/torch_utils.py:13
↓ 2 callersFunctionget_world_size
()
utils/comm.py:15
↓ 2 callersFunctiongsm_lottery_ticket
(choice, train_config:BaseConfigByEpoch, gsm_config, nonzero_ratio)
gsm/gsm_lottery_ticket.py:16
↓ 2 callersFunctiongsm_train
(cfg:BaseConfigByEpoch, net=None, train_dataloader=None, val_dataloader=None, show_variables=False, convbuilde
gsm/gsm_train.py:158
↓ 2 callersFunctionload_cuda_data
(data_loader, dataset_name)
dataset.py:11
↓ 2 callersFunctionload_cuda_data
(data_loader, dataset_name)
gsm/gsm_train.py:58
↓ 2 callersFunctionmask_out_weights
(initialized_weights, masked_weights, mask_dict)
gsm/gsm_prune_pipeline.py:25
↓ 2 callersMethodmax_pool2d
(self, in_features, kernel_size, stride, padding)
builder.py:113
↓ 2 callersFunctionnum_iters_per_epoch
(cfg)
dataset.py:126
↓ 2 callersFunctionnum_iters_per_epoch
(cfg)
gsm/gsm_train.py:137
↓ 2 callersFunctionnum_train_examples_per_epoch
(dataset_name)
dataset.py:116
↓ 2 callersFunctionreduce_loss_dict
Reduce the loss dictionary from all processes so that process with rank 0 has the averaged results. Returns a dict with the same fields as
utils/comm.py:49
↓ 2 callersFunctionresnet_bottleneck_internal_kernel_indices
(res_n)
constants.py:311
↓ 2 callersFunctionrun_eval
(ds_val, max_iters, net, criterion, discrip_str, dataset_name)
ding_test.py:23
↓ 2 callersMethodsave_and_link_checkpoint
(self, snapshot_dir)
utils/engine.py:208
↓ 2 callersFunctionsave_hdf5
(numpy_dict, file_path)
utils/misc.py:46
↓ 2 callersMethodset_value
(self, param, value)
utils/engine.py:116
↓ 2 callersMethodupdate_iteration
(self, iteration)
utils/engine.py:91
↓ 1 callersMethod__init__
( self, optimizer, final_lr, final_iters, warmup_factor=1.0 / 3,
utils/lr_scheduler.py:57
↓ 1 callersMethod__init__
(self, block_counts, num_classes, builder:ConvBuilder, deps, use_dropout)
base_model/wrn.py:35
↓ 1 callersMethod__init__
(self, builder:ConvBuilder, in_planes, out_planes, stride=1)
base_model/mobilenetv1.py:6
↓ 1 callersMethod__init__
(self, builder:ConvBuilder, deps)
base_model/lenet5.py:8
↓ 1 callersFunction_align_and_update_loaded_state_dicts
Strategy: suppose that the models that we will create will have prefixes appended to each of its keys, for example due to an extra level
utils/checkpoint.py:49
↓ 1 callersMethodadd
(self, time_diff)
utils/timer.py:28
↓ 1 callersFunctionall_gather
Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[data]
utils/comm.py:74
↓ 1 callersFunctionding_test
(cfg:BaseConfigByEpoch, net=None, val_dataloader=None, show_variables=False, convbuilder=None,
ding_test.py:65
↓ 1 callersFunctionextract_deps_from_weights_file
(file_path)
utils/misc.py:19
↓ 1 callersFunctionget_baseconfig_for_test
(network_type, dataset_subset, global_batch_size, init_weights, device='cuda', deps=None, se_reduce_scale=0)
base_config.py:42
↓ 1 callersFunctionget_criterion
(cfg)
ding_train.py:71
↓ 1 callersFunctionget_criterion
(cfg)
ding_test.py:62
↓ 1 callersFunctionget_criterion
(cfg)
gsm/gsm_train.py:134
↓ 1 callersFunctionget_dataset_name_by_model_name
(model_name)
model_map.py:83
↓ 1 callersFunctionget_lr_scheduler
(cfg, optimizer)
utils/lr_scheduler.py:104
↓ 1 callersFunctionget_lr_scheduler
(cfg, optimizer)
gsm/gsm_train.py:141
↓ 1 callersFunctionget_mask_by_gsm
(init_hdf5, gsm_config, nonzero_ratio)
gsm/gsm_lottery_ticket.py:9
↓ 1 callersFunctionget_optimizer
(cfg, model, no_l2_keywords, use_nesterov=False)
ding_train.py:68
↓ 1 callersFunctionget_optimizer
(cfg, model, no_l2_keywords, use_nesterov)
gsm/gsm_train.py:131
↓ 1 callersFunctionget_rank
()
utils/torch_utils.py:21
↓ 1 callersFunctionget_rank
()
utils/comm.py:23
↓ 1 callersFunctiongsm_lottery_ticket_lenet300
(compress_ratio)
gsm/gsm_lottery_ticket_lenet300.py:15
↓ 1 callersFunctiongsm_lottery_ticket_lenet5
(compress_ratio)
gsm/gsm_lottery_ticket_lenet5.py:15
↓ 1 callersFunctiongsm_prune_pipeline
(init_hdf5, base_train_config, gsm_config, nonzero_ratio)
gsm/gsm_prune_pipeline.py:36
↓ 1 callersFunctiongsm_rc56
()
gsm/gsm_rc56.py:4
↓ 1 callersFunctionis_main_process
()
utils/comm.py:31
↓ 1 callersFunctionload_model
(model, model_file, logger)
utils/checkpoint.py:6
↓ 1 callersFunctionlog_important
(message, log_file)
utils/misc.py:14
↓ 1 callersFunctionparse_top1_top5_loss_from_log_line
(log_line)
show_log.py:13
↓ 1 callersFunctionparse_torch_devices
Parse user's devices input string to standard format for Torch. e.g. [gpu0, gpu1, ...]
utils/pyt_utils.py:23
↓ 1 callersFunctionrc_convert_flattened_deps
(flattened)
constants.py:135
↓ 1 callersFunctionrc_pacesetter_dict
(n)
constants.py:170
↓ 1 callersMethodregister
(self, **kwargs)
utils/engine.py:31
next →1–100 of 233, ranked by callers