Index

_

_init_weights() (virtex.modules.textual_heads.TransformerDecoderTextualHead static method) _job_worker() (in module virtex.utils.distributed)

_ModelZooUrls (class in virtex.model_zoo.model_zoo) _state_dict() (virtex.utils.checkpointing.CheckpointManager method)

A

AutoRegressiveBeamSearch (class in virtex.utils.beam_search)

average_across_processes() (in module virtex.utils.distributed)

B

BidirectionalCaptioningModel (class in virtex.models.captioning)

C

CaptioningDataset (class in virtex.data.datasets.captioning) CaptioningModel (class in virtex.models.captioning) CenterSquareCrop (class in virtex.data.transforms) CheckpointManager (class in virtex.utils.checkpointing) cider() (in module virtex.utils.metrics) ClassificationModel (class in virtex.models.classification) CocoCaptionsDataset (class in virtex.data.datasets.coco_captions)

CocoCaptionsEvaluator (class in virtex.utils.metrics) common_parser() (in module virtex.utils.common) common_setup() (in module virtex.utils.common) Config (class in virtex.config) create() (virtex.factories.Factory class method) (virtex.factories.ImageTransformsFactory class method) cycle() (in module virtex.utils.common)

D

decode() (virtex.data.tokenizers.SentencePieceBPETokenizer method) decoding_step() (virtex.models.captioning.CaptioningModel method) DEFAULT_IMAGE_TRANSFORM (in module virtex.data.transforms)

detectron2_backbone_state_dict() (virtex.modules.visual_backbones.TorchvisionVisualBackbone method) DownstreamDatasetFactory (class in virtex.factories) dump() (virtex.config.Config method)

E

encode() (virtex.data.tokenizers.SentencePieceBPETokenizer method)

eta_hhmm (virtex.utils.timer.Timer property) evaluate() (virtex.utils.metrics.CocoCaptionsEvaluator method)

F

Factory (class in virtex.factories) forward() (virtex.models.captioning.CaptioningModel method) (virtex.models.classification.ClassificationModel method) (virtex.models.masked_lm.MaskedLMModel method) (virtex.modules.embedding.WordAndPositionalEmbedding method) (virtex.modules.textual_heads.LinearTextualHead method) (virtex.modules.textual_heads.TransformerDecoderTextualHead method) (virtex.modules.visual_backbones.TorchvisionVisualBackbone method) ForwardCaptioningModel (class in virtex.models.captioning)

from_config() (virtex.factories.DownstreamDatasetFactory class method) (virtex.factories.Factory class method) (virtex.factories.ImageTransformsFactory class method) (virtex.factories.LRSchedulerFactory class method) (virtex.factories.OptimizerFactory class method) (virtex.factories.PretrainingDatasetFactory class method) (virtex.factories.PretrainingModelFactory class method) (virtex.factories.TextualHeadFactory class method) (virtex.factories.TokenizerFactory class method) (virtex.factories.VisualBackboneFactory class method)

G

get() (in module virtex.model_zoo.model_zoo) get_rank() (in module virtex.utils.distributed)

get_vocab_size() (virtex.data.tokenizers.SentencePieceBPETokenizer method) get_world_size() (in module virtex.utils.distributed) gpu_mem_usage() (in module virtex.utils.distributed)

H

HorizontalFlip (class in virtex.data.transforms)

I

id_to_token() (virtex.data.tokenizers.SentencePieceBPETokenizer method) ImageDirectoryDataset (class in virtex.data.datasets.downstream) IMAGENET_COLOR_MEAN (in module virtex.data.transforms) IMAGENET_COLOR_STD (in module virtex.data.transforms)

ImageNetDataset (class in virtex.data.datasets.downstream) ImageTransformsFactory (class in virtex.factories) INaturalist2018Dataset (class in virtex.data.datasets.downstream) is_master_process() (in module virtex.utils.distributed)

L

launch() (in module virtex.utils.distributed) LinearTextualHead (class in virtex.modules.textual_heads) LinearWarmupCosineAnnealingLR (class in virtex.optim.lr_scheduler) LinearWarmupLinearDecayLR (class in virtex.optim.lr_scheduler) LinearWarmupMultiStepLR (class in virtex.optim.lr_scheduler)

