langvae package

Subpackages

• langvae.arch package
  • Submodules
  • langvae.arch.vae module
    • LangVAE
      • LangVAE.decode_sentences()
      • LangVAE.encode_z()
      • LangVAE.forward()
      • LangVAE.load_from_folder()
      • LangVAE.load_from_hf_hub()
      • LangVAE.loss_function()
      • LangVAE.push_to_hf_hub()
      • LangVAE.save()
    • vae_nll_loss()
  • Module contents
• langvae.data_conversion package
  • Submodules
  • langvae.data_conversion.sparse module
    • densify_w_padding()
  • langvae.data_conversion.tokenization module
    • TokenizedAnnotatedDataSet
      • TokenizedAnnotatedDataSet.source
      • TokenizedAnnotatedDataSet.tokenizer
      • TokenizedAnnotatedDataSet.max_len
      • TokenizedAnnotatedDataSet.annotations
      • TokenizedAnnotatedDataSet.caching
      • TokenizedAnnotatedDataSet.cache_persistence
      • TokenizedAnnotatedDataSet.tokenizer_options
    • TokenizedDataSet
      • TokenizedDataSet.source
      • TokenizedDataSet.tokenizer
      • TokenizedDataSet.max_len
      • TokenizedDataSet.caching
      • TokenizedDataSet.cache_persistence
      • TokenizedDataSet.tokenizer_options
      • TokenizedDataSet.vocab_size
    • collate_sparse_fn()
    • get_hash()
  • Module contents
• langvae.decoders package
  • Submodules
  • langvae.decoders.sentence module
    • SentenceDecoder
      • SentenceDecoder.model_path
      • SentenceDecoder.latent_size
      • SentenceDecoder.max_len
      • SentenceDecoder.device
      • SentenceDecoder.device_map
      • SentenceDecoder.args
      • SentenceDecoder.decoder
      • SentenceDecoder.forward()
      • SentenceDecoder.generate()
      • SentenceDecoder.init_pretrained_model()
      • SentenceDecoder.to()
      • SentenceDecoder.tokenizer
  • Module contents
• langvae.encoders package
  • Submodules
  • langvae.encoders.automodel_presets module
    • AutoModelPreset
      • AutoModelPreset.cls
      • AutoModelPreset.pooling_method
      • AutoModelPreset.normalize
      • AutoModelPreset.cls
      • AutoModelPreset.cls_type
      • AutoModelPreset.normalize
      • AutoModelPreset.pooling_method
    • PoolingMethod
      • PoolingMethod.CLS
      • PoolingMethod.LAST
      • PoolingMethod.MEAN
  • langvae.encoders.sentence module
    • SentenceEncoder
      • SentenceEncoder.encoder
      • SentenceEncoder.linear
      • SentenceEncoder.tokenizer
      • SentenceEncoder.decoder_tokenizer
      • SentenceEncoder.caching
      • SentenceEncoder.automodel_preset
      • SentenceEncoder.device
      • SentenceEncoder.debug
      • SentenceEncoder.annotated
      • SentenceEncoder.decoder_tokenizer
      • SentenceEncoder.encoder
      • SentenceEncoder.forward()
      • SentenceEncoder.init_pretrained_model()
      • SentenceEncoder.recode()
      • SentenceEncoder.to()
      • SentenceEncoder.tokenizer
    • cls_token_pooling()
    • last_token_pooling()
    • mean_pooling()
  • Module contents
• langvae.pipelines package
  • Submodules
  • langvae.pipelines.lang_training module
    • LanguageTrainingPipeline
  • Module contents
• langvae.trainers package
  • Submodules
  • langvae.trainers.cyclical_schedule_kl module
    • CyclicalScheduleKLThresholdTrainer
      • CyclicalScheduleKLThresholdTrainer.eval_step()
      • CyclicalScheduleKLThresholdTrainer.get_eval_dataloader()
      • CyclicalScheduleKLThresholdTrainer.get_train_dataloader()
      • CyclicalScheduleKLThresholdTrainer.train()
      • CyclicalScheduleKLThresholdTrainer.train_step()
    • CyclicalScheduleKLThresholdTrainerConfig
      • CyclicalScheduleKLThresholdTrainerConfig.max_beta
      • CyclicalScheduleKLThresholdTrainerConfig.n_cycles
      • CyclicalScheduleKLThresholdTrainerConfig.start_beta
      • CyclicalScheduleKLThresholdTrainerConfig.target_kl
    • copy_model_ptref()
    • frange_cycle_zero_linear()
  • langvae.trainers.training_callbacks module
  • Module contents

Module contents