MCPcopy Create free account

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

Functions321 in github.com/albanie/collaborative-experts

↓ 104 callersMethoditems
(self)
parse_config.py:194
↓ 62 callersMethodget
(self, name, default)
parse_config.py:164
↓ 39 callersMethodupdate
(self, val, n=1)
model/metric.py:320
↓ 31 callersMethodkeys
(self)
parse_config.py:167
↓ 14 callersMethodadd
Log a new value to the meter Args: value: Next restult to include.
model/metric.py:340
↓ 14 callersMethodinit
Finds a function handle with the name given as 'type' in config, and returns the instance initialized with corresponding keyword args given as
parse_config.py:136
↓ 12 callersMethod__init__
( self, task, use_ce, text_dim, l2renorm,
model/model.py:87
↓ 10 callersFunctionmemory_summary
()
utils/util.py:119
↓ 9 callersMethodload_challenge_text_features
Load the text features and raw captions to be used in the challenge.
base/base_dataset.py:691
↓ 9 callersMethodvisual_feat_paths
Canonical path lookup for visual features
base/base_dataset.py:578
↓ 8 callersMethodcommon_feat_names
Produce a common collection of feature names shared amongst datasets.
base/base_dataset.py:669
↓ 8 callersMethodcommon_text_feat_paths
Load the text features and raw captions to be used in the challenge.
base/base_dataset.py:660
↓ 7 callersMethodsave
save the current content to the HMTL file
utils/html.py:116
↓ 5 callersMethod__init__
(self, dict_path, num_classes=512, gating=True, space_to_depth=True)
model/s3dg.py:251
↓ 5 callersMethodlog_metrics
(self, metric_store, metric_name, mode)
trainer/trainer.py:179
↓ 4 callersMethod__init__
( self, dim: int, mirror: str, embedding_name: str,
model/text.py:258
↓ 4 callersFunctiondataset_paths
( dataset: str )
misc/gen_readme.py:172
↓ 4 callersFunctionget_archive_name
(dataset: str, release: str, archive_type: str)
misc/sync_experts.py:24
↓ 4 callersMethodget_logger
(self, name, verbosity=2)
parse_config.py:170
↓ 4 callersMethodhas_missing_values
(self, x)
base/base_dataset.py:575
↓ 3 callersMethodadd_header
Insert a header to the HTML file Parameters: text (str) -- the header text
utils/html.py:42
↓ 3 callersFunctioncols2metrics
(cols, num_queries)
model/metric.py:288
↓ 3 callersFunctionctxt_mgr
Provide a context for managing temporary, cloned copies of retrieval sample tensors. The rationale here is that to use nan-checking in the mo
trainer/trainer.py:20
↓ 3 callersFunctionfetch_model
(url: str, weights_path: Path)
model/text.py:188
↓ 3 callersFunctionfill_template
(template_path, rules)
misc/generate_slurm_scripts.py:31
↓ 3 callersFunctiongenerate_predictions
Generate a zip file for a submission in the challenge format.
misc/cvpr2020_challenge/prepare_submission.py:70
↓ 3 callersMethodlog_assert
Use assertions that will be written to the logs. This is a recipe from: http://code.activestate.com/recipes/577074-logging-asserts/
base/base_dataset.py:609
↓ 3 callersFunctionlog_summary
Extract performace statistics from experiment log files. Args: logger (logger): reference to primary logging instance log_path (P
logger/log_parser.py:8
↓ 3 callersFunctionmodel_specs2path
(feat_aggregation: Dict, keep: set, tag: str = None)
misc/gen_readme.py:142
↓ 3 callersMethodset_step
(self, step, mode='train')
logger/visualization.py:44
↓ 3 callersMethodvalue
Get the value of the meter in the current state.
model/metric.py:347
↓ 2 callersMethod__init__
(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-8, weight_decay=0)
utils/radam.py:7
↓ 2 callersMethod__init__
(self, weight=None)
model/loss.py:79
↓ 2 callersMethod__setstate__
(self, state)
utils/radam.py:12
↓ 2 callersMethod_progress
(self, batch_idx)
trainer/trainer.py:324
↓ 2 callersMethod_save_checkpoint
Saving checkpoints :param epoch: current epoch number :param log: logging information of the epoch :param save_best: if True,
base/base_trainer.py:238
↓ 2 callersFunctionaggregation_script_path2job_name
(aggregation_script_path)
misc/generate_slurm_scripts.py:97
↓ 2 callersFunctioncompute_dims
( config, logger: logging.Logger = None, )
utils/util.py:200
↓ 2 callersFunctioncompute_trn_config
(config, logger=None)
utils/util.py:189
↓ 2 callersFunctiondataset_loader
( text_dropout: float, fuse_captions: bool, spatial_feats: bool, use_zeros_for
data_loader/data_loaders.py:23
↓ 2 callersFunctionensure_tensor
(x)
utils/util.py:291
↓ 2 callersFunctionevaluation
(config, logger=None, trainer=None)
test.py:107
↓ 2 callersFunctionexpert_tensor_storage
(experts, feat_aggregation)
utils/util.py:140
↓ 2 callersFunctionextract_embeddings
( text_embedding_config_path: Path, rel_dest_dir: Path, data_dir: Path, refresh: bool, val
misc/prepare_text_embeddings.py:112
↓ 2 callersFunctionfilter_cmd_args
(cmd_args: List[str], remove: List[str])
utils/util.py:26
↓ 2 callersFunctiongenerate_readme
( experiments: Dict[str, List[str]], root_url: str, readme_templates: List[Path], readme_dests
misc/gen_readme.py:436
↓ 2 callersFunctiongenerate_results_string
(target, exp_name, results, latexify, drop=None)
misc/gen_readme.py:416
↓ 2 callersFunctiongenerate_url
(root_url: str, target: str, exp_name: str, experiments: Dict, fnames: dict,
misc/gen_readme.py:89
↓ 2 callersFunctionjobn_name2agg_log_path
(exp_dir, job_name)
misc/generate_slurm_scripts.py:84
↓ 2 callersFunctionlaunch_and_monitor_cmd
(cmd: str)
misc/cvpr2020_challenge/train_baselines.py:23
↓ 2 callersFunctionmkdir
create a single empty directory if it didn't exist Parameters: path (str) -- a single directory path
utils/util.py:374
↓ 2 callersFunctionparse_paths_from_logs
(logs: List[str], queries: List[str],)
misc/cvpr2020_challenge/train_baselines.py:41
↓ 2 callersFunctionparse_results
( experiments: Dict[str, List[str]], save_dir: Path, refresh_summaries: bool,
misc/gen_readme.py:371
↓ 2 callersFunctionread_json
(fname)
utils/util.py:162
↓ 2 callersMethodrebuild_keymap
(self)
utils/datastructures.py:34
↓ 2 callersFunctionsent_feat
(text, feat_type)
sent_feat_demo.py:1
↓ 2 callersFunctionsharded_cross_view_inner_product
Compute a similarity matrix from sharded vectors. Args: embds1 (dict[str:th.Tensor]): the set of sub-embeddings that, when co
model/model.py:1032
↓ 2 callersFunctionsharded_single_view_inner_product
Compute a similarity matrix from sharded vectors. Args: embds (dict[str:th.Tensor]): the set of sub-embeddings that, when concatenated,
model/model.py:1154
↓ 2 callersMethodstep
(self, closure=None)
utils/radam.py:15
↓ 2 callersFunctionsummarise
(group_id, log_dir="data/saved/log", model_dir="data/saved/models")
misc/aggregate_logs_and_stats.py:14
↓ 2 callersMethodtrain
Full training logic. Responsible for iterating over epochs, early stopping, checkpointing and logging metrics.
base/base_trainer.py:77
↓ 2 callersFunctiontrain_baselines
( dest_dir: Path, challenge_config_dir: Path, datasets: List[str], slurm: bool
misc/cvpr2020_challenge/train_baselines.py:80
↓ 2 callersFunctionupdate_src_web_video_dir
Provide backwards compatible support for web directories Args: config: a configuration object containing experiment paths
utils/util.py:96
↓ 2 callersFunctionverbose
(epoch, metrics, mode, name="TEST")
trainer/trainer.py:11
↓ 2 callersMethodvisualize_ranking
(self, sims, epoch, meta, nested_metrics)
utils/visualizer.py:37
↓ 1 callersMethod__init__
(self)
model/metric.py:366
↓ 1 callersFunction_get_by_path
Access a nested object in tree by sequence of keys.
parse_config.py:219
↓ 1 callersFunction_get_opt_name
(flags)
parse_config.py:207
↓ 1 callersMethod_get_padding_shape
(self, filter_shape, stride)
model/s3dg.py:170
↓ 1 callersMethod_prepare_device
setup GPU device if available, move model into configured device
base/base_trainer.py:220
↓ 1 callersMethod_resume_checkpoint
Resume from saved checkpoints :param resume_path: Checkpoint path to be resumed
base/base_trainer.py:268
↓ 1 callersFunction_set_by_path
Set a value in a nested object in tree by sequence of keys.
parse_config.py:214
↓ 1 callersMethod_space_to_depth
3D space to depth trick for TPU optimization.
model/s3dg.py:308
↓ 1 callersMethod_split_text
(self, sentence)
model/s3dg.py:221
↓ 1 callersMethod_train_epoch
Training logic for an epoch :param epoch: Current epoch number
base/base_trainer.py:70
↓ 1 callersFunction_update_config
(config, options, args)
parse_config.py:199
↓ 1 callersMethod_valid_epoch
Validate model after an epoch of training and store results to disk. Args: epoch (int): the current epoch Returns:
trainer/trainer.py:186
↓ 1 callersMethod_words_to_ids
(self, x)
model/s3dg.py:235
↓ 1 callersMethod_words_to_token
(self, words)
model/s3dg.py:225
↓ 1 callersMethod_zero_pad_tensor_token
(self, tensor, size)
model/s3dg.py:214
↓ 1 callersMethodadd
Add a new observation Args: output (Tensor): NxK tensor that for each of the N examples indicates the probability
model/metric.py:376
↓ 1 callersMethodadd_images
add images to the HTML file Parameters: ims (str list) -- a list of image paths txts (str list) -- a list of image
utils/html.py:91
↓ 1 callersMethodadd_videos
add images to the HTML file Parameters: vids (str list) -- a list of image paths txts (str list) -- a list of imag
utils/html.py:51
↓ 1 callersMethodcheck
(self)
utils/util.py:301
↓ 1 callersFunctioncompress_predictions
We store the indices of the top-k predictions, rather than the full similarity matrix, to reduce storage requirements. NOTE: The similarity m
test.py:24
↓ 1 callersMethodcompute_moe_weights
(self, text, ind)
model/model.py:671
↓ 1 callersMethodconfigure_train_test_splits
Partition the datset into train/val/test splits. Args: split_name (str): the name of the split
base/base_dataset.py:288
↓ 1 callersMethodcosine
(self, base_lr)
utils/cos_restart.py:45
↓ 1 callersFunctiondataset_name2json_key
(dataset: str)
misc/cvpr2020_challenge/train_baselines.py:33
↓ 1 callersMethoddataset_paths
Generates a datastructure containing all the paths required to load features
base/base_dataset.py:32
↓ 1 callersMethoddisplay_current_results
Display current results on visdom; save current results to an HTML file. Parameters: visuals (OrderedDict) - - dictionary of imag
utils/visualizer.py:74
↓ 1 callersFunctiondrop_nans
Remove nans, which we expect to find at missing indices. Args: x (th.Tensor): features ind (th.Tensor): binary values denoting wh
model/model.py:54
↓ 1 callersFunctionevaluate_from_ckpts
( ckpt_list_path: Path, challenge_config_dir: Path, dest_dir: Path, device: in
misc/cvpr2020_challenge/test_baselines.py:45
↓ 1 callersFunctionfetch_from_server
( dataset: str, root_url: str, purge_tar_file: bool, release: str, ref
misc/sync_experts.py:166
↓ 1 callersFunctionformatted_summary
(dataset, exp_root, fname)
misc/find_latest_checkpoints.py:10
↓ 1 callersFunctiongen_dict_store
(keylist, dim)
utils/datastructures.py:77
↓ 1 callersFunctiongen_latex_version_of_table
( latex_table_dir: Path, content: List[str], table_name: str, branch_name: str
misc/gen_readme.py:35
↓ 1 callersFunctiongenerate_aggregation_script
(exp_dir, group_id, aggregation_template, aggregation_script_path)
misc/generate_slurm_scripts.py:88
↓ 1 callersFunctiongenerate_configs
(base_config, grid)
misc/generate_exps.py:12
↓ 1 callersFunctiongenerate_script
(template_path, slurm_script_dir, job_queue, exp_dir, monitor_script, constraints, depende
misc/generate_slurm_scripts.py:101
next →1–100 of 321, ranked by callers