MCPcopy Create free account

hub / github.com/ant-research/DreamLIP / functions

Functions432 in github.com/ant-research/DreamLIP

⨍ Functions 432◇ Types & classes 72
↓ 103 callersFunction_pcfg
(url='', hf_hub='', **kwargs)
open_clip/pretrained.py:23
↓ 47 callersFunction_n2p
(w, t=True)
open_clip/big_vision.py:17
↓ 12 callersFunctionis_master
(args, local=False)
training/distributed.py:20
↓ 11 callersMethodupdate
(self, val, n=1)
training/train.py:35
↓ 9 callersMethod__init__
( self, dim, num_heads=8, qkv_bias=True, scaled_co
open_clip/transformer.py:90
↓ 9 callersFunction_slpcfg
(url='', hf_hub='', **kwargs)
open_clip/pretrained.py:36
↓ 7 callersFunction_apcfg
(url='', hf_hub='', **kwargs)
open_clip/pretrained.py:49
↓ 6 callersMethodsplit_caption
(self, text)
training/data.py:202
↓ 6 callersMethodsplit_caption
(self, text)
training/data.py:348
↓ 6 callersMethodsplit_caption
(self, text)
training/data.py:453
↓ 5 callersFunction_ntuple
(n)
open_clip/utils.py:49
↓ 5 callersMethoddecode
(self, tokens)
open_clip/tokenizer.py:215
↓ 4 callersMethod__init__
( self, local_loss=False, gather_with_grad=False, cache_labels
open_clip/loss.py:104
↓ 4 callersFunction_clean_tag
(tag: str)
open_clip/pretrained.py:430
↓ 4 callersMethod_encode_image
(self, images, normalize: bool = True)
open_clip/coca_model.py:139
↓ 4 callersMethod_loss
(self, image_features, text_features, logit_scale, logit_bias=None, negative_only=False)
open_clip/loss.py:658
↓ 4 callersMethod_make_layer
(self, planes, blocks, stride=1)
open_clip/modified_resnet.py:132
↓ 4 callersFunctionbasic_clean
(text)
open_clip/tokenizer.py:65
↓ 4 callersFunctiondownload_pretrained_from_hf
( model_id: str, filename: str = 'open_clip_pytorch_model.bin', revision=None,
open_clip/pretrained.py:532
↓ 4 callersMethodencode_text
(self, text, normalize: bool = False)
open_clip/model.py:317
↓ 4 callersFunctionenhance_level_to_args
(MAX_LEVEL)
training/random_aug.py:300
↓ 4 callersFunctiongather_only_features
( features, local_loss=False, gather_with_grad=False, rank=0, world_si
open_clip/loss.py:18
↓ 4 callersFunctionget_autocast
(precision)
training/precision.py:5
↓ 4 callersFunctionmerge_preprocess_dict
Merge overlay key-value pairs on top of base preprocess cfg or dict. Input dicts are filtered based on PreprocessCfg fields.
open_clip/transform.py:50
↓ 3 callersFunction_build_text_tower
( embed_dim: int, text_cfg: CLIPTextCfg, quick_gelu: bool = False, cast_dtype:
open_clip/model.py:173
↓ 3 callersFunction_build_vision_tower
( embed_dim: int, vision_cfg: CLIPVisionCfg, quick_gelu: bool = False, cast_dt
open_clip/model.py:104
↓ 3 callersMethod_global_pool
(self, x: torch.Tensor)
open_clip/transformer.py:493
↓ 3 callersFunction_warmup_lr
(base_lr, warmup_length, step)
training/scheduler.py:9
↓ 3 callersFunctionassign_learning_rate
(optimizer, new_lr)
training/scheduler.py:4
↓ 3 callersFunctioncount_params
(model)
training/profiler.py:125
↓ 3 callersMethodencode_image
(self, image, normalize: bool = False)
open_clip/model.py:445
↓ 3 callersFunctionexpand_urls
(urls, weights=None)
training/data.py:538
↓ 3 callersMethodfetch_file
Shortcut to reader's `fetch_file()`.
training/data.py:212
↓ 3 callersFunctionfreeze_batch_norm_2d
Converts all `BatchNorm2d` and `SyncBatchNorm` layers of provided module into `FrozenBatchNorm2d`. If `module` is itself an instance of eithe
open_clip/utils.py:9
↓ 3 callersMethodget_ground_truth
(self, device, num_logits)
open_clip/loss.py:165
↓ 3 callersFunctionget_input_dtype
(precision: str)
open_clip/model.py:95
↓ 3 callersMethodget_logits
(self, image_features, text_features, logit_scale)
open_clip/loss.py:202
↓ 3 callersFunctionget_model_config
(model_name)
open_clip/factory.py:70
↓ 3 callersFunctionget_tokenizer
( model_name: str = '', context_length: Optional[int] = None, cache_dir: str = '',
open_clip/factory.py:84
↓ 3 callersFunctionimage_transform_v2
( cfg: PreprocessCfg, is_train: bool, aug_cfg: Optional[Union[Dict[str, Any], Augmenta
open_clip/transform.py:416
↓ 3 callersMethodlock
lock modules Args: unlocked_groups (int): leave last n layer groups unlocked (default: 0)
open_clip/timm_model.py:110
↓ 3 callersFunctionremote_sync
(local_dir, remote_dir, protocol)
training/file_utils.py:44
↓ 3 callersMethodreset
(self)
training/train.py:29
↓ 3 callersMethodset_grad_checkpointing
(self, enable=True)
open_clip/model.py:306
↓ 3 callersFunctiontext_global_pool
(x, text: Optional[torch.Tensor] = None, pool_type: str = 'argmax')
open_clip/transformer.py:551
↓ 3 callersFunctionunwrap_model
(model)
training/train.py:50
↓ 2 callersMethod__init__
(self, inplanes, planes, stride=1)
open_clip/modified_resnet.py:13
↓ 2 callersMethod__init__
(self, use_pooler_output=True)
open_clip/hf_model.py:67
↓ 2 callersMethod__init__
( self, embed_dim: int, vision_cfg: CLIPVisionCfg, text_cfg: C
open_clip/model.py:254
↓ 2 callersMethod__init__
( self, root: str, ann_file: str, transform: Optional[Callable] = None,
dataloaders/flickr30k.py:70
↓ 2 callersMethod_encode_text
(self, text, normalize: bool = True)
open_clip/coca_model.py:144
↓ 2 callersFunction_expand_token
(token, batch_size: int)
open_clip/transformer.py:287
↓ 2 callersFunction_get_hf_config
(model_id, cache_dir=None)
open_clip/factory.py:77
↓ 2 callersFunction_process_batch
(batch_classnames)
open_clip/zero_shot_classifier.py:52
↓ 2 callersFunction_read_reqs
(relpath)
setup.py:13
↓ 2 callersFunction_rescan_model_configs
()
open_clip/factory.py:33
↓ 2 callersFunctionbackward
(total_loss, scaler)
training/train.py:57
↓ 2 callersMethodbackward
(ctx, grad_output)
open_clip/loss.py:590
↓ 2 callersFunctionbroadcast_object
(args, obj, src=0)
training/distributed.py:122
↓ 2 callersFunctionbuild_model_from_openai_state_dict
( state_dict: dict, quick_gelu=True, cast_dtype=torch.float16, )
open_clip/model.py:558
↓ 2 callersFunctionbytes_to_unicode
Returns list of utf-8 byte and a corresponding list of unicode strings. The reversible bpe codes work on unicode strings. This means you
open_clip/tokenizer.py:31
↓ 2 callersFunctioncanonicalize_text
Returns canonicalized `text` (lowercase and punctuation removed). From: https://github.com/google-research/big_vision/blob/53f18caf27a9419231bbf0
open_clip/tokenizer.py:103
↓ 2 callersFunctionconvert_weights_to_lp
Convert applicable model parameters to low-precision (bf16 or fp16)
open_clip/model.py:506
↓ 2 callersFunctioncreate_model
( model_name: str, pretrained: Optional[str] = None, precision: str = 'fp32',
open_clip/factory.py:168
↓ 2 callersFunctioncreate_model_and_transforms
( model_name: str, pretrained: Optional[str] = None, precision: str = 'fp32',
open_clip/factory.py:378
↓ 2 callersMethoddist_loss
(self, teacher_logits, student_logits)
open_clip/loss.py:491
↓ 2 callersFunctiondownload_pretrained_from_url
( url: str, cache_dir: Union[str, None] = None, )
open_clip/pretrained.py:478
↓ 2 callersMethoddraw_numbers
(self, n, k=4)
training/data.py:220
↓ 2 callersMethoddraw_numbers
(self, n, k=4)
training/data.py:366
↓ 2 callersMethoddraw_numbers
(self, n, k=4)
training/data.py:471
↓ 2 callersMethodencode_image
(self, image, normalize: bool = False)
open_clip/model.py:310
↓ 2 callersMethodencode_text
(self, text, normalize: bool = False)
open_clip/model.py:452
↓ 2 callersFunctionevaluate
(model, data, epoch, args, tb_writer=None, tokenizer=None)
training/train.py:265
↓ 2 callersFunctionget_1d_sincos_pos_embed_from_grid
embed_dim: output dimension for each position pos: a list of positions to be encoded: size (M,) out: (M, D)
open_clip/pos_embed.py:49
↓ 2 callersFunctionget_cast_dtype
(precision: str)
open_clip/model.py:86
↓ 2 callersMethodget_cast_dtype
(self)
open_clip/transformer.py:313
↓ 2 callersFunctionget_clean_fn
(type: str)
open_clip/tokenizer.py:92
↓ 2 callersFunctionget_dataset_fn
(data_path, dataset_type)
training/data.py:1115
↓ 2 callersFunctionget_imagenet
(args, preprocess_fns, split)
training/data.py:582
↓ 2 callersFunctionget_pairs
Return set of symbol pairs in a word. Word is represented as tuple of symbols (symbols being variable-length strings).
open_clip/tokenizer.py:53
↓ 2 callersFunctionget_pretrained_cfg
(model: str, tag: str)
open_clip/pretrained.py:466
↓ 2 callersFunctionget_pretrained_url
(model: str, tag: str)
open_clip/pretrained.py:473
↓ 2 callersMethodget_value
(self)
training/data.py:521
↓ 2 callersFunctionhas_hf_hub
(necessary=False)
open_clip/pretrained.py:524
↓ 2 callersFunctionload_checkpoint
(model, checkpoint_path, strict=True)
open_clip/factory.py:145
↓ 2 callersFunctionmerge_preprocess_kwargs
(base: PreprocessCfg, **kwargs)
open_clip/transform.py:67
↓ 2 callersFunctionneighbour_exchange_with_grad
(from_rank, to_rank, tensor, group=None)
open_clip/loss.py:594
↓ 2 callersFunctionpytorch_worker_seed
get dataloader worker seed from pytorch
training/data.py:733
↓ 2 callersFunctionrandom_seed
(seed=42, rank=0)
main.py:44
↓ 2 callersFunctionread_txt
(txt_file, serialize=True)
training/data.py:283
↓ 2 callersFunctionrun
(model, classifier, dataloader, args)
training/zero_shot.py:17
↓ 2 callersMethodrun
(self, src)
training/data.py:766
↓ 2 callersMethodsave_pretrained
(self, dest)
open_clip/tokenizer.py:415
↓ 2 callersFunctionshear_level_to_args
(MAX_LEVEL, replace_value)
training/random_aug.py:306
↓ 2 callersFunctiontranslate_level_to_args
(translate_const, MAX_LEVEL, replace_value)
training/random_aug.py:316
↓ 2 callersFunctiontune_channel
(ch)
training/random_aug.py:108
↓ 2 callersFunctionwhitespace_clean
(text)
open_clip/tokenizer.py:71
↓ 2 callersFunctionworld_info_from_env
()
training/distributed.py:43
↓ 1 callersMethodSinkhorn
(self, K, u, v)
open_clip/loss.py:236
↓ 1 callersMethod__init__
(self, p=0.2)
open_clip/transform.py:272
next →1–100 of 432, ranked by callers