Functions — github.com/Sense-GVT/DeCLIP

CodeHub MCPcopy Create free account

hub / github.com/Sense-GVT/DeCLIP / functions

Functions606 in github.com/Sense-GVT/DeCLIP

⨍ Functions 606◇ Types & classes 121

(self, input, return_dict=False)

prototype/model/declip.py:196

(self, text, mask_type=None, return_dense=False)

prototype/model/text_encoder/text_transformer.py:183

(self, x: torch.Tensor)

prototype/model/text_encoder/base_transformer.py:13

(self, x: torch.Tensor)

prototype/model/text_encoder/base_transformer.py:25

(self, x: torch.Tensor)

prototype/model/text_encoder/base_transformer.py:50

(self, x: torch.Tensor)

prototype/model/text_encoder/base_transformer.py:75

Query memory bank for additional negative samples Args: output: The output of the model. labels:

prototype/model/utils/nnclr_modules/memory_bank.py:89

Returns nearest neighbour of output tensor from memory bank Args: output: The torch tensor for which you want the nearest neighbo

prototype/model/utils/nnclr_modules/nn_memory_bank.py:42

Query memory bank for additional negative samples Args: output: The output of the model. labels:

prototype/model/utils/nnclr_modules/memory_bank_cuda.py:94

(self, x: torch.Tensor, return_dense=False, return_feature=False)

prototype/model/image_encoder/visual_transformer.py:55

(self, x: torch.Tensor)

prototype/model/image_encoder/modified_resnet_modified.py:44

prototype/model/image_encoder/modified_resnet_modified.py:71

(self, x, return_dense=False, return_feature=False)

prototype/model/image_encoder/modified_resnet_modified.py:189

(self, x: torch.Tensor)

prototype/model/image_encoder/modified_resnet.py:44

prototype/model/image_encoder/modified_resnet.py:71

(self, x, return_dense=False)

prototype/model/image_encoder/modified_resnet.py:192

(self, x: torch.Tensor)

prototype/model/image_encoder/base_transformer.py:13

(self, x: torch.Tensor)

prototype/model/image_encoder/base_transformer.py:25

(self, x: torch.Tensor)

prototype/model/image_encoder/base_transformer.py:50

(self, x: torch.Tensor)

prototype/model/image_encoder/base_transformer.py:75

(self, input, label)

prototype/loss_functions/loss.py:13

(self, logits_per_image, logits_per_text)

prototype/loss_functions/loss.py:37

(self, p1, z1, p2, z2, minimize_loss=False,)

prototype/loss_functions/loss.py:70

(self, zis, zjs, norm=True, weights=1.0)

prototype/loss_functions/nt_xent_ConVIRT.py:23

(self, z_i, z_j)

prototype/loss_functions/nt_xent.py:28

(self, z_i, z_ib, z_j, z_jb, temperature=None)

prototype/loss_functions/nt_xent.py:62

Methodfrom_args

prototype/data/metrics/imagenet_evaluator.py:76

Methodfrom_args

prototype/data/metrics/base_evaluator.py:39

prototype/utils/misc.py:112

Functionget_divisible_by

(num, divisible_by, min_val=None)

prototype/utils/fbnet_helper.py:94

Functionget_local_rank

linklink/__init__.py:37

Methodget_logits

(dense_feat_1, selected_feat_2)

prototype/model/filip.py:96

Methodget_logits

(dense_feat_1, selected_feat_2)

prototype/model/defilip.py:251

Functionget_merged_dict

(base, *new_dicts)

prototype/utils/fbnet_helper.py:133

Functionget_norm_layer

prototype/utils/misc.py:139

Functionget_rank

linklink/__init__.py:22

Methodgrad_clip_after

prototype/solver/slip_solver.py:664

Methodgrad_clip_after

prototype/solver/clip_solver.py:534

Methodgrad_clip_after

prototype/solver/defilip_solver.py:693

Methodgrad_clip_after

prototype/solver/declip_solver.py:659

Methodgrad_clip_after

prototype/solver/filip_solver.py:673

Methodgrad_clip_before

prototype/solver/slip_solver.py:654

Methodgrad_clip_before

prototype/solver/clip_solver.py:524

Methodgrad_clip_before

prototype/solver/defilip_solver.py:683

Methodgrad_clip_before

prototype/solver/declip_solver.py:649

