Functions — github.com/CrossmodalGroup/LAPS

CodeHub MCPcopy Create free account

hub / github.com/CrossmodalGroup/LAPS / functions

Functions123 in github.com/CrossmodalGroup/LAPS

⨍ Functions 123◇ Types & classes 17

↓ 12 callersFunctionprint

(*args, **kwargs)

lib/utils.py:325

↓ 9 callersMethodmax

lib/utils.py:70

↓ 4 callersFunctioncosine_similarity

(x1, x2, dim=1, eps=1e-8)

↓ 4 callersFunctionfunc_attention

(query, context, smooth=4, eps=1e-8, detach=True)

↓ 4 callersFunctionshard_attn_scores

(model, img_embs, cap_embs, cap_lens, opt, gpu=False)

lib/evaluation.py:301

↓ 4 callersMethodupdate

(self, val, n=1)

lib/evaluation.py:29

↓ 3 callersFunctioncount_params

↓ 3 callersFunctionget_world_size

lib/utils.py:347

↓ 3 callersFunctioni2t

(npts, sims, return_ranks=False, mode='coco')

lib/evaluation.py:231

↓ 3 callersFunctionis_dist_avail_and_initialized

lib/utils.py:336

↓ 3 callersFunctiont2i

(npts, sims, return_ranks=False, mode='coco')

lib/evaluation.py:266

↓ 3 callersMethodupdate

(self, **kwargs)

lib/utils.py:91

↓ 2 callersMethod__init__

(self, embed_dim=512, sparse_ratio=0.6)

lib/cross_net.py:15

↓ 2 callersFunctionencode_data

(model, data_loader, log_step=10, logging=logger.info)

lib/evaluation.py:72

↓ 2 callersMethodforward_sim

(self, img_embs, cap_embs, cap_lens)

↓ 2 callersFunctionget_loader

