Functions — github.com/DingXiaoH/RepVGG

CodeHub MCPcopy Create free account

hub / github.com/DingXiaoH/RepVGG / functions

Functions206 in github.com/DingXiaoH/RepVGG

⨍ Functions 206◇ Types & classes 26

↓ 20 callersMethodupdate

(self, val, n=1)

↓ 8 callersFunction_check_args_tf

train/randaug.py:35

↓ 8 callersFunctionunwrap_model

Remove the DistributedDataParallel wrapper if present.

↓ 5 callersFunctionloads_pyarrow

Args: buf: the output of `dumps`.

data/lmdb_dataset.py:21

↓ 5 callersFunctionlog_msg

(message, log_file)

↓ 5 callersMethodread

data/zipreader.py:79

↓ 4 callersMethod_make_stage

(self, planes, num_blocks, stride)

↓ 4 callersFunction_randomly_negate

With 50% prob, negate the value

train/randaug.py:168

↓ 4 callersFunctionaccuracy

Computes the accuracy over the k top predictions for the specified values of k

↓ 4 callersFunctionget_RepVGG_func_by_name

↓ 4 callersFunctionget_grad_norm

(parameters, norm_type=2)

↓ 4 callersMethodget_zipfile

data/zipreader.py:24

↓ 4 callersMethodsplit_zip_style_path

data/zipreader.py:32

↓ 3 callersMethod_build_aux_for_stage

repvggplus_custom_L2.py:201

↓ 3 callersMethod_build_aux_for_stage

repvggplus.py:203

↓ 3 callersMethod_fuse_bn_tensor

↓ 3 callersMethod_fuse_bn_tensor

repvggplus_custom_L2.py:102

↓ 3 callersMethod_fuse_bn_tensor

repvggplus.py:90

↓ 3 callersMethoddisplay

↓ 3 callersFunctiondumps_pyarrow

Serialize an object. Returns: Implementation-dependent bytes-like object

data/lmdb_dataset.py:102

↓ 3 callersFunctionload_checkpoint

(model, ckpt_path)

↓ 3 callersFunctionreduce_tensor

↓ 3 callersMethodswitch_to_deploy

↓ 3 callersFunctionvalidate

(config, data_loader, model)

↓ 2 callersMethod__init__

