MCPcopy Create free account

hub / github.com/albanie/collaborative-experts / functions

Functions321 in github.com/albanie/collaborative-experts

Methodsanity_checks
(self)
data_loader/MSRVTT_dataset.py:168
Methodsanity_checks
(self)
data_loader/MSVD_dataset.py:144
Methodsanity_checks
(self)
data_loader/DiDeMo_dataset.py:102
Methodsanity_checks
(self)
data_loader/ActivityNet_dataset.py:114
Methodsave_dir
(self)
parse_config.py:184
Functionsave_image
Save a numpy image to the disk Parameters: image_numpy (numpy array) -- input numpy array image_path (str) -- the path o
utils/util.py:334
Methodset_device
(self, device: torch.device)
model/text.py:91
Methodstep
(self, closure=None)
utils/radam.py:90
Methodstep
(self, closure=None)
utils/radam.py:155
Methodstep
(self, closure=None)
utils/ranger.py:82
Methodstep_n
(self)
utils/cos_restart.py:50
Functiont2v_metrics
Compute retrieval metrics from a similiarity matrix. Args: sims (th.Tensor): N x M matrix of similarities between embeddings, where
model/metric.py:14
Functiontensor2im
Converts a Tensor array into a numpy image array. Parameters: input_image (tensor) -- the input image tensor array imtype (type)
utils/util.py:311
Methodtext2vec
(self, text: str)
model/text.py:96
Methodtext2vec
(self, text: str)
model/text.py:345
Methodtext2vec
(self, text: str)
model/text.py:415
Functionv2t_metrics
Compute retrieval metrics from a similiarity matrix. Args: sims (th.Tensor): N x M matrix of similarities between embeddings, where
model/metric.py:132
Methodvalue
Returns the model's average precision for each class Return: ap (FloatTensor): 1xK tensor, with avg precision for each class k
model/metric.py:499
Methodvalue
(self, k=-1)
model/metric.py:558
Methodvocab
(self)
model/text.py:285
Methodwrapper
(tag, data, *args, **kwargs)
logger/visualization.py:64
← previous301–321 of 321, ranked by callers