Index

A

active_mask_avg (sparse_caption.pruning.prune.PruningMixin property) active_mask_sparsities (sparse_caption.pruning.prune.PruningMixin property) active_pruned_weights() (sparse_caption.pruning.prune.PruningMixin method) active_pruning_masks() (sparse_caption.pruning.prune.PruningMixin method) add_argparse_args() (sparse_caption.data.collate.AttCollate static method) (sparse_caption.data.collate.ObjectRelationCollate static method) (sparse_caption.data.collate.UpDownCollate static method) (sparse_caption.data.karpathy.KarpathyDataset static method) (sparse_caption.data.mscoco.MscocoDataset static method) (sparse_caption.models.att_model.UpDownModel static method) (sparse_caption.models.att_model_prune.UpDownModel static method) (sparse_caption.models.relation_transformer.RelationTransformerModel static method) (sparse_caption.models.relation_transformer_prune.RelationTransformerModel static method) (sparse_caption.models.transformer.CachedTransformerBase static method) (sparse_caption.models.transformer.Transformer static method) (sparse_caption.pruning.prune.PruningMixin static method) (sparse_caption.tokenizer.RadixTokenizer static method) (sparse_caption.tokenizer.SentencePieceUnigramTokenizer static method) (sparse_caption.utils.training.TrainingModule static method)

all_mask_avg (sparse_caption.pruning.prune.PruningMixin property) all_mask_sparsities (sparse_caption.pruning.prune.PruningMixin property) ALL_METRICS (sparse_caption.utils.training.TrainingModule attribute) all_pruned_weights() (sparse_caption.pruning.prune.PruningMixin method) all_pruning_masks() (sparse_caption.pruning.prune.PruningMixin method) all_weight_sparsities (sparse_caption.pruning.prune.PruningMixin property) all_weights() (sparse_caption.pruning.prune.PruningMixin method) ANNOTATION_FILE (sparse_caption.data.karpathy.KarpathyDataset attribute) (sparse_caption.data.mscoco.MscocoDataset attribute) assert_in_kwargs() (sparse_caption.pruning.masked_layer.MaskMixin static method) atoi() (in module sparse_caption.utils.natural_sort) AttCollate (class in sparse_caption.data.collate) Attention (class in sparse_caption.models.att_model) (class in sparse_caption.models.att_model_prune) attention() (sparse_caption.models.transformer.MultiHeadedAttention static method) AttModel (class in sparse_caption.models.att_model) (class in sparse_caption.models.att_model_prune)

B

backward() (sparse_caption.pruning.sampler.BernoulliSample static method) (sparse_caption.pruning.sampler.Round static method) base_to_decimal() (sparse_caption.tokenizer.RadixTokenizer static method) BasicCollate (class in sparse_caption.data.collate) batch_beam_search() (sparse_caption.models.caption_model.CaptionModel method) bernoulli_sample_sigmoid() (in module sparse_caption.pruning.sampler) BernoulliSample (class in sparse_caption.pruning.sampler) BernoulliSampleSigmoid (class in sparse_caption.pruning.sampler) Bleu (class in sparse_caption.coco_caption.pycocoevalcap.bleu.bleu) BleuScorer (class in sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer)

BleuSilent (class in sparse_caption.scst.scorers) bos_token (sparse_caption.tokenizer.Tokenizer property) bos_token_id (sparse_caption.tokenizer.RadixTokenizer property) (sparse_caption.tokenizer.SentencePieceUnigramTokenizer property) (sparse_caption.tokenizer.Tokenizer property) box_attention() (sparse_caption.models.relation_transformer.BoxMultiHeadedAttention static method) BoxMultiHeadedAttention (class in sparse_caption.models.relation_transformer) (class in sparse_caption.models.relation_transformer_prune) BoxRelationalEmbedding() (sparse_caption.models.relation_transformer.BoxMultiHeadedAttention static method) build_optimizer() (in module sparse_caption.utils.optim)

C

