MCPcopy Create free account

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

Functions321 in github.com/albanie/collaborative-experts

Method__init__
( self, input_dim, output_dim, kernel_size, stride=1, padding=0, separable=False )
model/s3dg.py:102
Method__init__
(self, kernel_size, stride=None, padding="SAME")
model/s3dg.py:162
Method__init__
( self, embd_dim, num_embeddings=66250, word_embedding_dim=300, token_
model/s3dg.py:195
Method__init__
( self, text_dim: int, expert_dims: Dict[str, Tuple[int, int]],
model/mil_nce_net.py:12
Method__init__
(self, cluster_size, feature_size, ghost_clusters=0, add_batch_norm=True)
model/net_vlad.py:26
Method__init__
(self, img_feature_dim, num_attention)
model/model.py:337
Method__init__
(self, img_feature_dim, num_frames, num_class)
model/model.py:367
Method__init__
(self, img_feature_dim, num_frames, num_class)
model/model.py:427
Method__init__
(self, expert_dims, text_dim, use_ce, verbose, l2renorm, num_classes, trn_config, trn_cat, us
model/model.py:490
Method__init__
(self, input_dimension, output_dimension, use_bn)
model/model.py:910
Method__init__
(self, input_dimension, output_dimension, use_bn)
model/model.py:924
Method__init__
(self, input_dimension, output_dimension)
model/model.py:935
Method__init__
(self, dimension, add_batch_norm=True)
model/model.py:950
Method__init__
(self, output_dimension)
model/model.py:965
Method__init__
(self, dimension)
model/model.py:976
Method__init__
(self, dimension, add_batch_norm=True)
model/model.py:987
Method__init__
(self, same_dim, num_inputs, non_lin)
model/model.py:1019
Method__init__
(self, model, loss, metrics, optimizer, config, mini_train, num_keep_ckpts, skip_tboard)
base/base_trainer.py:17
Method__init__
( self, data_dir: Path, fuse_captions: bool, spatial_feats: bo
base/base_dataset.py:50
Method__init__
(self, model, loss, metrics, optimizer, config, data_loaders, lr_scheduler, visualizer, disab
trainer/trainer.py:55
Method__len__
(self)
parse_config.py:153
Method__len__
(self)
utils/datastructures.py:62
Method__len__
(self)
base/base_dataset.py:433
Method__repr__
(self)
parse_config.py:191
Method__repr__
(self)
utils/datastructures.py:65
Method__setitem__
(self, name, value)
parse_config.py:158
Method__setitem__
(self, key, value)
utils/datastructures.py:25
Method__setstate__
(self, state)
utils/radam.py:87
Method__setstate__
(self, state)
utils/radam.py:152
Method__setstate__
(self, state)
utils/ranger.py:77
Method__str__
(self)
model/metric.py:326
Method__str__
Model prints with number of trainable parameters
base/base_model.py:19
Method_pad_top_bottom
(filter_dim, stride_val)
model/s3dg.py:171
Method_train_epoch
Training logic for an epoch :param epoch: Current training epoch. :return: A log that contains all information you want to s
trainer/trainer.py:79
Methodadd
(self, output, target)
model/metric.py:531
Methodcollate_data
(self, data)
base/base_dataset.py:316
Methodconfig
(self)
parse_config.py:180
Methoddataset_paths
(training_file=None)
data_loader/LSMDC_dataset.py:18
Methoddataset_paths
(training_file=-1)
data_loader/VaTeX_dataset.py:18
Methoddataset_paths
()
data_loader/YouCook2_dataset.py:16
Methoddataset_paths
(training_file=None)
data_loader/QuerYDSegments_dataset.py:16
Methoddataset_paths
(training_file=None)
data_loader/QuerYD_dataset.py:16
Methoddataset_paths
(training_file=None)
data_loader/MSRVTT_dataset.py:16
Methoddataset_paths
(training_file=None)
data_loader/MSVD_dataset.py:18
Methoddataset_paths
(training_file=None)
data_loader/DiDeMo_dataset.py:16
Methoddataset_paths
(training_file=None)
data_loader/ActivityNet_dataset.py:16
Functionextract_embeddings_for_video
( descriptions: List[str], key: str, model, )
misc/prepare_text_embeddings.py:93
Methodfilter_keys
(self, keys, tag, allow_mismatch="", exceptions=None)
utils/datastructures.py:38
Functionflatten_dict
(x, keysep="-")
utils/util.py:128
Methodforward
(self, x)
model/loss.py:37
Methodforward
(self, x, target)
model/loss.py:73
Methodforward
(self, x, target)
model/loss.py:83
Methodforward
Inception block
model/s3dg.py:69
Methodforward
Feature gating as used in S3D-G.
model/s3dg.py:92
Methodforward
(self, input)
model/s3dg.py:154
Methodforward
(self, inp)
model/s3dg.py:188
Methodforward
(self, x, device: th.device = None)
model/s3dg.py:239
Methodforward
Defines the S3DG base architecture.
model/s3dg.py:317
Methodforward
( self, text: torch.Tensor, ind: Dict[str, torch.Tensor], expe
model/mil_nce_net.py:25
Methodforward
Aggregates feature maps into a fixed size representation. In the following notation, B = batch_size, N = num_features, K = num_clusters, D =
model/net_vlad.py:44
Methodforward
Forward pass of the function.
model/model.py:41
Methodforward
(self, experts, ind, text=None, raw_captions=None, text_token_mask=None)
model/model.py:251
Methodforward
(self, input)
model/model.py:345
Methodforward
(self, input)
model/model.py:399
Methodforward
(self, input)
model/model.py:457
Methodforward
Compute joint embeddings and, if requested, a confusion matrix between video and text representations in the minibatch. Notation: B =
model/model.py:711
Methodforward
(self, x)
model/model.py:916
Methodforward
(self, x)
model/model.py:928
Methodforward
(self, x)
model/model.py:942
Methodforward
(self, x)
model/model.py:956
Methodforward
(self, x, mask)
model/model.py:969
Methodforward
(self, x)
model/model.py:981
Methodforward
(self, x, x1)
model/model.py:994
Methodforward
(self, x)
model/model.py:1025
Methodforward
Forward pass logic :return: Model output
base/base_model.py:11
Functiongenerate_tar_lists
(save_dir, experiments)
misc/gen_readme.py:200
Methodget_image_dir
Return the directory that stores images
utils/html.py:38
Functioninf_loop
wrapper function for endless data loader.
utils/util.py:183
Functioninit_weights
(m)
train.py:96
Methodload_features
(self)
data_loader/LSMDC_dataset.py:64
Methodload_features
(self)
data_loader/VaTeX_dataset.py:51
Methodload_features
(self)
data_loader/YouCook2_dataset.py:55
Methodload_features
(self)
data_loader/QuerYDSegments_dataset.py:45
Methodload_features
(self)
data_loader/QuerYD_dataset.py:45
Methodload_features
(self)
data_loader/MSRVTT_dataset.py:94
Methodload_features
(self)
data_loader/MSVD_dataset.py:64
Methodload_features
(self)
data_loader/DiDeMo_dataset.py:58
Methodload_features
(self)
data_loader/ActivityNet_dataset.py:62
Methodlog_dir
(self)
parse_config.py:188
Functionmean_average_precision
(sims, query_masks=None)
model/metric.py:301
Functionmkdirs
create empty directories if they don't exist Parameters: paths (str list) -- a list of directory paths
utils/util.py:361
Functionmultiprocessing_parsing
(exp_name: str, meta: list, save_dir: Path, refresh_summaries: bool, teachText: bool, pickle_files: st
misc/gen_readme.py:341
Functionprint_numpy
Print the mean, min, max, median, std, and size of a numpy array Parameters: val (bool) -- if print the values of the numpy array
utils/util.py:345
Methodreset
Resets the meter to default settings.
model/metric.py:336
Functionretrieval_as_classification
Compute classification metrics from a similiarity matrix.
model/metric.py:223
Methodsanity_checks
(self)
data_loader/LSMDC_dataset.py:108
Methodsanity_checks
(self)
data_loader/VaTeX_dataset.py:95
Methodsanity_checks
(self)
data_loader/YouCook2_dataset.py:94
Methodsanity_checks
(self)
data_loader/QuerYDSegments_dataset.py:85
Methodsanity_checks
(self)
data_loader/QuerYD_dataset.py:89
← previousnext →201–300 of 321, ranked by callers