LinearWarmupNoDecayLR (class in virtex.optim.lr_scheduler) load() (virtex.utils.checkpointing.CheckpointManager method) load_slow_weights() (virtex.optim.lookahead.Lookahead method) load_state_dict() (virtex.optim.lookahead.Lookahead method) Lookahead (class in virtex.optim.lookahead) LRSchedulerFactory (class in virtex.factories)

M

make_future_mask() (virtex.modules.textual_heads.TransformerDecoderTextualHead static method) MaskedLMModel (class in virtex.models.masked_lm) module virtex.config virtex.data.datasets.captioning virtex.data.datasets.classification virtex.data.datasets.coco_captions virtex.data.datasets.downstream virtex.data.tokenizers virtex.data.transforms virtex.factories virtex.model_zoo.model_zoo virtex.models.captioning virtex.models.classification virtex.models.masked_lm virtex.modules.embedding virtex.modules.textual_heads virtex.modules.visual_backbones virtex.optim.lookahead virtex.optim.lr_scheduler virtex.utils.beam_search virtex.utils.checkpointing virtex.utils.common virtex.utils.distributed virtex.utils.metrics virtex.utils.timer

MultiLabelClassificationDataset (class in virtex.data.datasets.classification) MultiLabelClassificationModel (class in virtex.models.classification)

O

OptimizerFactory (class in virtex.factories)

P

PretrainingDatasetFactory (class in virtex.factories)

PretrainingModelFactory (class in virtex.factories)

R

RandomResizedSquareCrop (class in virtex.data.transforms)

remove_earliest_checkpoint() (virtex.utils.checkpointing.CheckpointManager method) restore_fast_weights() (virtex.optim.lookahead.Lookahead method)

S

search() (virtex.utils.beam_search.AutoRegressiveBeamSearch method) SentencePieceBPETokenizer (class in virtex.data.tokenizers) spice() (in module virtex.utils.metrics) SquareResize (class in virtex.data.transforms)

state_dict() (virtex.optim.lookahead.Lookahead method) stats (virtex.utils.timer.Timer property) step() (virtex.optim.lookahead.Lookahead method) (virtex.utils.checkpointing.CheckpointManager method) synchronize() (in module virtex.utils.distributed)

T

textual_feature_size (virtex.modules.textual_heads.TextualHead property) TextualHead (class in virtex.modules.textual_heads) TextualHeadFactory (class in virtex.factories) tic() (virtex.utils.timer.Timer method) Timer (class in virtex.utils.timer) toc() (virtex.utils.timer.Timer method) token_to_id() (virtex.data.tokenizers.SentencePieceBPETokenizer method)

TokenClassificationDataset (class in virtex.data.datasets.classification) TokenClassificationModel (class in virtex.models.classification) tokenize() (in module virtex.utils.metrics) TokenizerFactory (class in virtex.factories) TopkAccuracy (class in virtex.utils.metrics) TorchvisionVisualBackbone (class in virtex.modules.visual_backbones) TransformerDecoderTextualHead (class in virtex.modules.textual_heads)

V

virtex.config module virtex.data.datasets.captioning module virtex.data.datasets.classification module virtex.data.datasets.coco_captions module virtex.data.datasets.downstream module virtex.data.tokenizers module virtex.data.transforms module virtex.factories module virtex.model_zoo.model_zoo module virtex.models.captioning module virtex.models.classification module virtex.models.masked_lm module virtex.modules.embedding module

virtex.modules.textual_heads module virtex.modules.visual_backbones module virtex.optim.lookahead module virtex.optim.lr_scheduler module virtex.utils.beam_search module virtex.utils.checkpointing module virtex.utils.common module virtex.utils.distributed module virtex.utils.metrics module virtex.utils.timer module VirTexModel (in module virtex.models.captioning) VisualBackbone (class in virtex.modules.visual_backbones) VisualBackboneFactory (class in virtex.factories) VOC07ClassificationDataset (class in virtex.data.datasets.downstream)

W

WordAndPositionalEmbedding (class in virtex.modules.embedding)

Z

zero_grad() (virtex.optim.lookahead.Lookahead method)