Functions — github.com/ZhangYuanhan-AI/OmniBenchmark

hub / github.com/ZhangYuanhan-AI/OmniBenchmark / functions

Functions269 in github.com/ZhangYuanhan-AI/OmniBenchmark

⨍ Functions 269◇ Types & classes 49

↓ 1 callersFunctiontest_build

()

linear_probe/models/timm_beit.py:58

↓ 1 callersFunctiontest_build

()

linear_probe/models/timm_vit_modified.py:90

↓ 1 callersFunctiontest_build

()

linear_probe/models/timm_effnet.py:49

↓ 1 callersFunctiontest_build

()

linear_probe/models/timm_vit.py:53

↓ 1 callersFunctiontest_build

()

linear_probe/models/timm_swinTransformer.py:39

↓ 1 callersFunctiontest_build

()

linear_probe/models/timm_inceptionv4.py:41

↓ 1 callersMethodtokenize

(self, texts: Union[str, List[str]], context_length: int = 77)

linear_probe/models/text_transformer.py:148

↓ 1 callersFunctiontrain

(train_loader, model, criterion, optimizer, epoch, scaler, args)

ReCo/reco_imagenet.py:423

↓ 1 callersFunctiontrain_feat_split

(seg_num, train_feats)

linear_probe/main.py:285

↓ 1 callersFunctiontrain_labels_split

(seg_num, train_feats)

linear_probe/main.py:303

↓ 1 callersFunctiontransform

(x, y, matrix)

ReCo/randaugment.py:98

↓ 1 callersFunctionwhitespace_clean

(text)

linear_probe/models/simple_tokenizer.py:56

FunctionBNFunc

(*args, **kwargs)

linear_probe/models/modified_resnet.py:15

MethodBNFunc

(*args, **kwargs)

linear_probe/models/resnet.py:103

Method__call__

(self, img)

ReCo/randaugment.py:304

Method__call__

(self, img)

ReCo/randaugment.py:399

Method__call__

(self, img)

ReCo/randaugment.py:475

Method__call__

(self, x)

ReCo/moco/loader.py:12

Method__call__

(self, x)

ReCo/moco/loader.py:23

Method__call__

(self, x)

ReCo/moco/loader.py:35

Method__getitem__

Args: index (int): Index Returns: tuple: (sample, target) where target is class_index of the target class.

linear_probe/dataset.py:179

Method__getitem__

Args: index (int): Index Returns: tuple: (sample, target) where target is class_index of the target class.

linear_probe/dataset.py:266

Method__getitem__

Args: index (int): Index Returns: tuple: (sample, target) where target is class_index of the target class.

linear_probe/dataset.py:319

Method__init__

(self, length=0)

linear_probe/utils.py:63

Method__init__

(self)

linear_probe/dataset.py:32

Method__init__

(self, root, extensions, transform=None, target_transform=None)

linear_probe/dataset.py:130

Method__init__

(self, root, transform=None, target_transform=None)

linear_probe/dataset.py:242

Method__init__

(self, root, meta, transform=None)

linear_probe/dataset.py:250

Method__init__

(self, inplanes, planes, stride=1)

linear_probe/models/vit.py:71

Method__init__

(self, spacial_dim: int, embed_dim: int, num_heads: int, output_dim: int = None)

linear_probe/models/vit.py:115

Method__init__

(self, layers, output_dim, heads, input_resolution=224, width=64)

linear_probe/models/vit.py:159

Method__init__

(self, d_model: int, n_head: int, attn_mask: torch.Tensor = None)

linear_probe/models/vit.py:226

Method__init__

(self, width: int, layers: int, heads: int, attn_mask: torch.Tensor = None)

linear_probe/models/vit.py:250

Method__init__

