torchseq.models package

Subpackages

• torchseq.models.rerankers package
  • Submodules
  • torchseq.models.rerankers.backtranslate_reranker module
  • torchseq.models.rerankers.combo module
  • torchseq.models.rerankers.ngram_reranker module
  • torchseq.models.rerankers.qa_reranker module
  • torchseq.models.rerankers.topk module
  • Module contents
• torchseq.models.samplers package
  • Submodules
  • torchseq.models.samplers.beam_search module
  • torchseq.models.samplers.diverse_beam module
  • torchseq.models.samplers.greedy module
  • torchseq.models.samplers.parallel_nucleus module
  • torchseq.models.samplers.teacher_force module
  • Module contents

Submodules

torchseq.models.activations module

torchseq.models.aq_transformer module

torchseq.models.bottleneck module

torchseq.models.bottleneck_autoencoder module

torchseq.models.decoder module

torchseq.models.encoder module

torchseq.models.hrq_vae module

torchseq.models.kl_divergence module

torchseq.models.lang_predict_loss module

torchseq.models.lm_transformer module

torchseq.models.lr_schedule module

torchseq.models.modular_bottleneck module

torchseq.models.multihead_output module

torchseq.models.parallel_wrapper module

torchseq.models.pooling module

torchseq.models.positional_embeddings module

torchseq.models.positional_embeddings_fb module

torchseq.models.pretrained_adapter module

torchseq.models.pythae_vq module

torchseq.models.ranger module

torchseq.models.suppression_loss module

torchseq.models.transformer module

torchseq.models.vmf module

torchseq.models.vq_code_predictor module

torchseq.models.vq_vae module

torchseq.models.vq_vae_legacy module

Module contents