Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/McGill-NLP/length-generalization
/ functions
Functions
942 in github.com/McGill-NLP/length-generalization
⨍
Functions
942
◇
Types & classes
118
↓ 1 callers
Method
extract_answer_from_final_output
(self, final_output: str)
src/data/data_instance_processor/data_instance_processor.py:716
↓ 1 callers
Method
extract_answer_from_final_output
(self, final_output: str)
src/data/data_instance_processor/addition_instance_processor.py:149
↓ 1 callers
Method
extract_answer_from_final_output
(self, final_output: str)
src/data/data_instance_processor/clutrr_instance_processor.py:80
↓ 1 callers
Method
extract_answer_from_final_output
(self, final_output: str)
src/data/data_instance_processor/poly_instance_processor.py:123
↓ 1 callers
Method
extract_answer_from_final_output
(self, final_output: str)
src/data/data_instance_processor/parity_instance_processor.py:108
↓ 1 callers
Method
extract_answer_from_final_output
(self, final_output: str)
src/data/data_instance_processor/sum_instance_processor.py:186
↓ 1 callers
Method
extract_answer_from_prediction
(self, prediction: str)
src/data/data_instance_processor/data_instance_processor.py:74
↓ 1 callers
Method
extract_answer_from_prediction
(self, prediction: str)
src/data/data_instance_processor/data_instance_processor.py:613
↓ 1 callers
Function
flatten_and_batch_shift_indices
This is a subroutine for [`batched_index_select`](./util.md#batched_index_select). The given `indices` of size `(batch_size, d_1, ..., d_n)`
src/modules/span_extractors/self_attentive_span_extractor.py:33
↓ 1 callers
Function
generate
( runtime, model, prompt, max_length=None, num_beams=1, process_output=None, io_ob
src/runtime/model_inference_shell.py:168
↓ 1 callers
Function
generate_all_boolean_combinations
(length: int)
scripts/generate_all_unified_scratchpad_configs.py:20
↓ 1 callers
Function
generate_all_scratchpad_configs
(ds_name, split_name)
src/tests/test_all_scratchpad_datasets_for_length.py:111
↓ 1 callers
Function
generate_all_scratchpad_configs
(ds_name, split_name)
src/tests/test_all_scratchpad_datasets.py:99
↓ 1 callers
Function
generate_all_scratchpad_configs
(ds_name, split_name)
scripts/upload_pos_enc_exp_for_scratchpad_ds.py:59
↓ 1 callers
Function
generate_all_scratchpad_configs
()
scripts/generate_all_unified_scratchpad_configs.py:31
↓ 1 callers
Function
generate_all_scratchpad_configs
(ds_name, split_name)
scripts/experiment_uploaders/scratchpad.py:62
↓ 1 callers
Function
generate_all_scratchpad_configs
(ds_name, split_name)
scripts/experiment_uploaders/attention_analysis_aggregate.py:62
↓ 1 callers
Function
generate_all_scratchpad_configs
(ds_name, split_name)
scripts/experiment_uploaders/classic_len_gen.py:55
↓ 1 callers
Function
generate_all_scratchpad_configs
(ds_name, split_name)
scripts/experiment_uploaders/attention_analysis.py:65
↓ 1 callers
Function
generate_all_scratchpad_configs
(ds_name, split_name)
scripts/experiment_uploaders/sanity_check.py:60
↓ 1 callers
Function
generate_all_scratchpad_configs
(ds_name, split_name)
scripts/experiment_uploaders/attention_kl_analysis.py:70
↓ 1 callers
Function
generate_boolean_configs
(config)
src/tests/test_all_scratchpad_datasets_for_length.py:97
↓ 1 callers
Function
generate_boolean_configs
(config)
scripts/upload_pos_enc_exp_for_scratchpad_ds.py:45
↓ 1 callers
Function
generate_boolean_configs
(config)
scripts/experiment_uploaders/scratchpad.py:48
↓ 1 callers
Function
generate_boolean_configs
(config)
scripts/experiment_uploaders/attention_analysis_aggregate.py:48
↓ 1 callers
Function
generate_boolean_configs
(config)
scripts/experiment_uploaders/final_attention_analysis.py:49
↓ 1 callers
Function
generate_boolean_configs
(config)
scripts/experiment_uploaders/final_attention_kl_analysis.py:58
↓ 1 callers
Function
generate_boolean_configs
(config)
scripts/experiment_uploaders/attention_analysis.py:51
↓ 1 callers
Function
generate_boolean_configs
(config)
scripts/experiment_uploaders/util.py:63
↓ 1 callers
Function
generate_boolean_configs
(config)
scripts/experiment_uploaders/attention_kl_analysis.py:56
↓ 1 callers
Function
generate_copy_half_task
(seq_length: int, num_examples: int)
scripts/dataset_builders/make_copy_half_dataset.py:73
↓ 1 callers
Function
generate_copy_task_data_copy_multi_char
( seq_length: int, num_examples: int )
scripts/dataset_builders/make_copy_dataset.py:137
↓ 1 callers
Function
generate_copy_task_data_copy_multi_char_2x
( seq_length: int, num_examples: int )
scripts/dataset_builders/make_copy_dataset.py:159
↓ 1 callers
Function
generate_copy_task_data_repeat_double_char
( seq_length: int, num_examples: int )
scripts/dataset_builders/make_copy_dataset.py:113
↓ 1 callers
Function
generate_copy_task_data_repeat_single_2x_char
( seq_length: int, num_examples: int )
scripts/dataset_builders/make_copy_dataset.py:93
↓ 1 callers
Function
generate_copy_task_data_repeat_single_char
( seq_length: int, num_examples: int )
scripts/dataset_builders/make_copy_dataset.py:73
↓ 1 callers
Function
generate_count_task_data_multi_char
( seq_length: int, num_examples: int, modular: bool = True, )
scripts/dataset_builders/make_count_dataset.py:95
↓ 1 callers
Function
generate_count_task_data_repeat_single_char
( seq_length: int, num_examples: int, modular: bool = True, )
scripts/dataset_builders/make_count_dataset.py:73
↓ 1 callers
Function
generate_data_for_bin
( a_num_digits: int, b_num_digits: int, num_examples: int )
scripts/dataset_builders/make_addition_dataset.py:35
↓ 1 callers
Function
generate_data_multi_digit
( seq_length: int, num_examples: int )
scripts/dataset_builders/make_sort_dataset.py:98
↓ 1 callers
Function
generate_data_single_digit
( seq_length: int, num_examples: int )
scripts/dataset_builders/make_sort_dataset.py:74
↓ 1 callers
Function
generate_deterministic_hp_run_id
(sweep_name: str, run_name: str)
src/common/py_utils.py:239
↓ 1 callers
Method
generate_dyck_one_sequence
Generates a random Dyck-1 sequence over the specified parenthesis pair. Args: parenthesis_pair: List containing an open parenthesi
scripts/dataset_builders/make_dyck_pred_dataset.py:81
↓ 1 callers
Function
generate_reverse_task_data_multi_char
( seq_length: int, num_examples: int )
scripts/dataset_builders/make_reverse_dataset.py:71
↓ 1 callers
Function
generate_reverse_task_data_multi_char_2x
( seq_length: int, num_examples: int )
scripts/dataset_builders/make_reverse_dataset.py:95
↓ 1 callers
Function
generate_reverse_task_data_multi_char_rev_cop
( seq_length: int, num_examples: int )
scripts/dataset_builders/make_reverse_dataset.py:118
↓ 1 callers
Method
generate_shuffle_sequence
Generates a random Shuffle-n sequence, where n indices the number of distinct parenthesis specified.
scripts/dataset_builders/make_dyck_pred_dataset.py:135
↓ 1 callers
Function
get_all_code_bundles_ids
()
scripts/wandb_download_all_bundles.py:18
↓ 1 callers
Function
get_cluster_name
(host: str)
scripts/download_nope_runs_to_local.py:96
↓ 1 callers
Function
get_compute_cluster
(host: str)
scripts/experiment_uploaders/attention_analysis_aggregate.py:165
↓ 1 callers
Function
get_compute_cluster
(host: str)
scripts/experiment_uploaders/final_attention_analysis.py:166
↓ 1 callers
Function
get_compute_cluster
(host: str)
scripts/experiment_uploaders/final_attention_kl_analysis.py:175
↓ 1 callers
Function
get_compute_cluster
(host: str)
scripts/experiment_uploaders/attention_analysis.py:168
↓ 1 callers
Function
get_compute_cluster
(host: str)
scripts/experiment_uploaders/util.py:44
↓ 1 callers
Function
get_compute_cluster
(host: str)
scripts/experiment_uploaders/attention_kl_analysis.py:173
↓ 1 callers
Method
get_compute_metric_fn
(self, stage: ExperimentStage)
src/data/base_dl_factory.py:122
↓ 1 callers
Method
get_compute_metric_fn_for_train
( self, )
src/data/sequence_classification_dl_factory.py:246
↓ 1 callers
Method
get_compute_metric_fn_for_train
( self, )
src/data/token_classification_dl_factory.py:243
↓ 1 callers
Method
get_compute_metric_fn_for_train
( self, )
src/data/s2s_dl_factory.py:578
↓ 1 callers
Method
get_compute_obj_fn
(self)
src/hp_search_space/base_search_space.py:27
↓ 1 callers
Function
get_config
(required_keys: List[str])
scripts/launcher.py:742
↓ 1 callers
Function
get_dataset_name
(cfg)
scripts/upload_result_csv.py:28
↓ 1 callers
Function
get_exp_name
(configs)
scripts/upload_experiment.py:455
↓ 1 callers
Function
get_file_name
(config: Dict[str, bool])
scripts/upload_pos_enc_exp_for_scratchpad_ds.py:112
↓ 1 callers
Function
get_file_name
(config: Dict[str, bool])
scripts/generate_all_unified_scratchpad_configs.py:56
↓ 1 callers
Function
get_file_name
(config: Dict[str, bool])
scripts/experiment_uploaders/scratchpad.py:127
↓ 1 callers
Function
get_file_name
(config: Dict[str, bool])
scripts/experiment_uploaders/attention_analysis.py:133
↓ 1 callers
Function
get_file_name
(config: Dict[str, bool])
scripts/experiment_uploaders/attention_kl_analysis.py:138
↓ 1 callers
Function
get_gpu_class
(length)
scripts/experiment_uploaders/scratchpad.py:210
↓ 1 callers
Function
get_gpu_class
(length)
scripts/experiment_uploaders/attention_analysis_all.py:99
↓ 1 callers
Function
get_gpu_class
(length)
scripts/experiment_uploaders/attention_analysis_aggregate.py:270
↓ 1 callers
Function
get_gpu_class
(length)
scripts/experiment_uploaders/classic_len_gen.py:116
↓ 1 callers
Function
get_gpu_class
(length)
scripts/experiment_uploaders/final_attention_analysis.py:271
↓ 1 callers
Function
get_gpu_class
(length)
scripts/experiment_uploaders/final_attention_kl_analysis.py:272
↓ 1 callers
Function
get_gpu_class
(length)
scripts/experiment_uploaders/attention_analysis.py:275
↓ 1 callers
Function
get_gpu_class
(length)
scripts/experiment_uploaders/sanity_check.py:121
↓ 1 callers
Function
get_gpu_class
(length)
scripts/experiment_uploaders/attention_kl_analysis.py:273
↓ 1 callers
Method
get_input_embeddings
(self)
src/models/custom_t5_decoder_only.py:1228
↓ 1 callers
Function
get_instance_processor_type
(ds_name: str, split_name: str)
src/tests/test_all_scratchpad_datasets_for_length.py:78
↓ 1 callers
Function
get_instance_processor_type
(ds_name: str, split_name: str)
src/tests/test_all_scratchpad_datasets.py:65
↓ 1 callers
Function
get_launcher_id
(run: Run)
src/common/wandb_utils.py:386
↓ 1 callers
Method
get_loaded_trainer
( self, load_best: bool = False, checkpoint_name: str = None )
src/runtime/seq2seq_runtime.py:1138
↓ 1 callers
Function
get_localhost
()
scripts/download_nope_runs_to_local.py:182
↓ 1 callers
Function
get_model_name
(cfg)
scripts/upload_result_csv.py:32
↓ 1 callers
Function
get_param
( run: wandb.apis.public.Run, params: List[Dict[str, Any]] )
scripts/fail_if_sweep_not_complete.py:49
↓ 1 callers
Function
get_pretty_pe_name
(pe_name: str)
src/common/wandb_utils.py:230
↓ 1 callers
Function
get_pretty_scratchpad_config_name
(config: str, ds: str, ds_split: str)
src/common/wandb_utils.py:158
↓ 1 callers
Function
get_pretty_scratchpad_config_name_for_lego
(config: str)
src/common/wandb_utils.py:145
↓ 1 callers
Function
get_pretty_scratchpad_config_name_for_sort
(config: str)
src/common/wandb_utils.py:132
↓ 1 callers
Function
get_result_name
(tags: List[str])
src/common/wandb_utils.py:382
↓ 1 callers
Function
get_scratchpad_config_name
(config: Dict[str, bool])
src/tests/test_all_scratchpad_datasets_for_length.py:248
↓ 1 callers
Method
get_search_space
(self, *args, **kwargs)
src/hp_search_space/base_search_space.py:24
↓ 1 callers
Function
get_seed
(cfg)
scripts/upload_result_csv.py:45
↓ 1 callers
Function
get_summary_key
(test_phase_shift: int, metric: str)
scripts/upload_result_csv.py:36
↓ 1 callers
Function
get_train_shift
(cfg)
scripts/upload_result_csv.py:24
↓ 1 callers
Function
get_unique_tokens
()
scripts/dataset_builders/make_copy_half_dataset.py:39
↓ 1 callers
Function
get_unique_tokens
()
scripts/dataset_builders/make_sort_dataset.py:37
↓ 1 callers
Function
get_wandb_api
()
src/common/wandb_utils.py:122
↓ 1 callers
Method
handle_tokenizer
(self, tokenizer: Optional[Tokenizer] = None)
src/models/gpt_neo.py:34
↓ 1 callers
Method
handle_tokenizer
(self, tokenizer: Optional[Tokenizer] = None)
src/models/gpt_neo.py:74
← previous
next →
301–400 of 942, ranked by callers