MCPcopy Create free account

hub / github.com/McGill-NLP/length-generalization / types & classes

Types & classes118 in github.com/McGill-NLP/length-generalization

↓ 39 callersClassParams
Represents a parameter dictionary with a history, and contains other functionality around parameter passing and validation for AllenNLP.
src/common/params.py:164
↓ 21 callersClassConfigurationError
The exception raised by any AllenNLP object when it's misconfigured (e.g. missing properties, invalid properties, unknown properties).
src/common/from_params.py:141
↓ 13 callersClassLazy
This class is for use when constructing objects using `FromParams`, when an argument to a constructor has a _sequential dependency_ with anot
src/common/lazy.py:10
↓ 10 callersClassUnifedScratchpadStep
src/data/data_instance_processor/data_instance_processor.py:132
↓ 3 callersClassCustomT5Stack
src/models/custom_t5_decoder_only.py:646
↓ 1 callersClassCustomT5Attention
src/models/custom_t5_decoder_only.py:149
↓ 1 callersClassCustomT5Block
src/models/custom_t5_decoder_only.py:536
↓ 1 callersClassCustomT5LayerSelfAttention
src/models/custom_t5_decoder_only.py:526
↓ 1 callersClassCustomTrainingArguments
src/trainers/custom_training_args.py:29
↓ 1 callersClassCustomWandbCallback
src/runtime/seq2seq_runtime.py:63
↓ 1 callersClassDataCollatorForSeq2SeqInCausalLMInEvaluation
src/data/s2s_dl_factory.py:640
↓ 1 callersClassDataCollatorForSeq2SeqInCausalLMInTraining
src/data/s2s_dl_factory.py:715
↓ 1 callersClassDataCollatorForSeqClassificationInCausalLM
src/data/sequence_classification_dl_factory.py:274
↓ 1 callersClassDataCollatorForSeqClassificationInCausalLM
src/data/token_classification_dl_factory.py:271
↓ 1 callersClassDyckLanguage
scripts/dataset_builders/make_dyck_pred_dataset2.py:17
↓ 1 callersClassEntryPoint
src/main.py:42
↓ 1 callersClassFixedAbsolutePositionalEmbedding
src/models/custom_t5_decoder_only.py:110
↓ 1 callersClassFixedRotaryPositionalEmbedding
src/models/custom_t5_decoder_only.py:123
↓ 1 callersClassMemoryTracker
src/analyzers/runtime_analyzer.py:23
↓ 1 callersClassModelInferenceShell
src/runtime/model_inference_shell.py:28
↓ 1 callersClassPositionalEmbedding
src/models/custom_t5_decoder_only.py:91
↓ 1 callersClassS2SAdditionDataInstanceProcessor
src/data/data_instance_processor/addition_instance_processor.py:14
↓ 1 callersClassS2SClutrrDataInstanceProcessor
src/data/data_instance_processor/clutrr_instance_processor.py:13
↓ 1 callersClassS2SCopyDataInstanceProcessor
src/data/data_instance_processor/copy_instance_processor.py:13
↓ 1 callersClassS2SLegoDataInstanceProcessor
src/data/data_instance_processor/lego_instance_processor.py:13
↓ 1 callersClassS2SPCFGBOSDataInstanceProcessor
src/data/data_instance_processor/pcfg_instance_processor.py:13
↓ 1 callersClassS2SParityDataInstanceProcessor
src/data/data_instance_processor/parity_instance_processor.py:13
↓ 1 callersClassS2SPolynomialDataInstanceProcessor
src/data/data_instance_processor/poly_instance_processor.py:14
↓ 1 callersClassS2SScanBOSDataInstanceProcessor
src/data/data_instance_processor/scan_instance_processor.py:41
↓ 1 callersClassS2SSortDataInstanceProcessor
src/data/data_instance_processor/sort_instance_processor.py:14
↓ 1 callersClassS2SSortMultiDigitDataInstanceProcessor
src/data/data_instance_processor/sort_instance_processor.py:152
↓ 1 callersClassS2SSumDataInstanceProcessor
src/data/data_instance_processor/sum_instance_processor.py:94
↓ 1 callersClassSearchSpace
src/hp_search_space/base_search_space.py:13
↓ 1 callersClassShuffleLanguageTask
A simple Shuffle-n language task.
scripts/dataset_builders/make_dyck_pred_dataset.py:6
↓ 1 callersClassTee
src/runtime/model_inference_shell.py:12
↓ 1 callersClassTimeDistributed
Given an input shaped like `(batch_size, time_steps, [rest])` and a `Module` that takes inputs like `(batch_size, [rest])`, `TimeDistributed`
src/modules/span_extractors/self_attentive_span_extractor.py:269
↓ 1 callersClass_DetHashPickler
src/common/from_params.py:94
↓ 1 callersClassscale_color_prism
Prism color scale Args: palette: The color palette name
src/common/plotnine_utils.py:72
↓ 1 callersClassscale_fill_prism
Prism fill scale Args: palette: The fill palette name
src/common/plotnine_utils.py:87
ClassActivation
Pytorch has a number of built-in activation functions. We group those here under a common type, just to make it easier to configure and inst
src/modules/activations.py:34
ClassAnalyzer
src/analyzers/analyzer.py:14
ClassAttentionAggregateAnalyzer
src/analyzers/attention_aggregate_analyzer.py:54
ClassAttentionAnalyzer
src/analyzers/attention_analyzer.py:16
ClassAttentionKLAnalyzer
src/analyzers/attention_kl_analyzer.py:61
ClassBaseTrainer
src/trainers/base_trainer.py:4
ClassCallback
src/callbacks/base_callback.py:8
ClassCartographyMeasureCallback
src/callbacks/cartography_measure_callback.py:27
ClassCausalGPT2
src/models/gpt2.py:18
ClassCausalGPTNeo
src/models/gpt_neo.py:22
ClassComputeCanadaCluster
scripts/launcher.py:398
ClassComputeFuncCreator
src/analyzers/runtime_analyzer.py:32
ClassComputingCluster
scripts/launcher.py:54
ClassCountDataInstanceProcessor
src/data/data_instance_processor/count_instance_processor.py:13
ClassCountModularDataInstanceProcessor
src/data/data_instance_processor/count_instance_processor.py:35
ClassCustomDecoderOnlyT5
src/models/custom_t5_decoder_only.py:1118
ClassCustomDecoderOnlyT5ForSeqClassification
src/models/custom_t5_decoder_only.py:1372
ClassCustomDecoderOnlyT5ForTokenClassification
src/models/custom_t5_decoder_only.py:1604
ClassCustomDetHash
src/common/from_params.py:44
ClassCustomProgressCallback
src/runtime/seq2seq_runtime.py:129
ClassDIDataloader
src/data/base_dl_factory.py:15
ClassDIPreTrainedTokenizer
src/tokenization_utils/pre_trained.py:9
ClassDataInstanceProcessor
src/data/data_instance_processor/data_instance_processor.py:13
ClassDataInstanceProcessorWithUnifiedScratchpad
This is the base class for all data instance processors that support a unified scratchpad. Unified scratchpad format looks like this: <sc
src/data/data_instance_processor/data_instance_processor.py:156
ClassDataLoaderFactory
src/data/base_dl_factory.py:19
ClassDecoderOnlyTrainer
src/trainers/decoder_only_trainer.py:22
ClassDetHashFromInitParams
Add this class as a mixin base class to make sure your class's det_hash is derived exclusively from the parameters passed to __init__().
src/common/from_params.py:58
ClassDetHashWithVersion
Add this class as a mixing base class to make sure your class's det_hash can be modified by altering a static `VERSION` member of your class.
src/common/from_params.py:79
ClassDiGPT2Config
src/models/gpt2.py:13
ClassDiGPTNeoConfig
src/models/gpt_neo.py:17
ClassDiRobertaConfig
src/models/roberta.py:13
ClassDiT5Config
src/models/t5.py:16
ClassExperimentStage
src/common/__init__.py:17
ClassFeedForward
This `Module` is a feed-forward neural network, just a sequence of `Linear` layers with activation functions in between. # Parameters
src/modules/feedforward.py:11
ClassFromParams
Mixin to give a from_params method to classes. We create a distinct base class for this because sometimes we want non-Registrable classes to
src/common/from_params.py:685
ClassHFDefaultSearchSpace
src/hp_search_space/base_search_space.py:48
ClassHPSearchSpace
src/hp_search_space/base_search_space.py:18
ClassHfModelConfig
src/models/base_model.py:10
ClassIBMCluster
scripts/launcher.py:544
ClassIdentityActivation
src/modules/activations.py:47
ClassIdentityDataInstanceProcessor
src/data/data_instance_processor/data_instance_processor.py:48
ClassMain
scripts/manual_sweep.py:126
ClassManualWhitespaceTokenizer
src/tokenization_utils/manual.py:27
ClassMetricsMixin
src/trainers/trainer_with_metrics.py:10
ClassMilaCluster
scripts/launcher.py:530
ClassModel
src/models/base_model.py:6
ClassNewLineFormatter
src/common/py_utils.py:13
ClassRegistrable
Any class that inherits from `Registrable` gains access to a named registry for its subclasses. To register them, just decorate them with the
src/common/registrable.py:34
ClassRuntime
src/runtime/base_runtime.py:4
ClassRuntimeAnalyzer
src/analyzers/runtime_analyzer.py:40
ClassS2SReverseDataInstanceProcessor
src/data/data_instance_processor/copy_instance_processor.py:34
ClassS2SScanDataInstanceProcessor
src/data/data_instance_processor/scan_instance_processor.py:13
ClassSavePredictionsCallback
src/callbacks/save_predictions_callback.py:22
ClassSelfAttentiveSpanExtractor
Computes span representations by generating an unnormalized attention score for each word in the document. Spans representations are computed
src/modules/span_extractors/self_attentive_span_extractor.py:376
ClassSeq2SeqAnalyzer
src/analyzers/seq2seq_analyzer.py:20
ClassSeq2SeqDataLoaderFactory
src/data/s2s_dl_factory.py:34
ClassSeq2SeqRuntime
src/runtime/seq2seq_runtime.py:149
ClassSeq2SeqT5
src/models/t5.py:21
ClassSeq2SeqTrainerWithMetrics
src/trainers/trainer_with_metrics.py:74
ClassSeqClassifierGPT2
src/models/gpt2.py:49
ClassSeqClassifierGPTNeo
src/models/gpt_neo.py:53
next →1–100 of 118, ranked by callers