Functions — github.com/cavalleria/cavaface

CodeHub MCPcopy Create free account

hub / github.com/cavalleria/cavaface / functions

Functions528 in github.com/cavalleria/cavaface

⨍ Functions 528◇ Types & classes 126↳ Endpoints 4

backbone/mobilenetv3.py:41

backbone/mobilenetv3.py:50

backbone/mobilenetv3.py:65

backbone/mobilenetv3.py:138

backbone/mobilenetv3.py:230

backbone/resattnet.py:78

backbone/resattnet.py:151

backbone/resattnet.py:207

backbone/resattnet.py:264

backbone/resattnet.py:342

backbone/rexnetv1.py:21

backbone/rexnetv1.py:70

backbone/rexnetv1.py:112

backbone/rexnetv1.py:207

backbone/common.py:57

backbone/common.py:66

backbone/common.py:76

backbone/common.py:94

backbone/common.py:157

backbone/common.py:213

backbone/common.py:226

backbone/common.py:254

backbone/common.py:283

backbone/common.py:309

backbone/common.py:389

backbone/common.py:429

backbone/common.py:484

backbone/common.py:606

backbone/common.py:623

backbone/common.py:691

backbone/common.py:890

backbone/common.py:941

backbone/common.py:1021

backbone/common.py:1043

backbone/common.py:1060

backbone/common.py:1087

backbone/common.py:1103

backbone/common.py:1120

backbone/efficientnet.py:98

backbone/efficientnet.py:189

backbone/efficientnet.py:240

backbone/efficientnet.py:378

backbone/mobilenetv2.py:92

backbone/mobilenetv2.py:202

backbone/mobilefacenet.py:29

backbone/mobilefacenet.py:53

backbone/mobilefacenet.py:76

backbone/mobilefacenet.py:118

backbone/mobilefacenet.py:161

backbone/mobilefacenet.py:174

backbone/mobilefacenet.py:197

backbone/mobilefacenet.py:283

backbone/mobilenext.py:128

backbone/mobilenext.py:244

backbone/resnet.py:45

backbone/resnet.py:79

backbone/resnet.py:160

backbone/proxylessnas.py:70

backbone/proxylessnas.py:128

backbone/proxylessnas.py:252

head/metrics.py:28

(self, embbedings, label)

head/metrics.py:66

(self, input, label)

head/metrics.py:118

(self, input, label)

head/metrics.py:160

(self, input, label=None)

head/metrics.py:202

(self, input, label)

head/metrics.py:261

(self, embbedings, label)

head/metrics.py:330

(self, embbedings, label)

head/metrics.py:373

(self, input, label)

head/metrics.py:436

(self, input, label)

head/metrics.py:507

(self, embbedings, label)

head/metrics.py:564

(self, embbedings, label)

head/metrics.py:605

(self, input, label)

head/metrics.py:636

head/metrics_parallel.py:61

(self, input, target)

loss/loss.py:13

(self, input, target)

loss/loss.py:26

(self, pred, target)

loss/loss.py:56

(self, input, target)

loss/loss.py:78

Methodget_sample_num_of_each_class

dataset/datasets.py:125

Methodget_tensor_stat

Get the stat of a single tensor Returns: - stat: a tuple containing (tensor_name, tensor_size, tensor_numel,

evaluation/utils/pytorch_memlab/mem_reporter.py:73

Functionget_val_dataset

(data_dir, val_set_name)

util/verification.py:259

Functionhard_sigmoid

(x, inplace: bool = False)

backbone/ghostnet.py:35

Functioninner_decorator

evaluation/utils/pytorch_memlab/line_profiler.py:159

Functionl2_norm

(input, axis=1)

backbone/common.py:230

Functionl2_norm

(input, axis=1)

util/utils.py:19

Functionlinear_hook

(self, input, output)

util/flops_counter.py:61

Functionload_ipython_extension

evaluation/utils/pytorch_memlab/extension.py:86

Methodload_state_dict

(self, state_dict)

optimizer/optimizer.py:57

Methodload_state_dict

Loads the schedulers state. Arguments: state_dict (dict): scheduler state. Should be an object returned from a ca

optimizer/lr_scheduler.py:81

Functionmain_worker

(gpu, ngpus_per_node, cfg)

Functionmerge_func

evaluation/evaluate_service.py:69

evaluation/eval_megaface.py:291

Execute a statement/cell under the PyTorch Memlab profiler to collect CUDA memory allocation information on a per-line basis.

evaluation/utils/pytorch_memlab/extension.py:50

Methodparse_results_into_file

evaluation/eval_megaface.py:463

Functionpooling_hook

(self, input, output)

util/flops_counter.py:84

Functionprint_pass

Functionprint_stats_atexit

evaluation/utils/pytorch_memlab/line_profiler.py:209

Functionprofile

Profile the CUDA memory usage of target function line by line The profiling results will be printed at exiting, KeyboardInterrupt raised. The

evaluation/utils/pytorch_memlab/line_profiler.py:178

Functionprofile_every

Profile the CUDA memory usage of target function line by line Prints the profiling output every `output_interval` execution of the target fun

evaluation/utils/pytorch_memlab/line_profiler.py:145

Functionproxylessnas

Create ProxylessNAS model with specific parameters. Parameters: ---------- version : str Version of ProxylessNAS ('cpu', 'gp

backbone/proxylessnas.py:258

← previous next →401–500 of 528, ranked by callers