CachedMultiHeadedAttention (class in sparse_caption.models.relation_transformer_prune) (class in sparse_caption.models.transformer) CachedTransformerBase (class in sparse_caption.models.transformer) calc_score() (sparse_caption.coco_caption.pycocoevalcap.rouge.rouge.Rouge method) calculate_sparsities() (sparse_caption.pruning.prune.PruningMixin static method) CaptionModel (class in sparse_caption.models.caption_model) CaptionScorer (class in sparse_caption.scst.scorers) CharacterTokenizer (class in sparse_caption.tokenizer) check_loaded_version() (sparse_caption.utils.config.Config method) checkpoint_path (sparse_caption.utils.training.TrainingModule attribute) ChoiceList (class in sparse_caption.utils.misc) Cider (class in sparse_caption.coco_caption.pycocoevalcap.cider.cider) (class in sparse_caption.scst.cider.pyciderevalcap.cider.cider) CiderD (class in sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD) CIDErEvalCap (class in sparse_caption.scst.cider.pyciderevalcap.eval) CiderScorer (class in sparse_caption.coco_caption.pycocoevalcap.cider.cider_scorer) (class in sparse_caption.scst.cider.pyciderevalcap.cider.cider_scorer) (class in sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD_scorer) clear() (sparse_caption.scst.cider.pyciderevalcap.cider.cider_scorer.CiderScorer method) (sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD_scorer.CiderScorer method) clip_att() (sparse_caption.models.att_model.AttModel method) (sparse_caption.models.relation_transformer.RelationTransformerModel static method) clip_gradient() (in module sparse_caption.utils.optim) clones() (in module sparse_caption.utils.model_utils) CMHA (in module sparse_caption.models.transformer) coco_annot_json_dump() (sparse_caption.data.karpathy.KarpathyDataset method) coco_caption_json_dump() (sparse_caption.data.karpathy.KarpathyDataset method) COCOEvalCap (class in sparse_caption.coco_caption.pycocoevalcap.eval) COLLATE_FN (sparse_caption.models.att_model.UpDownModel attribute) (sparse_caption.models.att_model_prune.UpDownModel attribute) (sparse_caption.models.relation_transformer.RelationTransformerModel attribute) (sparse_caption.models.transformer.Transformer attribute) collate_fn (sparse_caption.utils.training.TrainingModule attribute) compat() (sparse_caption.utils.config.Config method) compatible() (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer method) compute_cider() (sparse_caption.coco_caption.pycocoevalcap.cider.cider_scorer.CiderScorer method) (sparse_caption.scst.cider.pyciderevalcap.cider.cider_scorer.CiderScorer method) (sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD_scorer.CiderScorer method) compute_doc_freq() (sparse_caption.coco_caption.pycocoevalcap.cider.cider_scorer.CiderScorer method) (sparse_caption.scst.cider.pyciderevalcap.cider.cider_scorer.CiderScorer method) (sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD_scorer.CiderScorer method) compute_mask() (sparse_caption.pruning.prune.PruningMixin static method)

compute_score() (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu.Bleu method) (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer method) (sparse_caption.coco_caption.pycocoevalcap.cider.cider.Cider method) (sparse_caption.coco_caption.pycocoevalcap.cider.cider_scorer.CiderScorer method) (sparse_caption.coco_caption.pycocoevalcap.meteor.meteor.Meteor method) (sparse_caption.coco_caption.pycocoevalcap.rouge.rouge.Rouge method) (sparse_caption.coco_caption.pycocoevalcap.spice.spice.Spice method) (sparse_caption.scst.cider.pyciderevalcap.cider.cider.Cider method) (sparse_caption.scst.cider.pyciderevalcap.cider.cider_scorer.CiderScorer method) (sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD.CiderD method) (sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD_scorer.CiderScorer method) (sparse_caption.scst.scorers.BleuSilent method) compute_scst_loss() (sparse_caption.utils.training.TrainingModule method) compute_sparsity_loss() (sparse_caption.pruning.prune.PruningMixin method) Config (class in sparse_caption.utils.config) config (sparse_caption.utils.training.TrainingModule attribute) configure_logging() (in module sparse_caption.utils.misc) control_symbols (sparse_caption.tokenizer.Tokenizer attribute) cook_append() (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer method) (sparse_caption.coco_caption.pycocoevalcap.cider.cider_scorer.CiderScorer method) (sparse_caption.scst.cider.pyciderevalcap.cider.cider_scorer.CiderScorer method) (sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD_scorer.CiderScorer method) cook_refs() (in module sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer) (in module sparse_caption.coco_caption.pycocoevalcap.cider.cider_scorer) (in module sparse_caption.scst.cider.pyciderevalcap.cider.cider_scorer) (in module sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD_scorer) cook_test() (in module sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer) (in module sparse_caption.coco_caption.pycocoevalcap.cider.cider_scorer) (in module sparse_caption.scst.cider.pyciderevalcap.cider.cider_scorer) (in module sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD_scorer) copy() (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer method) (sparse_caption.coco_caption.pycocoevalcap.cider.cider_scorer.CiderScorer method) (sparse_caption.scst.cider.pyciderevalcap.cider.cider_scorer.CiderScorer method) (sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD_scorer.CiderScorer method) copy_empty() (sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD_scorer.CiderScorer method) CosineOpt (class in sparse_caption.utils.optim) count_nonzero() (in module sparse_caption.utils.model_utils) crefs (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer attribute) csv_to_float_list() (in module sparse_caption.utils.misc) csv_to_int_list() (in module sparse_caption.utils.misc) csv_to_str_list() (in module sparse_caption.utils.misc) ctest (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer attribute)

D

data (sparse_caption.utils.training.TrainingModule attribute) decimal_to_base() (sparse_caption.tokenizer.RadixTokenizer static method) decode() (sparse_caption.models.relation_transformer.EncoderDecoder method) (sparse_caption.models.transformer.EncoderDecoder method) (sparse_caption.tokenizer.RadixTokenizer method) (sparse_caption.tokenizer.SentencePieceUnigramTokenizer method) (sparse_caption.tokenizer.Tokenizer method) Decoder (class in sparse_caption.models.transformer)