(opt, data_path, split, tokenizer, batch_size=128, shuffle=True, num_workers=2

lib/image_caption.py:188

↓ 2 callersFunctionget_rank

lib/utils.py:355

↓ 2 callersFunctionget_sim

(images, captions)

↓ 2 callersFunctionis_main_process

lib/utils.py:362

↓ 2 callersMethodloss_forward

(self, sim_mat, pos_mask, neg_mask)

lib/loss.py:113

↓ 2 callersMethodmedian

lib/utils.py:56

↓ 2 callersMethodupdate

(self, value, n=1)

lib/utils.py:37

↓ 1 callersMethod__init__

(self, opt=None, margin=0.2, )

↓ 1 callersMethod__init__

lib/encoders.py:76

↓ 1 callersFunctionadd_path

lib/_init_paths.py:5

↓ 1 callersFunctionadjust_learning_rate

(opt, optimizer, epoch)

↓ 1 callersMethodbackward

lib/utils.py:431

↓ 1 callersFunctionbuild_transforms

(img_size=224, is_train=True)

lib/image_caption.py:16

↓ 1 callersFunctioncreate_optimizer

↓ 1 callersMethodforward_emb

(self, images, captions, lengths)

↓ 1 callersFunctionget_image_encoder

lib/encoders.py:21

↓ 1 callersFunctionget_text_encoder

lib/encoders.py:16

↓ 1 callersFunctionis_sqr

lib/cross_net.py:9

↓ 1 callersFunctionl2norm

(X, dim, eps=1e-8)

↓ 1 callersFunctionloss_select

(opt, loss_type='vse')

↓ 1 callersFunctionmain

↓ 1 callersFunctionmain

↓ 1 callersFunctionmask_xattn_one_text

(img_embs, cap_i_expand, img_mask=None, i2t=True, scan=True,)

lib/xttn.py:225

↓ 1 callersFunctionmatching_max_mean

(img_regions, cap_words, cap_len, i2t=False, scan=False, bi_norm=False)

lib/xttn.py:164

↓ 1 callersMethodmax_violation_off

↓ 1 callersMethodmax_violation_on

↓ 1 callersFunctionpos_neg_mask

↓ 1 callersFunctionprocess_caption_bert

(tokenizer, tokens, train=True, mask_rate=0.2, size_augment=True)

lib/image_caption.py:103

↓ 1 callersMethodreset

lib/evaluation.py:23

↓ 1 callersFunctionsave_checkpoint

(state, is_best, filename='checkpoint.pth', prefix='')

↓ 1 callersMethodset_max_violation

(self, max_violation=True)

↓ 1 callersFunctionsetup_for_distributed

This function disables printing when not in master process

lib/utils.py:318

↓ 1 callersMethodtb_log

(self, tb_logger, prefix='', step=None)

lib/evaluation.py:67

↓ 1 callersFunctiontrain

(opt, train_loader, model, model_without_ddp, optimizer, epoch)

↓ 1 callersFunctionvalidate

(opt, val_loader, model)

↓ 1 callersFunctionxattn_score_i2t

(images, captions, cap_lens, smooth=4)

lib/xttn.py:102

↓ 1 callersFunctionxattn_score_t2i

(images, captions, cap_lens, smooth=9.0)

↓ 1 callersFunctionxattn_score_two

(images, captions, cap_lens, smooth_t2i=9, smooth_i2t=4)

lib/xttn.py:132

Method__getattr__

lib/utils.py:98

Method__getitem__

lib/image_caption.py:80

(self, opt, margin=0.2, max_violation=False)

lib/encoders.py:28

lib/evaluation.py:20

lib/evaluation.py:49

(self, window_size=20, fmt=None)

lib/utils.py:29

(self, delimiter="\t")

lib/utils.py:87

(self, *args, **kwargs)

lib/utils.py:177

(self, dim=512, keeped_patches=64, dim_ratio=0.2)

lib/cross_net.py:62

(self, opt=None)

lib/cross_net.py:102

(self, opt, data_path, split, tokenizer, train)

lib/image_caption.py:40

lib/image_caption.py:99

lib/evaluation.py:39

lib/evaluation.py:59

lib/utils.py:77

lib/utils.py:106

Methodadd_meter

(self, name, meter)

lib/utils.py:126

Functionall_gather_with_grad

lib/utils.py:442

lib/utils.py:61

(ctx, grad_output)

lib/utils.py:471

(ctx, grad_output: torch.Tensor)

lib/utils.py:498

Functioncollate_fn_ours

lib/image_caption.py:162

Functioncompute_acc

(logits, label, reduction='mean')

lib/utils.py:182

Functioncompute_n_params

(model, return_str=True)

lib/utils.py:190

Functionconcat_all_gather

(tensor, keep_grad=True)

lib/utils.py:402

Functioncosine_lr_schedule

Decay the learning rate

lib/utils.py:280

Functionevalrank

(model_path, model=None, data_path=None, split='dev', fold5=False, save_path=None)

lib/evaluation.py:129

(self, im, s, img_ids=None, scores=None)

(self, im, s, img_ids, sim_mat=None)

lib/encoders.py:54

(self, x, lengths)

lib/encoders.py:93

lib/utils.py:425

lib/utils.py:460

(ctx, tensor: torch.Tensor, reduce_dtype: torch.dtype = torch.float32)

lib/utils.py:488

(self, tokens, attention_x, attention_y)

lib/cross_net.py:21

(self, x, keep_policy=None)

lib/cross_net.py:76

(self, img_embs, cap_embs, cap_lens)

lib/cross_net.py:128

(self, images, captions, lengths, img_ids=None, warmup_alpha=1.,)

Methodfreeze_backbone

lib/encoders.py:65

Methodfreeze_backbone

lib/encoders.py:112

Methodfreeze_backbone

Functiongather_result

lib/utils.py:509

Functionget_argument_parser

Functionget_padding_mask

(features, lengths)

Functionget_test_loader

(opt, data_path, tokenizer, batch_size, workers, split='test')

lib/image_caption.py:225

next →1–100 of 123, ranked by callers