Index

_

__init__() (rau.models.LSTM method) (rau.models.SimpleRNN method) (rau.models.SinusoidalPositionalEncodingCacher method) (rau.models.TransformerDecoderLayers method) (rau.models.TransformerDecoderLayers.State method) (rau.models.TransformerEncoderDecoder method) (rau.models.TransformerEncoderLayers method) (rau.models.UnidirectionalTransformerEncoderLayers method) (rau.models.UnidirectionalTransformerEncoderLayers.State method) (rau.tasks.common.DictScoreAccumulator method) (rau.tasks.common.MicroAveragedScoreAccumulator method) (rau.tasks.common.OutOfCUDAMemoryError method) (rau.tasks.common.TrainingLoop method) (rau.unidirectional.ComposedUnidirectional method) (rau.unidirectional.ComposedUnidirectional.State method) (rau.unidirectional.DropoutUnidirectional method) (rau.unidirectional.EmbeddingUnidirectional method) (rau.unidirectional.ForwardResult method) (rau.unidirectional.OutputUnidirectional method) (rau.unidirectional.PositionalUnidirectional.State method) (rau.unidirectional.ResidualUnidirectional method) (rau.unidirectional.ResidualUnidirectional.State method) (rau.unidirectional.SimpleLayerUnidirectional method) (rau.unidirectional.SimpleUnidirectional.State method) (rau.unidirectional.Unidirectional method) (rau.vocab.ToStringVocabulary method)

A

add_prepare_data_args() (in module rau.tasks.common) add_training_loop_arguments() (in module rau.tasks.common)

args (rau.unidirectional.SimpleUnidirectional.State attribute) as_composable() (rau.unidirectional.Unidirectional method)

B

batch_size() (rau.models.TransformerDecoderLayers.State method) (rau.models.UnidirectionalTransformerEncoderLayers.State method) (rau.unidirectional.ComposedUnidirectional.State method) (rau.unidirectional.PositionalUnidirectional.State method) (rau.unidirectional.ResidualUnidirectional.State method) (rau.unidirectional.SimpleUnidirectional.State method) (rau.unidirectional.Unidirectional.State method)

beam_search() (in module rau.generation) build_to_int_vocabulary() (in module rau.vocab) build_to_string_vocabulary() (in module rau.vocab)

C

catchall() (rau.vocab.ToIntVocabularyBuilder method) (rau.vocab.ToStringVocabularyBuilder method) (rau.vocab.VocabularyBuilder method) clear() (rau.models.SinusoidalPositionalEncodingCacher method)

ComposedUnidirectional (class in rau.unidirectional) ComposedUnidirectional.State (class in rau.unidirectional) content() (rau.vocab.ToIntVocabularyBuilder method) (rau.vocab.ToStringVocabularyBuilder method) (rau.vocab.VocabularyBuilder method)

D

decoder (rau.models.TransformerDecoderLayers.State attribute) detach() (rau.unidirectional.ComposedUnidirectional.State method) (rau.unidirectional.Unidirectional.State method)

DictScoreAccumulator (class in rau.tasks.common) DropoutUnidirectional (class in rau.unidirectional)

E

early_stopping_patience (rau.tasks.common.TrainingLoop attribute) EmbeddingUnidirectional (class in rau.unidirectional) encoder (rau.models.UnidirectionalTransformerEncoderLayers.State attribute) encoder_is_padding_mask (rau.models.TransformerDecoderLayers.State attribute) encoder_sequence (rau.models.TransformerDecoderLayers.State attribute)

evaluate() (in module rau.tasks.common) (rau.tasks.common.TrainingLoop method) evaluate_batch() (rau.tasks.common.TrainingLoop method) examples_per_checkpoint (rau.tasks.common.TrainingLoop attribute) extra_outputs (rau.unidirectional.ForwardResult attribute)

F

fastforward() (rau.models.TransformerDecoderLayers.State method) (rau.models.UnidirectionalTransformerEncoderLayers.State method) (rau.unidirectional.ComposedUnidirectional.State method) (rau.unidirectional.PositionalUnidirectional.State method) (rau.unidirectional.ResidualUnidirectional.State method) (rau.unidirectional.SimpleUnidirectional.State method) (rau.unidirectional.Unidirectional.State method) first_state (rau.unidirectional.ComposedUnidirectional.State attribute) forward() (rau.models.TransformerDecoderLayers method) (rau.models.TransformerDecoderLayers.State method) (rau.models.TransformerEncoderDecoder method) (rau.models.TransformerEncoderLayers method) (rau.models.UnidirectionalTransformerEncoderLayers method) (rau.models.UnidirectionalTransformerEncoderLayers.State method) (rau.unidirectional.ComposedUnidirectional method) (rau.unidirectional.ComposedUnidirectional.State method) (rau.unidirectional.PositionalUnidirectional.State method) (rau.unidirectional.ResidualUnidirectional method) (rau.unidirectional.ResidualUnidirectional.State method) (rau.unidirectional.SimpleUnidirectional.State method) (rau.unidirectional.Unidirectional method) (rau.unidirectional.Unidirectional.State method)