DecoderLayer (class in sparse_caption.models.transformer) deepcopy() (sparse_caption.utils.config.Config method) DEFAULT_ANNOT_DIR (sparse_caption.data.karpathy.KarpathyDataset attribute) densify_state_dict() (in module sparse_caption.utils.model_utils) dict() (sparse_caption.utils.config.Config method) disable_incremental_decoding() (sparse_caption.models.transformer.CachedTransformerBase static method) download_and_process_karpathy_json() (sparse_caption.data.karpathy.KarpathyDataset method) dump_json() (in module sparse_caption.utils.file) dumps_file() (in module sparse_caption.utils.file)

E

Embeddings (class in sparse_caption.models.relation_transformer_prune) enable_incremental_decoding() (sparse_caption.models.transformer.CachedTransformerBase static method) encode() (sparse_caption.models.relation_transformer.EncoderDecoder method) (sparse_caption.models.transformer.EncoderDecoder method) (sparse_caption.tokenizer.CharacterTokenizer method) (sparse_caption.tokenizer.RadixTokenizer method) (sparse_caption.tokenizer.SentencePieceUnigramTokenizer method) (sparse_caption.tokenizer.Tokenizer method) encode_tokenized() (sparse_caption.tokenizer.RadixTokenizer method) (sparse_caption.tokenizer.SentencePieceUnigramTokenizer method) (sparse_caption.tokenizer.Tokenizer method) Encoder (class in sparse_caption.models.relation_transformer) (class in sparse_caption.models.transformer) EncoderDecoder (class in sparse_caption.models.relation_transformer) (class in sparse_caption.models.relation_transformer_prune) (class in sparse_caption.models.transformer)

EncoderLayer (class in sparse_caption.models.relation_transformer) (class in sparse_caption.models.transformer) end (sparse_caption.tokenizer.Token property) eos_token (sparse_caption.tokenizer.Tokenizer property) eos_token_id (sparse_caption.tokenizer.RadixTokenizer property) (sparse_caption.tokenizer.SentencePieceUnigramTokenizer property) (sparse_caption.tokenizer.Tokenizer property) eval_model() (sparse_caption.utils.training.TrainingModule class method) eval_on_split() (sparse_caption.utils.training.TrainingModule method) evaluate() (sparse_caption.coco_caption.pycocoevalcap.eval.COCOEvalCap method) (sparse_caption.scst.cider.pyciderevalcap.eval.CIDErEvalCap method) evaluate_caption_json() (in module sparse_caption.coco_caption.eval)

F

file_size() (in module sparse_caption.utils.file) float_convert() (sparse_caption.coco_caption.pycocoevalcap.spice.spice.Spice method) forward() (sparse_caption.models.att_model.Attention method) (sparse_caption.models.att_model.UpDownCore method) (sparse_caption.models.caption_model.CaptionModel method) (sparse_caption.models.relation_transformer.BoxMultiHeadedAttention method) (sparse_caption.models.relation_transformer.Encoder method) (sparse_caption.models.relation_transformer.EncoderDecoder method) (sparse_caption.models.relation_transformer.EncoderLayer method) (sparse_caption.models.transformer.Decoder method) (sparse_caption.models.transformer.DecoderLayer method) (sparse_caption.models.transformer.Encoder method) (sparse_caption.models.transformer.EncoderDecoder method) (sparse_caption.models.transformer.EncoderLayer method) (sparse_caption.models.transformer.InputEmbedding method) (sparse_caption.models.transformer.LayerNorm method) (sparse_caption.models.transformer.MultiHeadedAttention method) (sparse_caption.models.transformer.OutputEmbedding method) (sparse_caption.models.transformer.PositionalEncoding method) (sparse_caption.models.transformer.PositionwiseFeedForward method) (sparse_caption.models.transformer.SublayerConnection method) (sparse_caption.pruning.masked_layer.MaskedEmbedding method) (sparse_caption.pruning.masked_layer.MaskedLinear method) (sparse_caption.pruning.masked_layer.MaskedLSTMCell method) (sparse_caption.pruning.masked_layer.MaskedLSTMCellCheckpoint method) (sparse_caption.pruning.sampler.BernoulliSample static method) (sparse_caption.pruning.sampler.BernoulliSampleSigmoid static method) (sparse_caption.pruning.sampler.Round static method) (sparse_caption.pruning.sampler.RoundSigmoid static method) (sparse_caption.utils.losses.LabelSmoothing method) (sparse_caption.utils.losses.LanguageModelCriterion method) (sparse_caption.utils.losses.RewardCriterion method)

from_pretrained() (sparse_caption.pruning.masked_layer.MaskedEmbedding class method)

G

