MCPcopy Create free account

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

Functions321 in github.com/albanie/collaborative-experts

↓ 1 callersFunctiongenerate_slurm_dependency_script
(group_ids, dependency_template, aggregation_scripts, generated_script_pa
misc/generate_slurm_scripts.py:69
↓ 1 callersFunctiongenerate_tar_lists
( save_dir: Path, experiments: Dict[str, Tuple[str, str]], datasets: List[str],
misc/gen_tar_lists.py:17
↓ 1 callersFunctiongenerate_tar_lists_for_challenge
( refresh: bool, datasets: List[str], challenge_phase: str, data_dir: Path, )
misc/gen_tar_lists.py:81
↓ 1 callersFunctionget_dataset_num_queries
(dataset: str, challenge_phase: str)
misc/cvpr2020_challenge/prepare_submission.py:21
↓ 1 callersMethodget_lr
(self)
utils/cos_restart.py:53
↓ 1 callersFunctionget_model_and_data_loaders
( config: ConfigParser, logger: logging.Logger, ckpt_path: Path, )
test.py:51
↓ 1 callersMethodget_retrieval_data
(self)
base/base_dataset.py:558
↓ 1 callersFunctionget_short_uuid
Return a 7 alpha-numeric character random string. We could use the full uuid() for better uniqueness properties, but it makes the filenames long
utils/util.py:41
↓ 1 callersFunctionhandle_moee_config
For the official ablations on MSRVTT, we provide MoEE with the same hyperparam budget as CE and run a search to find the best hyperparams. For th
utils/gen_ablations_for_dataset.py:13
↓ 1 callersFunctionjson_key2dataset_name
Convert json key used for a given datset into the name used in the codebase. Args: json_key: the key to be converted Returns:
misc/cvpr2020_challenge/test_baselines.py:28
↓ 1 callersFunctionkronecker_prod
(t1, t2)
model/model.py:48
↓ 1 callersMethodload_features
Load features from disk
base/base_dataset.py:44
↓ 1 callersFunctionload_w2v_model_from_cache
( w2v_weights: Path, )
model/text.py:176
↓ 1 callersFunctionmain
()
train.py:176
↓ 1 callersFunctionmain
()
utils/gen_ablations_for_dataset.py:43
↓ 1 callersFunctionmain
()
utils/datastructures.py:84
↓ 1 callersFunctionmain
()
misc/gen_tar_lists.py:168
↓ 1 callersFunctionmain
()
misc/gen_readme.py:632
↓ 1 callersFunctionmain
()
misc/generate_exps.py:54
↓ 1 callersFunctionmain
()
misc/generate_slurm_scripts.py:200
↓ 1 callersFunctionmain
()
misc/aggregate_logs_and_stats.py:65
↓ 1 callersFunctionmain
()
misc/find_latest_checkpoints.py:37
↓ 1 callersFunctionmain
()
misc/prepare_text_embeddings.py:240
↓ 1 callersFunctionmain
()
misc/sync_experts.py:201
↓ 1 callersFunctionmain
()
misc/cvpr2020_challenge/test_baselines.py:88
↓ 1 callersFunctionmain
()
misc/cvpr2020_challenge/train_baselines.py:160
↓ 1 callersFunctionmain
()
misc/cvpr2020_challenge/prepare_submission.py:107
↓ 1 callersFunctionparse_generate_readme
( experiments: Dict[str, List[str]], root_url: str, readme_templates: List[Path], readme_dests
misc/gen_readme.py:561
↓ 1 callersFunctionparse_geom_means_from_val_runs
TODO: Samuel - this is redundant due to log_summary() func in log_parser should refactor after deadline.
misc/gen_readme.py:246
↓ 1 callersFunctionparse_grid
Parse compact command line strings of the form: --key1 val_a|val_b --key2 val_c|val_d (here a vertical bar represents multiple values)
utils/util.py:51
↓ 1 callersFunctionparse_grid
(key_val_strs)
misc/generate_exps.py:36
↓ 1 callersFunctionparse_group_ids
(parsed_cmds)
misc/generate_slurm_scripts.py:58
↓ 1 callersFunctionparse_log
(log_path)
misc/gen_readme.py:289
↓ 1 callersFunctionpath2str
Recursively convert pathlib objects to strings to enable serialization
utils/util.py:167
↓ 1 callersFunctionprepare_embedding_model
( embedding_name: str, text_embedding_config: Dict[str, Dict[str, Union[str, int]]], )
misc/prepare_text_embeddings.py:45
↓ 1 callersFunctionprepare_text_with_yaspi
( yaspi_defaults: Dict[str, Union[str, int]], common_kwargs: Dict, datasets: List[str]
misc/prepare_text_embeddings.py:207
↓ 1 callersMethodpurge_stale_checkpoints
Remove checkpoints that are no longer neededself. NOTE: This function assumes that the `best` checkpoint has already been renamed to
base/base_trainer.py:193
↓ 1 callersMethodrandomise_feats
(self, experts, key)
model/model.py:241
↓ 1 callersFunctionremove_audio_streams
Prune audio-based features from the config and dest_path name (necessary for datasets like MSVD which do not possess sound.) If the audio feature
utils/gen_ablations_for_dataset.py:25
↓ 1 callersMethodreset
(self)
utils/util.py:307
↓ 1 callersMethodreset
(self)
model/metric.py:314
↓ 1 callersMethodreset
Resets the meter with empty member variables
model/metric.py:370
↓ 1 callersMethodreset
(self)
model/metric.py:527
↓ 1 callersMethodrestart
(self)
utils/cos_restart.py:41
↓ 1 callersMethodreturn_relationset
(self, num_frames, num_frames_relation)
model/model.py:419
↓ 1 callersMethodreturn_relationset
(self, num_frames, num_frames_relation)
model/model.py:483
↓ 1 callersFunctionrun_exp
(config)
train.py:32
↓ 1 callersMethodsanity_checks
( self, text: torch.Tensor, ind: Dict[str, torch.Tensor], expe
model/mil_nce_net.py:48
↓ 1 callersMethodsanity_checks
Catch any nans in the inputs/clusters
model/net_vlad.py:89
↓ 1 callersMethodsanity_checks
Run sanity checks on loaded data
base/base_dataset.py:38
↓ 1 callersMethodset_device
(self, device: torch.device)
model/text.py:59
↓ 1 callersMethodset_exper_name
(self, args, config)
parse_config.py:106
↓ 1 callersFunctionset_seeds
Set seeds for randomisation libraries. Args: seed: the seed value
utils/util.py:85
↓ 1 callersFunctionsetup_logging
Setup logging configuration.
logger/logger.py:8
↓ 1 callersFunctionsmall_font_str
(tokens)
misc/gen_readme.py:106
↓ 1 callersMethodsummary_stats
Report basic statistics about feature availability and variable lengths across the different subsets of the data.
base/base_dataset.py:630
↓ 1 callersFunctionsync_files
(experiments, save_dir, webserver, web_dir)
misc/gen_readme.py:111
↓ 1 callersMethodtext2vec
Convert a string of text into an embedding. Args: text: the content to be embedded Returns: (d x n) array, w
model/text.py:42
↓ 1 callersFunctionto_iterator
(obj_ids)
misc/prepare_text_embeddings.py:173
↓ 1 callersFunctiontrain_baseline_for_dataset
( challenge_config_dir: Path, mini_train: bool, train_single_epoch: bool, devi
misc/cvpr2020_challenge/train_baselines.py:58
↓ 1 callersFunctiontrain_baselines_with_yaspi
( yaspi_defaults_path: Path, common_kwargs: Dict, timestamp: str, datasets: Li
misc/cvpr2020_challenge/train_baselines.py:133
↓ 1 callersFunctionupload_models_to_robots
(web_dir: Path, experiments: Path, save_dir: Path, webserver: str)
misc/sync_experts.py:118
↓ 1 callersFunctionupload_to_server
( web_dir: Path, dataset: str, release: str, webserver: str, refresh: Dict[str, bool], )
misc/sync_experts.py:41
↓ 1 callersFunctionvalidate_embeddings_against_reference
( computed_embeddings: Dict[str, List[np.ndarray]], embedding_name: str, dataset: str, )
misc/prepare_text_embeddings.py:66
↓ 1 callersFunctionvalidate_predictions
( preds: np.ndarray, dataset: str, challenge_phase: str, topk: int = 10, )
misc/cvpr2020_challenge/prepare_submission.py:55
↓ 1 callersMethodvalue
Returns the model's average precision for each class Return: ap (FloatTensor): 1xK tensor, with avg precision for each class k
model/metric.py:442
↓ 1 callersFunctionwrite_json
(content, fname, paths2strs=False)
utils/util.py:176
Method__call__
(self, text: str)
model/text.py:143
Method__call__
(self, key)
model/text.py:228
Method__call__
(self, key)
model/text.py:281
Method__contains__
(self, name)
parse_config.py:161
Method__getattr__
If visualization is configured to use: return add_data() methods of tensorboard with additional information (step, ta
logger/visualization.py:53
Method__getitem__
(self, name)
parse_config.py:150
Method__getitem__
(self, key)
utils/datastructures.py:59
Method__getitem__
(self, key)
data_loader/data_loaders.py:220
Method__getitem__
(self, idx)
base/base_dataset.py:436
Method__init__
(self, args, options='', timestamp=True, slave_mode=False)
parse_config.py:20
Method__init__
(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-8, weight_decay=0)
utils/radam.py:82
Method__init__
(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-8, weight_decay=0, warmup = 0)
utils/radam.py:147
Method__init__
Initialize the HTML classes Parameters: web_dir (str) -- a directory that stores the webpage. HTML file will be creat
utils/html.py:16
Method__init__
Initialize the Visualizer class Create an HTML object for saveing HTML filters
utils/visualizer.py:19
Method__init__
(self)
utils/util.py:298
Method__init__
(self, keylist, dim, dtype=np.float16)
utils/datastructures.py:16
Method__init__
(self, params, lr=1e-3, alpha=0.5, k=6, N_sma_threshhold=5, betas=(.95,0.999), eps=1e-5, weight_decay=0)
utils/ranger.py:28
Method__init__
(self, optimizer, T_max, eta_min=0, last_epoch=-1, T_mult=1)
utils/cos_restart.py:32
Method__init__
( self, eval_only: bool, fuse_captions: bool, challenge_mode:
data_loader/data_loaders.py:105
Method__init__
(self, log_dir, logger, enabled)
logger/visualization.py:6
Method__init__
(self, name, fmt=':f')
model/metric.py:309
Method__init__
(self, topk=[1, 5, 10, 50], accuracy=True)
model/metric.py:521
Method__init__
(self, margin=1, fix_norm=True)
model/loss.py:31
Method__init__
(self, weight=None)
model/loss.py:69
Method__init__
( self, model: Callable, tokenizer: Union[Callable, None], dim
model/text.py:28
Method__init__
( self, model: Callable, dim: int, remove_stopwords: bool,
model/text.py:67
Method__init__
(self, vocab: Set[str])
model/text.py:121
Method__init__
( self, dim: int, mirror: str, embedding_name: str,
model/text.py:208
Method__init__
(self, w2v)
model/text.py:224
Method__init__
( self, word_dict_path: Path, weights_path: Path, embedding_na
model/text.py:316
Method__init__
(self, dim: int, embedding_name: str)
model/text.py:358
Method__init__
( self, input_dim, num_outputs_0_0a, num_outputs_1_0a, num_outputs_1_0
model/s3dg.py:36
Method__init__
(self, input_dim)
model/s3dg.py:88
← previousnext →101–200 of 321, ranked by callers