(self, input_resolution: int, patch_size: int, width: int, layers: int, heads: int, output_dim: int, clip_pret

linear_probe/models/vit.py:261

Method__init__

(self, num_classes=1000, pretrain_path=None, enable_fc=False)

linear_probe/models/timm_rn101.py:21

Method__init__

(self, num_classes=1000, pretrain_path=None, enable_fc=False)

linear_probe/models/timm_mlp_mixer.py:21

Method__init__

(self, bpe_path: str = default_bpe())

linear_probe/models/simple_tokenizer.py:63

Method__init__

(self, spacial_dim: int, embed_dim: int, num_heads: int, output_dim: int = None)

linear_probe/models/modified_resnet.py:121

Method__init__

(self, layers, output_dim, heads, input_resolution=224, width=64, num_classes=1000, bn=None,

linear_probe/models/modified_resnet.py:165

Method__init__

(self, num_classes=1000, pretrain_path=None, enable_fc=False)

linear_probe/models/timm_beit.py:21

Method__init__

(self, d_model: int, n_head: int, attn_mask: torch.Tensor = None)

linear_probe/models/text_transformer.py:32

Method__init__

(self, width: int, layers: int, heads: int, attn_mask: torch.Tensor = None, checkpoint: bool = False)

linear_probe/models/text_transformer.py:56

Method__init__

(self, num_classes=1000, pretrain_path=None, enable_fc=False)

linear_probe/models/timm_vit_modified.py:55

Method__init__

(self, num_classes=1000, pretrain_path=None, enable_fc=False)

linear_probe/models/timm_effnet.py:21

Method__init__

(self, num_classes=119035, pretrain_path=None, enable_fc=False)

linear_probe/models/timm_vit.py:23

Method__init__

(self, num_classes=1000, pretrain_path=None, enable_fc=False)

linear_probe/models/timm_swinTransformer.py:11

Method__init__

(self, num_classes=1001, pretrain_path=None, enable_fc=False)

linear_probe/models/timm_inceptionv4.py:21

Method__init__

(self, inplanes, planes, stride=1, downsample=None)

linear_probe/models/resnet.py:28

Method__init__

(self, inplanes, planes, stride=1, downsample=None)

linear_probe/models/resnet.py:60

Method__init__

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

ReCo/reco_imagenet.py:532

Method__init__

(self, num_batches, meters, prefix="")

ReCo/reco_imagenet.py:555

Method__init__

(self, alpha=1.0, positive_realm_weight=1.0,beta=1.0, gamma=1.0, supt=1.0, temperature=1.0, base_temperature=N

ReCo/losses.py:106

Method__init__

(self, name, prob=0.5, magnitude=10, hparams=None)

ReCo/randaugment.py:286

Method__init__

(self, ops, num_layers=2, choice_weights=None)

ReCo/randaugment.py:394

Method__init__

(self, kernel_size)

ReCo/randaugment.py:456

Method__init__

(self, base_transform)

ReCo/moco/loader.py:9

Method__init__

(self, base_transform)

ReCo/moco/loader.py:20

Method__init__

(self, sigma=[.1, 2.])

ReCo/moco/loader.py:32

Method__init__

(self, num_classes=1000, feat_dim=2048)

ReCo/moco/builder.py:9

Method__init__

( self, inplanes: int, planes: int, stride: int = 1, downsample: Optio

ReCo/models/resnet.py:39

Method__init__

( self, inplanes: int, planes: int, stride: int = 1, downsample: Optio

ReCo/models/resnet.py:94

Method__repr__

(self)

linear_probe/dataset.py:206

Method__repr__

(self)

linear_probe/dataset.py:293

Method__str__

(self)

ReCo/reco_imagenet.py:549

Function_enhance_level_to_arg

(level, _hparams)

ReCo/randaugment.py:181

Function_get_clones

(module, N)

linear_probe/models/timm_vit_modified.py:23

Function_get_clones

(module, N)

linear_probe/models/timm_vit.py:18

Function_get_clones

(module, N)

linear_probe/models/timm_swinTransformer.py:6

Method_hook_k

(self, _, __, output)

ReCo/moco/builder.py:82

Method_hook_q

(self, _, __, output)

ReCo/moco/builder.py:88

Function_init_dist_pytorch

(backend, **kwargs)

linear_probe/main.py:82

Function_posterize_original_level_to_arg

(level, _hparams)

ReCo/randaugment.py:207

Function_posterize_research_level_to_arg

(level, _hparams)

ReCo/randaugment.py:213

Function_posterize_tpu_level_to_arg

(level, _hparams)

ReCo/randaugment.py:219

Function_rotate_level_to_arg

(level, _hparams)

ReCo/randaugment.py:174

Function_shear_level_to_arg

(level, _hparams)

ReCo/randaugment.py:186

Function_solarize_add_level_to_arg

(level, _hparams)

ReCo/randaugment.py:230

Function_solarize_level_to_arg

(level, _hparams)

ReCo/randaugment.py:225

Function_translate_abs_level_to_arg

(level, hparams)

ReCo/randaugment.py:193

Function_translate_rel_level_to_arg

(level, _hparams)

ReCo/randaugment.py:200

Functionauto_contrast

(img, **__)

ReCo/randaugment.py:112

Functionbrightness

(img, factor, **__)

ReCo/randaugment.py:161

Functionclip_text

(clip_pretrain_path=None, **kwargs)

linear_probe/models/text_transformer.py:248

Functionclipvitb16

Constructs a ResNet-152 model.

linear_probe/models/vit.py:433

Functionclipvitb32

Constructs a ResNet-152 model.

linear_probe/models/vit.py:449

Functionclosure

()

linear_probe/main.py:329

Functioncollect_env_info

()

linear_probe/utils.py:18

Functioncolor

(img, factor, **__)

ReCo/randaugment.py:157

Functioncontrast

(img, factor, **__)

ReCo/randaugment.py:153

Functiondownload_image

(url, label, timeout)

download_tool/download_image.py:18

Methoddtype

(self)

linear_probe/models/vit.py:393

Methoddtype

(self)

linear_probe/models/text_transformer.py:145

Functionequalize

(img, **__)

ReCo/randaugment.py:124

Methodforward

(self, x: torch.Tensor)

linear_probe/models/vit.py:98

Methodforward

(self, x)

linear_probe/models/vit.py:124

Methodforward

(self, x)

linear_probe/models/vit.py:193

Methodforward

(self, x: torch.Tensor)

linear_probe/models/vit.py:221

Methodforward

(self, x: torch.Tensor)

linear_probe/models/vit.py:243

Methodforward

(self, x: torch.Tensor)

linear_probe/models/vit.py:256

Methodforward

(self, x: torch.Tensor)

linear_probe/models/vit.py:280

← previous next →101–200 of 269, ranked by callers