generate() (sparse_caption.models.transformer.EncoderDecoder method) Generator (class in sparse_caption.models.relation_transformer_prune) get() (sparse_caption.utils.config.Config method) get_dataloader() (sparse_caption.utils.training.TrainingModule method) get_dataset() (in module sparse_caption.data) get_file() (in module sparse_caption.utils.file) get_logprobs_state() (sparse_caption.models.att_model.AttModel method) (sparse_caption.models.relation_transformer.RelationTransformerModel method) (sparse_caption.models.transformer.Transformer method)

get_masked_weight() (sparse_caption.pruning.masked_layer.MaskMixin method) get_memory_info() (in module sparse_caption.utils.misc) get_model() (in module sparse_caption.models) get_optim() (in module sparse_caption.utils.optim) get_split() (sparse_caption.data.karpathy.KarpathyDataset method) get_stanford_models() (in module sparse_caption.coco_caption.pycocoevalcap.spice.get_stanford_models) get_test2014_split() (sparse_caption.data.mscoco.MscocoDataset method) get_tokenizer() (in module sparse_caption.tokenizer) grouper() (in module sparse_caption.utils.misc) (sparse_caption.tokenizer.RadixTokenizer static method)

H

humanise_number() (in module sparse_caption.utils.misc)

I

id_to_token() (sparse_caption.tokenizer.RadixTokenizer method) (sparse_caption.tokenizer.SentencePieceUnigramTokenizer method) (sparse_caption.tokenizer.Tokenizer method) image_filename_to_id() (sparse_caption.data.karpathy.KarpathyDataset static method) (sparse_caption.data.mscoco.MscocoDataset static method)

init_hidden() (sparse_caption.models.att_model.AttModel method) input_check() (sparse_caption.scst.scorers.CaptionScorer static method) InputEmbedding (class in sparse_caption.models.transformer)

J

join_default_bu_dir() (sparse_caption.data.collate.UpDownCollate method)

json() (sparse_caption.utils.config.Config method)

K

KarpathyDataset (class in sparse_caption.data.karpathy)

L

LabelSmoothing (class in sparse_caption.utils.losses) LanguageModelCriterion (class in sparse_caption.utils.losses) LayerNorm (class in sparse_caption.models.transformer) length_average() (in module sparse_caption.utils.model_utils) length_wu() (in module sparse_caption.utils.model_utils) list_dir() (in module sparse_caption.utils.file)

list_files() (in module sparse_caption.utils.file) ListDataset (class in sparse_caption.data.collate) load_config_json() (sparse_caption.utils.config.Config class method) load_pil_image_from_url() (in module sparse_caption.utils.file) load_sparse_state_dict() (sparse_caption.pruning.prune.PruningMixin method) load_state_dict (sparse_caption.pruning.prune.PruningMixin attribute) LoadData (class in sparse_caption.scst.cider.pydataformat.loadData)

M