forward_at_position() (rau.unidirectional.PositionalUnidirectional method) forward_from_position() (rau.unidirectional.PositionalUnidirectional method) forward_sequence() (rau.unidirectional.SimpleLayerUnidirectional method) (rau.unidirectional.SimpleUnidirectional method) forward_single() (rau.unidirectional.SimpleLayerUnidirectional method) (rau.unidirectional.SimpleReshapingLayerUnidirectional method) (rau.unidirectional.SimpleUnidirectional method) ForwardResult (class in rau.unidirectional)

G

generate_batches() (rau.tasks.common.TrainingLoop method) get_encodings() (rau.models.SinusoidalPositionalEncodingCacher method) get_loss() (rau.tasks.common.TrainingLoop method) get_prepared_batch_and_loss() (rau.tasks.common.TrainingLoop method) get_prepared_batch_info() (rau.tasks.common.TrainingLoop method) get_random_generator_and_seed() (in module rau.tasks.common) get_random_seed() (in module rau.tasks.common) get_shared_embeddings() (in module rau.models) get_token_types() (in module rau.tasks.common) get_token_types_in_file() (in module rau.tasks.common)

get_training_loop_kwargs() (in module rau.tasks.common) get_transformer_decoder() (in module rau.models) get_transformer_encoder() (in module rau.models) get_transformer_encoder_decoder() (in module rau.models) get_unidirectional_transformer_encoder() (in module rau.models) get_validation_metric_mode() (rau.tasks.common.TrainingLoop method) get_validation_metric_name() (rau.tasks.common.TrainingLoop method) get_value() (rau.tasks.common.DictScoreAccumulator method) (rau.tasks.common.MicroAveragedScoreAccumulator method) gradient_clipping_threshold (rau.tasks.common.TrainingLoop attribute)

H

handle_out_of_cuda_memory() (rau.tasks.common.TrainingLoop method)

has_catchall() (rau.vocab.ToIntVocabulary method)

I

info (rau.tasks.common.OutOfCUDAMemoryError attribute) initial_decoder_state() (rau.models.TransformerEncoderDecoder method) initial_learning_rate (rau.tasks.common.TrainingLoop attribute) initial_output() (rau.unidirectional.SimpleUnidirectional method) initial_state() (rau.models.TransformerDecoderLayers method) (rau.models.UnidirectionalTransformerEncoderLayers method) (rau.unidirectional.ComposedUnidirectional method) (rau.unidirectional.PositionalUnidirectional method) (rau.unidirectional.ResidualUnidirectional method) (rau.unidirectional.SimpleUnidirectional method) (rau.unidirectional.Unidirectional method)

input_tensor (rau.unidirectional.PositionalUnidirectional.State attribute) (rau.unidirectional.ResidualUnidirectional.State attribute) (rau.unidirectional.SimpleUnidirectional.State attribute)

K

kwargs (rau.unidirectional.SimpleUnidirectional.State attribute)

L

label_smoothing_factor (rau.tasks.common.TrainingLoop attribute) learning_rate_decay_factor (rau.tasks.common.TrainingLoop attribute) learning_rate_patience (rau.tasks.common.TrainingLoop attribute)

load_prepared_data_file() (in module rau.tasks.common) log_failed_batch() (rau.tasks.common.TrainingLoop method) LSTM (class in rau.models)

M

main() (rau.unidirectional.Unidirectional method) max_epochs (rau.tasks.common.TrainingLoop attribute) max_tokens_per_batch (rau.tasks.common.TrainingLoop attribute) MicroAveragedScoreAccumulator (class in rau.tasks.common) module rau.generation rau.models rau.tasks.common rau.tools rau.tools.torch rau.training rau.unidirectional rau.vocab

N

next() (rau.models.TransformerDecoderLayers.State method) (rau.models.UnidirectionalTransformerEncoderLayers.State method) (rau.unidirectional.ComposedUnidirectional.State method) (rau.unidirectional.PositionalUnidirectional.State method) (rau.unidirectional.ResidualUnidirectional.State method) (rau.unidirectional.SimpleUnidirectional.State method) (rau.unidirectional.Unidirectional.State method)

O

optimizer (rau.tasks.common.TrainingLoop attribute) OutOfCUDAMemoryError output (rau.unidirectional.ForwardResult attribute) output() (rau.models.TransformerDecoderLayers.State method) (rau.models.UnidirectionalTransformerEncoderLayers.State method) (rau.unidirectional.ComposedUnidirectional.State method) (rau.unidirectional.PositionalUnidirectional.State method) (rau.unidirectional.ResidualUnidirectional.State method) (rau.unidirectional.SimpleUnidirectional.State method) (rau.unidirectional.Unidirectional.State method)

