Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/albanie/collaborative-experts
/ functions
Functions
321 in github.com/albanie/collaborative-experts
⨍
Functions
321
◇
Types & classes
64
Method
sanity_checks
(self)
data_loader/MSRVTT_dataset.py:168
Method
sanity_checks
(self)
data_loader/MSVD_dataset.py:144
Method
sanity_checks
(self)
data_loader/DiDeMo_dataset.py:102
Method
sanity_checks
(self)
data_loader/ActivityNet_dataset.py:114
Method
save_dir
(self)
parse_config.py:184
Function
save_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
Method
set_device
(self, device: torch.device)
model/text.py:91
Method
step
(self, closure=None)
utils/radam.py:90
Method
step
(self, closure=None)
utils/radam.py:155
Method
step
(self, closure=None)
utils/ranger.py:82
Method
step_n
(self)
utils/cos_restart.py:50
Function
t2v_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
Function
tensor2im
Converts a Tensor array into a numpy image array. Parameters: input_image (tensor) -- the input image tensor array imtype (type)
utils/util.py:311
Method
text2vec
(self, text: str)
model/text.py:96
Method
text2vec
(self, text: str)
model/text.py:345
Method
text2vec
(self, text: str)
model/text.py:415
Function
v2t_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
Method
value
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
Method
value
(self, k=-1)
model/metric.py:558
Method
vocab
(self)
model/text.py:285
Method
wrapper
(tag, data, *args, **kwargs)
logger/visualization.py:64
← previous
301–321 of 321, ranked by callers