make_model() (sparse_caption.models.att_model.AttModel method) (sparse_caption.models.att_model_prune.AttModel method) (sparse_caption.models.relation_transformer.RelationTransformerModel method) (sparse_caption.models.relation_transformer_prune.RelationTransformerModel method) (sparse_caption.models.transformer.Transformer method) map_recursive() (in module sparse_caption.utils.model_utils) map_to_cuda() (in module sparse_caption.utils.model_utils) mask_init_value (sparse_caption.pruning.masked_layer.MaskedEmbedding attribute) (sparse_caption.pruning.masked_layer.MaskedLinear attribute) (sparse_caption.pruning.masked_layer.MaskedLSTMCell attribute) (sparse_caption.pruning.masked_layer.MaskedLSTMCellCheckpoint attribute) (sparse_caption.pruning.masked_layer.MaskMixin attribute) mask_token (sparse_caption.tokenizer.Tokenizer property) mask_token_id (sparse_caption.tokenizer.RadixTokenizer property) (sparse_caption.tokenizer.SentencePieceUnigramTokenizer property) (sparse_caption.tokenizer.Tokenizer property) mask_trainable (sparse_caption.pruning.masked_layer.MaskedEmbedding attribute) (sparse_caption.pruning.masked_layer.MaskedLinear attribute) (sparse_caption.pruning.masked_layer.MaskedLSTMCell attribute) (sparse_caption.pruning.masked_layer.MaskedLSTMCellCheckpoint attribute) (sparse_caption.pruning.masked_layer.MaskMixin attribute) mask_type (sparse_caption.pruning.masked_layer.MaskedEmbedding attribute) (sparse_caption.pruning.masked_layer.MaskedLinear attribute) (sparse_caption.pruning.masked_layer.MaskedLSTMCell attribute) (sparse_caption.pruning.masked_layer.MaskedLSTMCellCheckpoint attribute) (sparse_caption.pruning.masked_layer.MaskMixin attribute) MaskedEmbedding (class in sparse_caption.pruning.masked_layer) MaskedLinear (class in sparse_caption.pruning.masked_layer) MaskedLSTMCell (class in sparse_caption.pruning.masked_layer) MaskedLSTMCellCheckpoint (class in sparse_caption.pruning.masked_layer) MaskMixin (class in sparse_caption.pruning.masked_layer) maybe_load_checkpoint() (sparse_caption.utils.training.TrainingModule method) Meteor (class in sparse_caption.coco_caption.pycocoevalcap.meteor.meteor) method() (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu.Bleu method) (sparse_caption.coco_caption.pycocoevalcap.cider.cider.Cider method) (sparse_caption.coco_caption.pycocoevalcap.meteor.meteor.Meteor method) (sparse_caption.coco_caption.pycocoevalcap.rouge.rouge.Rouge method) (sparse_caption.coco_caption.pycocoevalcap.spice.spice.Spice method) (sparse_caption.scst.cider.pyciderevalcap.cider.cider.Cider method) (sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD.CiderD method) MHA (in module sparse_caption.models.transformer) model (sparse_caption.utils.training.TrainingModule attribute) MODEL_TYPE (sparse_caption.tokenizer.CharacterTokenizer attribute) (sparse_caption.tokenizer.RadixTokenizer attribute) (sparse_caption.tokenizer.SentencePieceBPETokenizer attribute) (sparse_caption.tokenizer.SentencePieceUnigramTokenizer attribute) (sparse_caption.tokenizer.WordTokenizer attribute) module sparse_caption sparse_caption.coco_caption sparse_caption.coco_caption.eval sparse_caption.coco_caption.pycocoevalcap sparse_caption.coco_caption.pycocoevalcap.bleu sparse_caption.coco_caption.pycocoevalcap.bleu.bleu sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer sparse_caption.coco_caption.pycocoevalcap.cider sparse_caption.coco_caption.pycocoevalcap.cider.cider sparse_caption.coco_caption.pycocoevalcap.cider.cider_scorer sparse_caption.coco_caption.pycocoevalcap.eval sparse_caption.coco_caption.pycocoevalcap.meteor sparse_caption.coco_caption.pycocoevalcap.meteor.meteor sparse_caption.coco_caption.pycocoevalcap.rouge sparse_caption.coco_caption.pycocoevalcap.rouge.rouge sparse_caption.coco_caption.pycocoevalcap.spice sparse_caption.coco_caption.pycocoevalcap.spice.get_stanford_models sparse_caption.coco_caption.pycocoevalcap.spice.spice sparse_caption.coco_caption.pycocoevalcap.tokenizer sparse_caption.coco_caption.pycocoevalcap.tokenizer.ptbtokenizer sparse_caption.data sparse_caption.data.collate sparse_caption.data.karpathy sparse_caption.data.mscoco sparse_caption.models sparse_caption.models.att_model sparse_caption.models.att_model_prune sparse_caption.models.caption_model sparse_caption.models.relation_transformer sparse_caption.models.relation_transformer_prune sparse_caption.models.transformer sparse_caption.opts sparse_caption.pruning sparse_caption.pruning.masked_layer sparse_caption.pruning.prune sparse_caption.pruning.sampler sparse_caption.scst sparse_caption.scst.cider sparse_caption.scst.cider.pyciderevalcap sparse_caption.scst.cider.pyciderevalcap.cider sparse_caption.scst.cider.pyciderevalcap.cider.cider sparse_caption.scst.cider.pyciderevalcap.cider.cider_scorer sparse_caption.scst.cider.pyciderevalcap.ciderD sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD_scorer sparse_caption.scst.cider.pyciderevalcap.eval sparse_caption.scst.cider.pyciderevalcap.tokenizer sparse_caption.scst.cider.pyciderevalcap.tokenizer.ptbtokenizer sparse_caption.scst.cider.pydataformat sparse_caption.scst.cider.pydataformat.loadData sparse_caption.scst.scorers sparse_caption.tokenizer sparse_caption.utils sparse_caption.utils.config sparse_caption.utils.file sparse_caption.utils.losses sparse_caption.utils.misc sparse_caption.utils.model_utils sparse_caption.utils.natural_sort sparse_caption.utils.optim sparse_caption.utils.training sparse_caption.version

MscocoDataset (class in sparse_caption.data.mscoco) MscocoTesting (class in sparse_caption.data.mscoco) MultiHeadedAttention (class in sparse_caption.models.transformer) my_lcs() (in module sparse_caption.coco_caption.pycocoevalcap.rouge.rouge)

N

n (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer attribute) named_parameters (sparse_caption.pruning.prune.PruningMixin attribute)

natural_keys() (in module sparse_caption.utils.natural_sort) NoamOpt (class in sparse_caption.utils.optim)

O

ObjectRelationCollate (class in sparse_caption.data.collate)

optimizer (sparse_caption.utils.training.TrainingModule attribute) OutputEmbedding (class in sparse_caption.models.transformer)

P