outputs() (rau.models.TransformerDecoderLayers.State method) (rau.models.UnidirectionalTransformerEncoderLayers.State method) (rau.unidirectional.ComposedUnidirectional.State method) (rau.unidirectional.PositionalUnidirectional.State method) (rau.unidirectional.ResidualUnidirectional.State method) (rau.unidirectional.SimpleUnidirectional.State method) (rau.unidirectional.Unidirectional.State method) OutputUnidirectional (class in rau.unidirectional)

P

pad_sequences() (in module rau.tasks.common) parent (rau.unidirectional.PositionalUnidirectional.State attribute) (rau.unidirectional.SimpleUnidirectional.State attribute) position (rau.unidirectional.PositionalUnidirectional.State attribute)

PositionalUnidirectional (class in rau.unidirectional) PositionalUnidirectional.State (class in rau.unidirectional) prepare_file() (in module rau.tasks.common) previous_inputs (rau.models.TransformerDecoderLayers.State attribute) (rau.models.UnidirectionalTransformerEncoderLayers.State attribute)

R

random_shuffling_seed (rau.tasks.common.TrainingLoop attribute) rau.generation module rau.models module rau.tasks.common module rau.tools module rau.tools.torch module rau.training module

rau.unidirectional module rau.vocab module reserved() (rau.vocab.ToIntVocabularyBuilder method) (rau.vocab.ToStringVocabularyBuilder method) (rau.vocab.VocabularyBuilder method) ResidualUnidirectional (class in rau.unidirectional) ResidualUnidirectional.State (class in rau.unidirectional) RNN_CLASS (rau.models.LSTM attribute) (rau.models.SimpleRNN attribute) run() (rau.tasks.common.TrainingLoop method) run_parameter_update() (rau.tasks.common.TrainingLoop method)

S

second_state (rau.unidirectional.ComposedUnidirectional.State attribute) set_allow_reallocation() (rau.models.SinusoidalPositionalEncodingCacher method) show_progress (rau.tasks.common.TrainingLoop attribute) SimpleLayerUnidirectional (class in rau.unidirectional) SimpleReshapingLayerUnidirectional (class in rau.unidirectional) SimpleRNN (class in rau.models) SimpleUnidirectional (class in rau.unidirectional) SimpleUnidirectional.State (class in rau.unidirectional)

SinusoidalPositionalEncodingCacher (class in rau.models) slice_batch() (rau.unidirectional.ComposedUnidirectional.State method) (rau.unidirectional.Unidirectional.State method) state (rau.unidirectional.ForwardResult attribute) states() (rau.models.TransformerDecoderLayers.State method) (rau.models.UnidirectionalTransformerEncoderLayers.State method) (rau.unidirectional.ResidualUnidirectional.State method) (rau.unidirectional.Unidirectional.State method)

T

tag() (rau.unidirectional.Unidirectional method) to_int() (rau.vocab.ToIntVocabulary method) to_string() (rau.vocab.ToStringVocabulary method) ToIntVocabulary (class in rau.vocab) ToIntVocabularyBuilder (class in rau.vocab) ToStringVocabulary (class in rau.vocab) ToStringVocabularyBuilder (class in rau.vocab) TrainingLoop (class in rau.tasks.common) transform_args() (rau.unidirectional.SimpleUnidirectional method) transform_kwargs() (rau.unidirectional.SimpleUnidirectional method)

transform_tensors() (rau.models.TransformerDecoderLayers.State method) (rau.models.UnidirectionalTransformerEncoderLayers.State method) (rau.unidirectional.ComposedUnidirectional.State method) (rau.unidirectional.PositionalUnidirectional.State method) (rau.unidirectional.ResidualUnidirectional.State method) (rau.unidirectional.SimpleUnidirectional.State method) (rau.unidirectional.Unidirectional.State method) TransformerDecoderLayers (class in rau.models) TransformerDecoderLayers.State (class in rau.models) TransformerEncoderDecoder (class in rau.models) TransformerEncoderLayers (class in rau.models)

U

Unidirectional (class in rau.unidirectional) Unidirectional.State (class in rau.unidirectional) UnidirectionalTransformerEncoderLayers (class in rau.models)

UnidirectionalTransformerEncoderLayers.State (class in rau.models) update() (rau.tasks.common.DictScoreAccumulator method) (rau.tasks.common.MicroAveragedScoreAccumulator method)

V

validate_prepare_data_args() (in module rau.tasks.common)

Vocabulary (class in rau.vocab) VocabularyBuilder (class in rau.vocab)

W

wrapped_state (rau.unidirectional.ResidualUnidirectional.State attribute)