(self, num_blocks, num_classes, width_multiplier, override_groups_map=None,

repvggplus_custom_L2.py:170

↓ 2 callersMethod__init__

(self, num_blocks, num_classes, width_multiplier,

repvggplus.py:173

↓ 2 callersMethod_get_lr

train/lr_scheduler.py:82

↓ 2 callersFunctionbuild_dataset

(is_train, config)

data/build.py:78

↓ 2 callersFunctionconv_bn

(in_channels, out_channels, kernel_size, stride, padding, groups=1)

↓ 2 callersFunctionconv_bn

(in_channels, out_channels, kernel_size, stride, padding, groups=1)

repvggplus_custom_L2.py:21

↓ 2 callersFunctionconv_bn

(in_channels, out_channels, kernel_size, stride, padding, groups=1)

repvggplus.py:21

↓ 2 callersFunctioncreate_RepVGGplus_by_name

(name, deploy=False, use_checkpoint=False)

repvggplus.py:255

↓ 2 callersFunctionfolder2lmdb

(dpath, name="train", write_frequency=5000)

data/lmdb_dataset.py:111

↓ 2 callersFunctionpil_loader

data/cached_image_folder.py:171

↓ 2 callersFunctionsave_checkpoint

(config, epoch, model, max_accuracy, optimizer, lr_scheduler, logger, is_best=False, model_ema=None)

↓ 2 callersMethodset_epoch

data/samplers.py:20

↓ 2 callersMethodswitch_to_deploy

repvggplus.py:111

↓ 2 callersFunctionvalidate

(val_loader, model, criterion, args)

quantization/quant_qat_train.py:373

↓ 1 callersMethod__init__

(self, in_dim, reduction_dim, bins, BatchNorm)

example_pspnet.py:10

↓ 1 callersMethod__init__

(self, num_blocks, num_classes=1000, width_multiplier=None, override_groups_map=None, deploy=False, use_se=Fal

↓ 1 callersMethod__init__

(self, name, fmt=':f')

↓ 1 callersMethod__init__

(self, num_features)

tools/insert_bn.py:72

↓ 1 callersMethod__init__

(self, root, loader, extensions, ann_file='', img_prefix='', transform=None, target_transform=None,

data/cached_image_folder.py:85

↓ 1 callersMethod__len__

data/cached_image_folder.py:154

↓ 1 callersMethod_create_cutout_mask

Creates a zero mask used for cutout of shape `img_height` x `img_width`. Args: img_height: Height of image cutout mask will be appli

train/cutout.py:8

↓ 1 callersMethod_get_batch_fmtstr

(self, num_batches)

↓ 1 callersMethod_get_qconfig

quantization/repvgg_quantized.py:52

↓ 1 callersFunction_interpolation

train/randaug.py:27

↓ 1 callersMethod_pad_1x1_to_3x3_tensor

(self, kernel1x1)

↓ 1 callersMethod_pad_1x1_to_3x3_tensor

(self, kernel1x1)

repvggplus_custom_L2.py:96

↓ 1 callersMethod_pad_1x1_to_3x3_tensor

(self, kernel1x1)

repvggplus.py:84

↓ 1 callersFunctionaccimage_loader

data/cached_image_folder.py:184

↓ 1 callersFunctionauto_resume_helper

↓ 1 callersFunctionbuild_loader

data/build.py:22

↓ 1 callersFunctionbuild_optimizer

Build optimizer, set weight decay of normalization to 0 by default.

train/optimizer.py:11

↓ 1 callersFunctionbuild_scheduler

(config, optimizer, n_iter_per_epoch)

train/lr_scheduler.py:14

↓ 1 callersFunctionbuild_transform

(is_train, config)

data/build.py:117

↓ 1 callersFunctioncheck_keywords_in_name

(name, keywords=())

train/optimizer.py:66

↓ 1 callersFunctionconv_bn_relu

(in_channels, out_channels, kernel_size, stride, padding, groups=1)

repvggplus_custom_L2.py:13

↓ 1 callersFunctionconv_bn_relu

(in_channels, out_channels, kernel_size, stride, padding, groups=1)

repvggplus.py:13

↓ 1 callersFunctionconvert

tools/convert.py:20

↓ 1 callersFunctioncreate_RepVGG_B1

(deploy=False, use_checkpoint=False)

↓ 1 callersFunctioncreate_logger

(output_dir, dist_rank=0, name='')

train/logger.py:16

↓ 1 callersFunctiondirectly_insert_bn_without_init

tools/insert_bn.py:124

↓ 1 callersFunctionfind_classes

data/cached_image_folder.py:22

↓ 1 callersMethodfuse_model

quantization/repvgg_quantized.py:45

↓ 1 callersFunctionget_config

Get a yacs CfgNode object with default values.

train/config.py:210

↓ 1 callersFunctionget_default_train_trans

quantization/quant_qat_train.py:137

↓ 1 callersMethodget_equivalent_kernel_bias

↓ 1 callersMethodget_equivalent_kernel_bias

repvggplus_custom_L2.py:90

↓ 1 callersMethodget_equivalent_kernel_bias

repvggplus.py:78

↓ 1 callersMethodget_lr

↓ 1 callersFunctionhas_file_allowed_extension

Checks if a file is an allowed extension. Args: filename (string): path to a file Returns: bool: True if the filename ends wit

data/cached_image_folder.py:11

↓ 1 callersMethodinit_cache

data/cached_image_folder.py:116

↓ 1 callersFunctioninsert_bn

tools/insert_bn.py:145

↓ 1 callersFunctionis_zip_path

judge if this is a zip path

data/zipreader.py:11

↓ 1 callersFunctionload_weights

↓ 1 callersFunctionmain

↓ 1 callersFunctionmain

quantization/quant_qat_train.py:100

↓ 1 callersFunctionmain_worker

(gpu, ngpus_per_node, args)

quantization/quant_qat_train.py:169

↓ 1 callersFunctionmake_dataset

(dir, class_to_idx, extensions)

data/cached_image_folder.py:29

↓ 1 callersFunctionmake_dataset_with_ann

(ann_file, img_prefix, extensions)

data/cached_image_folder.py:47

↓ 1 callersFunctionparse_option

↓ 1 callersMethodprepare_quant

quantization/repvgg_quantized.py:55

↓ 1 callersFunctionrepvgg_model_convert

(model:torch.nn.Module, save_path=None, do_copy=True)

repvggplus.py:284

↓ 1 callersMethodreset

↓ 1 callersFunctionsave_checkpoint

(state, is_best, filename, best_filename)

quantization/quant_qat_train.py:416

↓ 1 callersFunctionsave_latest

(config, epoch, model, max_accuracy, optimizer, lr_scheduler, logger, model_ema=None)

↓ 1 callersFunctionset_weight_decay

(model, skip_list=(), skip_keywords=(), echo=False)

train/optimizer.py:41

↓ 1 callersFunctionsgd_optimizer

(model, lr, momentum, weight_decay)

quantization/quant_qat_train.py:84

↓ 1 callersFunctionswitch_bnstat_to_convbn

tools/insert_bn.py:96

↓ 1 callersFunctionswitch_repvggblock_to_bnstat

tools/insert_bn.py:78

↓ 1 callersMethodswitch_repvggplus_to_deploy

repvggplus.py:229

↓ 1 callersMethodswitch_to_deploy

repvggplus_custom_L2.py:123

↓ 1 callersFunctionthroughput

(data_loader, model, logger)

↓ 1 callersFunctiontrain

(train_loader, model, criterion, optimizer, epoch, args, lr_scheduler, is_main)

quantization/quant_qat_train.py:318

↓ 1 callersFunctiontrain_one_epoch

(config, model, criterion, data_loader, optimizer, epoch, mixup_fn, lr_scheduler, model_ema=None)

↓ 1 callersFunctiontransform

train/randaug.py:96

↓ 1 callersFunctionupdate_config

train/config.py:165

↓ 1 callersFunctionupdate_model_ema

Update exponential moving average (ema) of model weights.

next →1–100 of 206, ranked by callers