pack_wrapper() (in module sparse_caption.utils.model_utils) pad_token (sparse_caption.tokenizer.Tokenizer property) pad_token_id (sparse_caption.tokenizer.RadixTokenizer property) (sparse_caption.tokenizer.SentencePieceUnigramTokenizer property) (sparse_caption.tokenizer.Tokenizer property) pad_unsort_packed_sequence() (in module sparse_caption.utils.model_utils) parse_opt() (in module sparse_caption.opts) penalty_builder() (in module sparse_caption.utils.model_utils) PositionalEncoding (class in sparse_caption.models.transformer) PositionwiseFeedForward (class in sparse_caption.models.relation_transformer_prune) (class in sparse_caption.models.transformer) precook() (in module sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer) (in module sparse_caption.coco_caption.pycocoevalcap.cider.cider_scorer) (in module sparse_caption.scst.cider.pyciderevalcap.cider.cider_scorer) (in module sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD_scorer)

prepare() (sparse_caption.utils.training.TrainingModule method) prepare_data() (sparse_caption.data.karpathy.KarpathyDataset method) (sparse_caption.data.mscoco.MscocoDataset method) print_progress() (in module sparse_caption.coco_caption.pycocoevalcap.spice.get_stanford_models) process_tokens() (sparse_caption.tokenizer.Tokenizer static method) processor (sparse_caption.tokenizer.Tokenizer attribute) prune_weights() (sparse_caption.pruning.prune.PruningMixin method) PruningMixin (class in sparse_caption.pruning.prune) PTBTokenizer (class in sparse_caption.coco_caption.pycocoevalcap.tokenizer.ptbtokenizer) (class in sparse_caption.scst.cider.pyciderevalcap.tokenizer.ptbtokenizer)

R

RadixTokenizer (class in sparse_caption.tokenizer) random_image_check() (sparse_caption.data.karpathy.KarpathyDataset method) rate() (sparse_caption.utils.optim.CosineOpt method) (sparse_caption.utils.optim.NoamOpt method) (sparse_caption.utils.optim.StepLROpt method) RateOpt (class in sparse_caption.utils.optim) ratio() (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer method) RAW_JSON_FILE (sparse_caption.data.karpathy.KarpathyDataset attribute) (sparse_caption.data.mscoco.MscocoDataset attribute) (sparse_caption.data.mscoco.MscocoTesting attribute) read_file() (in module sparse_caption.utils.file) read_json() (in module sparse_caption.utils.file) readJson() (sparse_caption.scst.cider.pydataformat.loadData.LoadData method) recompute_score() (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer method) reflen() (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer method) register_dataset() (in module sparse_caption.data) register_model() (in module sparse_caption.models)

register_tokenizer() (in module sparse_caption.tokenizer) RelationTransformerModel (class in sparse_caption.models.relation_transformer) (class in sparse_caption.models.relation_transformer_prune) reorder_beam() (in module sparse_caption.utils.model_utils) repeat_tensors() (in module sparse_caption.utils.model_utils) replace_from_right() (in module sparse_caption.utils.misc) requires_grad() (in module sparse_caption.utils.model_utils) rescore() (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer method) reset_cache() (sparse_caption.models.transformer.CachedMultiHeadedAttention method) (sparse_caption.models.transformer.PositionalEncoding method) reset_masks() (sparse_caption.pruning.masked_layer.MaskMixin method) retest() (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer method) RewardCriterion (class in sparse_caption.utils.losses) Rouge (class in sparse_caption.coco_caption.pycocoevalcap.rouge.rouge) Round (class in sparse_caption.pruning.sampler) rounding_sigmoid() (in module sparse_caption.pruning.sampler) RoundSigmoid (class in sparse_caption.pruning.sampler)

S