Methodgrad_clip_before

prototype/solver/filip_solver.py:663

prototype/utils/dist.py:72

Methodinference

Arguments: - res_file (:obj:`str`): filename of result

prototype/data/datasets/base_dataset.py:141

linklink/fp16.py:16

Functioninitialize

(backend='nccl')

linklink/__init__.py:42

Functionkestrel_loader

(img_bytes, filepath)

prototype/data/image_reader.py:22

Functionlink_dist

prototype/utils/dist.py:8

kwargs will override other arguments

prototype/utils/fbnet_helper.py:124

Functionmerge_unify_args

prototype/utils/fbnet_helper.py:57

Functionmix_criterion

(criterion, pred, y_a, y_b, lam)

prototype/utils/misc.py:497

Functionmixup_data

Returns mixed inputs, pairs of targets, and lambda

prototype/utils/misc.py:482

Functionmodified_resnet_R101

prototype/model/image_encoder/modified_resnet_modified.py:234

Functionmodified_resnet_R101

prototype/model/image_encoder/modified_resnet.py:230

Functionmodified_resnet_R50

prototype/model/image_encoder/modified_resnet_modified.py:220

Methodoptimizer

linklink/fp16.py:7

Methodparam_clip_after

prototype/solver/slip_solver.py:638

Methodparam_clip_after

prototype/solver/clip_solver.py:509

Methodparam_clip_after

prototype/solver/defilip_solver.py:667

Methodparam_clip_after

prototype/solver/declip_solver.py:634

Methodparam_clip_after

prototype/solver/filip_solver.py:647

Methodparam_clip_before

prototype/solver/slip_solver.py:628

Methodparam_clip_before

prototype/solver/clip_solver.py:500

Methodparam_clip_before

prototype/solver/defilip_solver.py:657

Methodparam_clip_before

prototype/solver/declip_solver.py:625

Methodparam_clip_before

prototype/solver/filip_solver.py:637

Functionpil_loader

(img_bytes, filepath)

prototype/data/image_reader.py:12

Functionregister_float_module

(param, cast_args=True)

linklink/fp16.py:11

Methodreset_pipeline

(self, pipeline)

prototype/data/nvidia_dali_dataloader.py:63

Methodrotate_with_fill

(img, magnitude)

prototype/data/auto_augmentation.py:232

Methodsave_feature

(self, module, input, output)

prototype/tools/inference.py:197

Methodsave_gradient

(self, module, grad_in, grad_out)

prototype/tools/inference.py:200

Functionsend_info

prototype/utils/user_analysis_helper.py:7

Methodset_cmp_key

prototype/data/metrics/base_evaluator.py:14

Methodset_epoch

prototype/data/sampler.py:53

Methodset_layers

(self, num_layers)

prototype/model/slip.py:81

Methodset_layers

(self, num_layers)

prototype/model/defilip.py:80

Methodset_layers

(self, num_layers)

prototype/model/declip.py:63

Methodsetup_envs

prototype/solver/base_solver.py:11

Functionslip_res50

Constructs a slip_res50 model.

prototype/model/slip.py:289

Functionslip_vitb32

Constructs a slip_vitb32 model.

prototype/model/slip.py:299

prototype/model/image_encoder/modified_resnet_modified.py:190

prototype/model/image_encoder/modified_resnet.py:193

(self, this_iter=None)

prototype/lr_scheduler/scheduler.py:33

Performs a single optimization step. Arguments: - closure (:obj:`callable`, optional): A closure that reevaluates the model and r

prototype/optimizer/lars.py:50

(self, closure=None)

prototype/optimizer/adam_clip.py:30

Performs a single optimization step. Arguments: - closure (:obj:`callable`, optional): A closure that reevaluates the model and r

prototype/optimizer/AdamW_SGD.py:120

Methodtext_modules

prototype/model/slip.py:218

Methodtext_modules

prototype/model/clip.py:75

Methodtext_modules

prototype/model/defilip.py:205

Methodtext_modules

prototype/model/declip.py:177

Methodtext_parameters

prototype/model/clip.py:64

(self, mode=True)

prototype/model/image_encoder/visual_transformer.py:40

prototype/solver/base_solver.py:20

Methodvisual_modules

prototype/model/slip.py:222

Methodvisual_modules

prototype/model/clip.py:91

← previous next →501–600 of 606, ranked by callers