Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/McGill-NLP/length-generalization
/ types & classes
Types & classes
118 in github.com/McGill-NLP/length-generalization
⨍
Functions
942
◇
Types & classes
118
↓ 39 callers
Class
Params
Represents a parameter dictionary with a history, and contains other functionality around parameter passing and validation for AllenNLP.
src/common/params.py:164
↓ 21 callers
Class
ConfigurationError
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 callers
Class
Lazy
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 callers
Class
UnifedScratchpadStep
src/data/data_instance_processor/data_instance_processor.py:132
↓ 3 callers
Class
CustomT5Stack
src/models/custom_t5_decoder_only.py:646
↓ 1 callers
Class
CustomT5Attention
src/models/custom_t5_decoder_only.py:149
↓ 1 callers
Class
CustomT5Block
src/models/custom_t5_decoder_only.py:536
↓ 1 callers
Class
CustomT5LayerSelfAttention
src/models/custom_t5_decoder_only.py:526
↓ 1 callers
Class
CustomTrainingArguments
src/trainers/custom_training_args.py:29
↓ 1 callers
Class
CustomWandbCallback
src/runtime/seq2seq_runtime.py:63
↓ 1 callers
Class
DataCollatorForSeq2SeqInCausalLMInEvaluation
src/data/s2s_dl_factory.py:640
↓ 1 callers
Class
DataCollatorForSeq2SeqInCausalLMInTraining
src/data/s2s_dl_factory.py:715
↓ 1 callers
Class
DataCollatorForSeqClassificationInCausalLM
src/data/sequence_classification_dl_factory.py:274
↓ 1 callers
Class
DataCollatorForSeqClassificationInCausalLM
src/data/token_classification_dl_factory.py:271
↓ 1 callers
Class
DyckLanguage
scripts/dataset_builders/make_dyck_pred_dataset2.py:17
↓ 1 callers
Class
EntryPoint
src/main.py:42
↓ 1 callers
Class
FixedAbsolutePositionalEmbedding
src/models/custom_t5_decoder_only.py:110
↓ 1 callers
Class
FixedRotaryPositionalEmbedding
src/models/custom_t5_decoder_only.py:123
↓ 1 callers
Class
MemoryTracker
src/analyzers/runtime_analyzer.py:23
↓ 1 callers
Class
ModelInferenceShell
src/runtime/model_inference_shell.py:28
↓ 1 callers
Class
PositionalEmbedding
src/models/custom_t5_decoder_only.py:91
↓ 1 callers
Class
S2SAdditionDataInstanceProcessor
src/data/data_instance_processor/addition_instance_processor.py:14
↓ 1 callers
Class
S2SClutrrDataInstanceProcessor
src/data/data_instance_processor/clutrr_instance_processor.py:13
↓ 1 callers
Class
S2SCopyDataInstanceProcessor
src/data/data_instance_processor/copy_instance_processor.py:13
↓ 1 callers
Class
S2SLegoDataInstanceProcessor
src/data/data_instance_processor/lego_instance_processor.py:13
↓ 1 callers
Class
S2SPCFGBOSDataInstanceProcessor
src/data/data_instance_processor/pcfg_instance_processor.py:13
↓ 1 callers
Class
S2SParityDataInstanceProcessor
src/data/data_instance_processor/parity_instance_processor.py:13
↓ 1 callers
Class
S2SPolynomialDataInstanceProcessor
src/data/data_instance_processor/poly_instance_processor.py:14
↓ 1 callers
Class
S2SScanBOSDataInstanceProcessor
src/data/data_instance_processor/scan_instance_processor.py:41
↓ 1 callers
Class
S2SSortDataInstanceProcessor
src/data/data_instance_processor/sort_instance_processor.py:14
↓ 1 callers
Class
S2SSortMultiDigitDataInstanceProcessor
src/data/data_instance_processor/sort_instance_processor.py:152
↓ 1 callers
Class
S2SSumDataInstanceProcessor
src/data/data_instance_processor/sum_instance_processor.py:94
↓ 1 callers
Class
SearchSpace
src/hp_search_space/base_search_space.py:13
↓ 1 callers
Class
ShuffleLanguageTask
A simple Shuffle-n language task.
scripts/dataset_builders/make_dyck_pred_dataset.py:6
↓ 1 callers
Class
Tee
src/runtime/model_inference_shell.py:12
↓ 1 callers
Class
TimeDistributed
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 callers
Class
_DetHashPickler
src/common/from_params.py:94
↓ 1 callers
Class
scale_color_prism
Prism color scale Args: palette: The color palette name
src/common/plotnine_utils.py:72
↓ 1 callers
Class
scale_fill_prism
Prism fill scale Args: palette: The fill palette name
src/common/plotnine_utils.py:87
Class
Activation
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
Class
Analyzer
src/analyzers/analyzer.py:14
Class
AttentionAggregateAnalyzer
src/analyzers/attention_aggregate_analyzer.py:54
Class
AttentionAnalyzer
src/analyzers/attention_analyzer.py:16
Class
AttentionKLAnalyzer
src/analyzers/attention_kl_analyzer.py:61
Class
BaseTrainer
src/trainers/base_trainer.py:4
Class
Callback
src/callbacks/base_callback.py:8
Class
CartographyMeasureCallback
src/callbacks/cartography_measure_callback.py:27
Class
CausalGPT2
src/models/gpt2.py:18
Class
CausalGPTNeo
src/models/gpt_neo.py:22
Class
ComputeCanadaCluster
scripts/launcher.py:398
Class
ComputeFuncCreator
src/analyzers/runtime_analyzer.py:32
Class
ComputingCluster
scripts/launcher.py:54
Class
CountDataInstanceProcessor
src/data/data_instance_processor/count_instance_processor.py:13
Class
CountModularDataInstanceProcessor
src/data/data_instance_processor/count_instance_processor.py:35
Class
CustomDecoderOnlyT5
src/models/custom_t5_decoder_only.py:1118
Class
CustomDecoderOnlyT5ForSeqClassification
src/models/custom_t5_decoder_only.py:1372
Class
CustomDecoderOnlyT5ForTokenClassification
src/models/custom_t5_decoder_only.py:1604
Class
CustomDetHash
src/common/from_params.py:44
Class
CustomProgressCallback
src/runtime/seq2seq_runtime.py:129
Class
DIDataloader
src/data/base_dl_factory.py:15
Class
DIPreTrainedTokenizer
src/tokenization_utils/pre_trained.py:9
Class
DataInstanceProcessor
src/data/data_instance_processor/data_instance_processor.py:13
Class
DataInstanceProcessorWithUnifiedScratchpad
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
Class
DataLoaderFactory
src/data/base_dl_factory.py:19
Class
DecoderOnlyTrainer
src/trainers/decoder_only_trainer.py:22
Class
DetHashFromInitParams
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
Class
DetHashWithVersion
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
Class
DiGPT2Config
src/models/gpt2.py:13
Class
DiGPTNeoConfig
src/models/gpt_neo.py:17
Class
DiRobertaConfig
src/models/roberta.py:13
Class
DiT5Config
src/models/t5.py:16
Class
ExperimentStage
src/common/__init__.py:17
Class
FeedForward
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
Class
FromParams
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
Class
HFDefaultSearchSpace
src/hp_search_space/base_search_space.py:48
Class
HPSearchSpace
src/hp_search_space/base_search_space.py:18
Class
HfModelConfig
src/models/base_model.py:10
Class
IBMCluster
scripts/launcher.py:544
Class
IdentityActivation
src/modules/activations.py:47
Class
IdentityDataInstanceProcessor
src/data/data_instance_processor/data_instance_processor.py:48
Class
Main
scripts/manual_sweep.py:126
Class
ManualWhitespaceTokenizer
src/tokenization_utils/manual.py:27
Class
MetricsMixin
src/trainers/trainer_with_metrics.py:10
Class
MilaCluster
scripts/launcher.py:530
Class
Model
src/models/base_model.py:6
Class
NewLineFormatter
src/common/py_utils.py:13
Class
Registrable
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
Class
Runtime
src/runtime/base_runtime.py:4
Class
RuntimeAnalyzer
src/analyzers/runtime_analyzer.py:40
Class
S2SReverseDataInstanceProcessor
src/data/data_instance_processor/copy_instance_processor.py:34
Class
S2SScanDataInstanceProcessor
src/data/data_instance_processor/scan_instance_processor.py:13
Class
SavePredictionsCallback
src/callbacks/save_predictions_callback.py:22
Class
SelfAttentiveSpanExtractor
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
Class
Seq2SeqAnalyzer
src/analyzers/seq2seq_analyzer.py:20
Class
Seq2SeqDataLoaderFactory
src/data/s2s_dl_factory.py:34
Class
Seq2SeqRuntime
src/runtime/seq2seq_runtime.py:149
Class
Seq2SeqT5
src/models/t5.py:21
Class
Seq2SeqTrainerWithMetrics
src/trainers/trainer_with_metrics.py:74
Class
SeqClassifierGPT2
src/models/gpt2.py:49
Class
SeqClassifierGPTNeo
src/models/gpt_neo.py:53
next →
1–100 of 118, ranked by callers