sample_next_word() (sparse_caption.models.caption_model.CaptionModel static method) save_config() (sparse_caption.utils.config.Config method) score_ratio() (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer method) score_ratio_str() (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer method) SCST_BASELINE (sparse_caption.utils.training.TrainingModule attribute) SCST_SAMPLE (sparse_caption.utils.training.TrainingModule attribute) scst_scorer (sparse_caption.utils.training.TrainingModule attribute) SentencePieceBPETokenizer (class in sparse_caption.tokenizer) SentencePieceUnigramTokenizer (class in sparse_caption.tokenizer) sequence_from_numpy() (in module sparse_caption.utils.model_utils) set_seed() (in module sparse_caption.utils.model_utils) setEval() (sparse_caption.coco_caption.pycocoevalcap.eval.COCOEvalCap method) setEvalImgs() (sparse_caption.coco_caption.pycocoevalcap.eval.COCOEvalCap method) setImgToEvalImgs() (sparse_caption.coco_caption.pycocoevalcap.eval.COCOEvalCap method) setup_masks() (sparse_caption.pruning.masked_layer.MaskMixin method) single_reflen() (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer method) size() (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer method) (sparse_caption.coco_caption.pycocoevalcap.cider.cider_scorer.CiderScorer method) (sparse_caption.scst.cider.pyciderevalcap.cider.cider_scorer.CiderScorer method) (sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD_scorer.CiderScorer method) sort_pack_padded_sequence() (in module sparse_caption.utils.model_utils) sparse_caption module sparse_caption.coco_caption module sparse_caption.coco_caption.eval module sparse_caption.coco_caption.pycocoevalcap module sparse_caption.coco_caption.pycocoevalcap.bleu module sparse_caption.coco_caption.pycocoevalcap.bleu.bleu module sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer module sparse_caption.coco_caption.pycocoevalcap.cider module sparse_caption.coco_caption.pycocoevalcap.cider.cider module sparse_caption.coco_caption.pycocoevalcap.cider.cider_scorer module sparse_caption.coco_caption.pycocoevalcap.eval module sparse_caption.coco_caption.pycocoevalcap.meteor module sparse_caption.coco_caption.pycocoevalcap.meteor.meteor module sparse_caption.coco_caption.pycocoevalcap.rouge module sparse_caption.coco_caption.pycocoevalcap.rouge.rouge module sparse_caption.coco_caption.pycocoevalcap.spice module sparse_caption.coco_caption.pycocoevalcap.spice.get_stanford_models module sparse_caption.coco_caption.pycocoevalcap.spice.spice module sparse_caption.coco_caption.pycocoevalcap.tokenizer module sparse_caption.coco_caption.pycocoevalcap.tokenizer.ptbtokenizer module sparse_caption.data module sparse_caption.data.collate module sparse_caption.data.karpathy module sparse_caption.data.mscoco module sparse_caption.models module sparse_caption.models.att_model module sparse_caption.models.att_model_prune module sparse_caption.models.caption_model module sparse_caption.models.relation_transformer module sparse_caption.models.relation_transformer_prune module

sparse_caption.models.transformer module sparse_caption.opts module sparse_caption.pruning module sparse_caption.pruning.masked_layer module sparse_caption.pruning.prune module sparse_caption.pruning.sampler module sparse_caption.scst module sparse_caption.scst.cider module sparse_caption.scst.cider.pyciderevalcap module sparse_caption.scst.cider.pyciderevalcap.cider module sparse_caption.scst.cider.pyciderevalcap.cider.cider module sparse_caption.scst.cider.pyciderevalcap.cider.cider_scorer module sparse_caption.scst.cider.pyciderevalcap.ciderD module sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD module sparse_caption.scst.cider.pyciderevalcap.ciderD.ciderD_scorer module sparse_caption.scst.cider.pyciderevalcap.eval module sparse_caption.scst.cider.pyciderevalcap.tokenizer module sparse_caption.scst.cider.pyciderevalcap.tokenizer.ptbtokenizer module sparse_caption.scst.cider.pydataformat module sparse_caption.scst.cider.pydataformat.loadData module sparse_caption.scst.scorers module sparse_caption.tokenizer module sparse_caption.utils module sparse_caption.utils.config module sparse_caption.utils.file module sparse_caption.utils.losses module sparse_caption.utils.misc module sparse_caption.utils.model_utils module sparse_caption.utils.natural_sort module sparse_caption.utils.optim module sparse_caption.utils.training module sparse_caption.version module sparsity_check() (sparse_caption.pruning.prune.PruningMixin method) special_reflen (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer attribute) special_token_attributes (sparse_caption.tokenizer.Tokenizer attribute) Spice (class in sparse_caption.coco_caption.pycocoevalcap.spice.spice) split_tensors() (in module sparse_caption.utils.model_utils) start (sparse_caption.tokenizer.Token property) state_dict (sparse_caption.pruning.prune.PruningMixin attribute) state_dict_dense() (sparse_caption.pruning.prune.PruningMixin method) state_dict_sparse() (sparse_caption.pruning.prune.PruningMixin method) step() (sparse_caption.utils.optim.RateOpt method) StepLROpt (class in sparse_caption.utils.optim) str_to_bool() (in module sparse_caption.utils.misc) str_to_none() (in module sparse_caption.utils.misc) str_to_sequence() (in module sparse_caption.utils.misc) SublayerConnection (class in sparse_caption.models.transformer) subsequent_mask() (sparse_caption.models.relation_transformer.RelationTransformerModel static method)

T

test_dataloader() (sparse_caption.utils.training.TrainingModule method) testlen() (sparse_caption.coco_caption.pycocoevalcap.bleu.bleu_scorer.BleuScorer method) time_function() (in module sparse_caption.utils.misc) to_cuda() (in module sparse_caption.utils.model_utils) Token (class in sparse_caption.tokenizer) token_to_id() (sparse_caption.tokenizer.RadixTokenizer method) (sparse_caption.tokenizer.SentencePieceUnigramTokenizer method) (sparse_caption.tokenizer.Tokenizer method) tokenize() (sparse_caption.coco_caption.pycocoevalcap.tokenizer.ptbtokenizer.PTBTokenizer method) (sparse_caption.scst.cider.pyciderevalcap.tokenizer.ptbtokenizer.PTBTokenizer method) (sparse_caption.tokenizer.SentencePieceUnigramTokenizer method) (sparse_caption.tokenizer.Tokenizer method) Tokenizer (class in sparse_caption.tokenizer) tokenizer (sparse_caption.utils.training.TrainingModule attribute) total_mask_params (sparse_caption.pruning.prune.PruningMixin property) total_weight_params (sparse_caption.pruning.prune.PruningMixin property) tqdm_hook() (in module sparse_caption.utils.file) train() (sparse_caption.tokenizer.SentencePieceUnigramTokenizer method) train_captions_txt_dump() (sparse_caption.data.karpathy.KarpathyDataset method) train_dataloader() (sparse_caption.utils.training.TrainingModule method) train_size (sparse_caption.data.karpathy.KarpathyDataset property) trainable_pruning_masks() (sparse_caption.pruning.prune.PruningMixin method) training (sparse_caption.models.att_model.Attention attribute) (sparse_caption.models.att_model.AttModel attribute) (sparse_caption.models.att_model.UpDownCore attribute) (sparse_caption.models.att_model.UpDownModel attribute) (sparse_caption.models.att_model_prune.Attention attribute) (sparse_caption.models.att_model_prune.AttModel attribute) (sparse_caption.models.att_model_prune.UpDownCore attribute) (sparse_caption.models.caption_model.CaptionModel attribute) (sparse_caption.models.relation_transformer.BoxMultiHeadedAttention attribute) (sparse_caption.models.relation_transformer.Encoder attribute) (sparse_caption.models.relation_transformer.EncoderDecoder attribute) (sparse_caption.models.relation_transformer.EncoderLayer attribute) (sparse_caption.models.relation_transformer.RelationTransformerModel attribute) (sparse_caption.models.relation_transformer_prune.BoxMultiHeadedAttention attribute) (sparse_caption.models.relation_transformer_prune.CachedMultiHeadedAttention attribute) (sparse_caption.models.relation_transformer_prune.Embeddings attribute) (sparse_caption.models.relation_transformer_prune.Generator attribute) (sparse_caption.models.relation_transformer_prune.PositionwiseFeedForward attribute) (sparse_caption.models.transformer.CachedMultiHeadedAttention attribute) (sparse_caption.models.transformer.CachedTransformerBase attribute) (sparse_caption.models.transformer.Decoder attribute) (sparse_caption.models.transformer.DecoderLayer attribute) (sparse_caption.models.transformer.Encoder attribute) (sparse_caption.models.transformer.EncoderDecoder attribute) (sparse_caption.models.transformer.EncoderLayer attribute) (sparse_caption.models.transformer.InputEmbedding attribute) (sparse_caption.models.transformer.LayerNorm attribute) (sparse_caption.models.transformer.MultiHeadedAttention attribute) (sparse_caption.models.transformer.OutputEmbedding attribute) (sparse_caption.models.transformer.PositionalEncoding attribute) (sparse_caption.models.transformer.PositionwiseFeedForward attribute) (sparse_caption.models.transformer.SublayerConnection attribute) (sparse_caption.models.transformer.Transformer attribute) (sparse_caption.pruning.masked_layer.MaskedEmbedding attribute) (sparse_caption.pruning.masked_layer.MaskedLinear attribute) (sparse_caption.pruning.masked_layer.MaskedLSTMCell attribute) (sparse_caption.pruning.masked_layer.MaskedLSTMCellCheckpoint attribute) (sparse_caption.pruning.masked_layer.MaskMixin attribute) (sparse_caption.utils.losses.LabelSmoothing attribute) (sparse_caption.utils.losses.LanguageModelCriterion attribute) (sparse_caption.utils.losses.RewardCriterion attribute)

TrainingModule (class in sparse_caption.utils.training) Transformer (class in sparse_caption.models.transformer)

U

unk_token (sparse_caption.tokenizer.Tokenizer property) unk_token_id (sparse_caption.tokenizer.RadixTokenizer property) (sparse_caption.tokenizer.SentencePieceUnigramTokenizer property) (sparse_caption.tokenizer.Tokenizer property) update() (sparse_caption.utils.config.Config method) update_masks_gradual() (sparse_caption.pruning.prune.PruningMixin method)

update_masks_once() (sparse_caption.pruning.prune.PruningMixin method) UpDownCollate (class in sparse_caption.data.collate) UpDownCore (class in sparse_caption.models.att_model) (class in sparse_caption.models.att_model_prune) UpDownModel (class in sparse_caption.models.att_model) (class in sparse_caption.models.att_model_prune)

V

val_dataloader() (sparse_caption.utils.training.TrainingModule method) validate_file() (in module sparse_caption.utils.file)

value (sparse_caption.tokenizer.Token property) vocab_size (sparse_caption.tokenizer.Tokenizer property)

W

WordTokenizer (class in sparse_caption.tokenizer)

Z

zip_dir() (in module sparse_caption.utils.file)