Index

_

__call__() (pythae.pipelines.GenerationPipeline method) (pythae.pipelines.TrainingPipeline method)

A

Adversarial_AE (class in pythae.models) Adversarial_AE_Config (class in pythae.models) AdversarialTrainer (class in pythae.trainers)

AdversarialTrainerConfig (class in pythae.trainers) AE (class in pythae.models) AEConfig (class in pythae.models) AutoModel (class in pythae.models)

B

BaseAE (class in pythae.models) BaseAEConfig (class in pythae.models) BaseDecoder (class in pythae.models.nn) BaseDiscriminator (class in pythae.models.nn) BaseEncoder (class in pythae.models.nn) BaseMetric (class in pythae.models.nn) BaseNF (class in pythae.models.normalizing_flows) BaseNFConfig (class in pythae.models.normalizing_flows)

BaseSampler (class in pythae.samplers) BaseSamplerConfig (class in pythae.samplers) BaseTrainer (class in pythae.trainers) BaseTrainerConfig (class in pythae.trainers) BetaTCVAE (class in pythae.models) BetaTCVAEConfig (class in pythae.models) BetaVAE (class in pythae.models) BetaVAEConfig (class in pythae.models)

C

CallbackHandler (class in pythae.trainers.training_callbacks) CIWAE (class in pythae.models) CIWAEConfig (class in pythae.models) CometCallback (class in pythae.trainers.training_callbacks)

CoupledOptimizerAdversarialTrainer (class in pythae.trainers) CoupledOptimizerAdversarialTrainerConfig (class in pythae.trainers) CoupledOptimizerTrainer (class in pythae.trainers) CoupledOptimizerTrainerConfig (class in pythae.trainers)

D

Decoder_Conv_AE_CELEBA (class in pythae.models.nn.benchmarks.celeba) Decoder_Conv_AE_CIFAR (class in pythae.models.nn.benchmarks.cifar) Decoder_Conv_AE_MNIST (class in pythae.models.nn.benchmarks.mnist) Decoder_ResNet_AE_CELEBA (class in pythae.models.nn.benchmarks.celeba) Decoder_ResNet_AE_CIFAR (class in pythae.models.nn.benchmarks.cifar) Decoder_ResNet_AE_MNIST (class in pythae.models.nn.benchmarks.mnist) Decoder_ResNet_VQVAE_CELEBA (class in pythae.models.nn.benchmarks.celeba)

Decoder_ResNet_VQVAE_CIFAR (class in pythae.models.nn.benchmarks.cifar) Decoder_ResNet_VQVAE_MNIST (class in pythae.models.nn.benchmarks.mnist) Discriminator_Conv_CELEBA (class in pythae.models.nn.benchmarks.celeba) Discriminator_Conv_CIFAR (class in pythae.models.nn.benchmarks.cifar) Discriminator_Conv_MNIST (class in pythae.models.nn.benchmarks.mnist) DisentangledBetaVAE (class in pythae.models) DisentangledBetaVAEConfig (class in pythae.models)

E

embed() (pythae.models.BaseAE method) Encoder_Conv_AE_CELEBA (class in pythae.models.nn.benchmarks.celeba) Encoder_Conv_AE_CIFAR (class in pythae.models.nn.benchmarks.cifar) Encoder_Conv_AE_MNIST (class in pythae.models.nn.benchmarks.mnist) Encoder_Conv_SVAE_CELEBA (class in pythae.models.nn.benchmarks.celeba) Encoder_Conv_SVAE_CIFAR (class in pythae.models.nn.benchmarks.cifar) Encoder_Conv_SVAE_MNIST (class in pythae.models.nn.benchmarks.mnist) Encoder_Conv_VAE_CELEBA (class in pythae.models.nn.benchmarks.celeba) Encoder_Conv_VAE_CIFAR (class in pythae.models.nn.benchmarks.cifar) Encoder_Conv_VAE_MNIST (class in pythae.models.nn.benchmarks.mnist) Encoder_ResNet_AE_CELEBA (class in pythae.models.nn.benchmarks.celeba) Encoder_ResNet_AE_CIFAR (class in pythae.models.nn.benchmarks.cifar) Encoder_ResNet_AE_MNIST (class in pythae.models.nn.benchmarks.mnist)

Encoder_ResNet_SVAE_CELEBA (class in pythae.models.nn.benchmarks.celeba) Encoder_ResNet_SVAE_CIFAR (class in pythae.models.nn.benchmarks.cifar) Encoder_ResNet_SVAE_MNIST (class in pythae.models.nn.benchmarks.mnist) Encoder_ResNet_VAE_CELEBA (class in pythae.models.nn.benchmarks.celeba) Encoder_ResNet_VAE_CIFAR (class in pythae.models.nn.benchmarks.cifar) Encoder_ResNet_VAE_MNIST (class in pythae.models.nn.benchmarks.mnist) Encoder_ResNet_VQVAE_CELEBA (class in pythae.models.nn.benchmarks.celeba) Encoder_ResNet_VQVAE_CIFAR (class in pythae.models.nn.benchmarks.cifar) Encoder_ResNet_VQVAE_MNIST (class in pythae.models.nn.benchmarks.mnist) eval_step() (pythae.trainers.AdversarialTrainer method) (pythae.trainers.BaseTrainer method) (pythae.trainers.CoupledOptimizerAdversarialTrainer method) (pythae.trainers.CoupledOptimizerTrainer method)

F

FactorVAE (class in pythae.models) FactorVAEConfig (class in pythae.models) fit() (pythae.samplers.BaseSampler method) (pythae.samplers.GaussianMixtureSampler method) (pythae.samplers.IAFSampler method) (pythae.samplers.MAFSampler method) (pythae.samplers.PixelCNNSampler method) (pythae.samplers.TwoStageVAESampler method) forward() (pythae.models.Adversarial_AE method) (pythae.models.AE method) (pythae.models.BaseAE method) (pythae.models.BetaTCVAE method) (pythae.models.BetaVAE method) (pythae.models.CIWAE method) (pythae.models.DisentangledBetaVAE method) (pythae.models.FactorVAE method) (pythae.models.HVAE method) (pythae.models.INFOVAE_MMD method) (pythae.models.IWAE method) (pythae.models.MIWAE method) (pythae.models.MSSSIM_VAE method) (pythae.models.nn.BaseDecoder method) (pythae.models.nn.BaseDiscriminator method) (pythae.models.nn.BaseEncoder method) (pythae.models.nn.BaseMetric method) (pythae.models.nn.benchmarks.celeba.Decoder_Conv_AE_CELEBA method) (pythae.models.nn.benchmarks.celeba.Decoder_ResNet_AE_CELEBA method) (pythae.models.nn.benchmarks.celeba.Decoder_ResNet_VQVAE_CELEBA method) (pythae.models.nn.benchmarks.celeba.Discriminator_Conv_CELEBA method) (pythae.models.nn.benchmarks.celeba.Encoder_Conv_AE_CELEBA method) (pythae.models.nn.benchmarks.celeba.Encoder_Conv_SVAE_CELEBA method) (pythae.models.nn.benchmarks.celeba.Encoder_Conv_VAE_CELEBA method) (pythae.models.nn.benchmarks.celeba.Encoder_ResNet_AE_CELEBA method) (pythae.models.nn.benchmarks.celeba.Encoder_ResNet_SVAE_CELEBA method) (pythae.models.nn.benchmarks.celeba.Encoder_ResNet_VAE_CELEBA method) (pythae.models.nn.benchmarks.celeba.Encoder_ResNet_VQVAE_CELEBA method) (pythae.models.nn.benchmarks.cifar.Decoder_Conv_AE_CIFAR method) (pythae.models.nn.benchmarks.cifar.Decoder_ResNet_AE_CIFAR method) (pythae.models.nn.benchmarks.cifar.Decoder_ResNet_VQVAE_CIFAR method) (pythae.models.nn.benchmarks.cifar.Discriminator_Conv_CIFAR method) (pythae.models.nn.benchmarks.cifar.Encoder_Conv_AE_CIFAR method) (pythae.models.nn.benchmarks.cifar.Encoder_Conv_SVAE_CIFAR method) (pythae.models.nn.benchmarks.cifar.Encoder_Conv_VAE_CIFAR method) (pythae.models.nn.benchmarks.cifar.Encoder_ResNet_AE_CIFAR method) (pythae.models.nn.benchmarks.cifar.Encoder_ResNet_SVAE_CIFAR method) (pythae.models.nn.benchmarks.cifar.Encoder_ResNet_VAE_CIFAR method) (pythae.models.nn.benchmarks.cifar.Encoder_ResNet_VQVAE_CIFAR method) (pythae.models.nn.benchmarks.mnist.Decoder_Conv_AE_MNIST method) (pythae.models.nn.benchmarks.mnist.Decoder_ResNet_AE_MNIST method) (pythae.models.nn.benchmarks.mnist.Decoder_ResNet_VQVAE_MNIST method) (pythae.models.nn.benchmarks.mnist.Discriminator_Conv_MNIST method) (pythae.models.nn.benchmarks.mnist.Encoder_Conv_AE_MNIST method) (pythae.models.nn.benchmarks.mnist.Encoder_Conv_SVAE_MNIST method) (pythae.models.nn.benchmarks.mnist.Encoder_Conv_VAE_MNIST method) (pythae.models.nn.benchmarks.mnist.Encoder_ResNet_AE_MNIST method) (pythae.models.nn.benchmarks.mnist.Encoder_ResNet_SVAE_MNIST method) (pythae.models.nn.benchmarks.mnist.Encoder_ResNet_VAE_MNIST method) (pythae.models.nn.benchmarks.mnist.Encoder_ResNet_VQVAE_MNIST method) (pythae.models.normalizing_flows.BaseNF method) (pythae.models.normalizing_flows.IAF method) (pythae.models.normalizing_flows.MADE method) (pythae.models.normalizing_flows.MAF method) (pythae.models.normalizing_flows.PixelCNN method) (pythae.models.normalizing_flows.PlanarFlow method) (pythae.models.normalizing_flows.RadialFlow method) (pythae.models.PIWAE method) (pythae.models.PoincareVAE method) (pythae.models.RAE_GP method) (pythae.models.RAE_L2 method) (pythae.models.RHVAE method) (pythae.models.SVAE method) (pythae.models.VAE method) (pythae.models.VAE_IAF method) (pythae.models.VAE_LinNF method) (pythae.models.VAEGAN method) (pythae.models.VAMP method) (pythae.models.VQVAE method) (pythae.models.WAE_MMD method)

G

GaussianMixtureSampler (class in pythae.samplers) GaussianMixtureSamplerConfig (class in pythae.samplers) GenerationPipeline (class in pythae.pipelines) get_nll() (pythae.models.HVAE method) (pythae.models.PoincareVAE method) (pythae.models.RHVAE method) (pythae.models.SVAE method) (pythae.models.VAE method) (pythae.models.VAE_IAF method) (pythae.models.VAE_LinNF method) (pythae.models.VAMP method)

H

HVAE (class in pythae.models)

HVAEConfig (class in pythae.models) HypersphereUniformSampler (class in pythae.samplers)

I

IAF (class in pythae.models.normalizing_flows) IAFConfig (class in pythae.models.normalizing_flows) IAFSampler (class in pythae.samplers) IAFSamplerConfig (class in pythae.samplers) imq_kernel() (pythae.models.INFOVAE_MMD method) (pythae.models.WAE_MMD method) INFOVAE_MMD (class in pythae.models) INFOVAE_MMD_Config (class in pythae.models)

interpolate() (pythae.models.BaseAE method) (pythae.models.FactorVAE method) (pythae.models.PoincareVAE method) inverse() (pythae.models.normalizing_flows.BaseNF method) (pythae.models.normalizing_flows.IAF method) (pythae.models.normalizing_flows.MAF method) IWAE (class in pythae.models) IWAEConfig (class in pythae.models)

L

load_from_folder() (pythae.models.Adversarial_AE class method) (pythae.models.AutoModel class method) (pythae.models.BaseAE class method) (pythae.models.normalizing_flows.BaseNF class method) (pythae.models.RHVAE class method) (pythae.models.VAEGAN class method)

load_from_hf_hub() (pythae.models.Adversarial_AE class method) (pythae.models.AutoModel class method) (pythae.models.BaseAE class method) (pythae.models.RHVAE class method) (pythae.models.VAEGAN class method)

M

MADE (class in pythae.models.normalizing_flows) MADEConfig (class in pythae.models.normalizing_flows) MAF (class in pythae.models.normalizing_flows) MAFConfig (class in pythae.models.normalizing_flows) MAFSampler (class in pythae.samplers) MAFSamplerConfig (class in pythae.samplers) MetricConsolePrinterCallback (class in pythae.trainers.training_callbacks) MIWAE (class in pythae.models) MIWAEConfig (class in pythae.models) MLFlowCallback (class in pythae.trainers.training_callbacks) ModelOutput (class in pythae.models.base.base_utils) module pythae.models pythae.models.adversarial_ae pythae.models.ae pythae.models.auto_model pythae.models.base pythae.models.beta_tc_vae pythae.models.beta_vae pythae.models.ciwae pythae.models.disentangled_beta_vae pythae.models.factor_vae pythae.models.hvae pythae.models.info_vae pythae.models.iwae pythae.models.miwae pythae.models.msssim_vae pythae.models.nn pythae.models.nn.benchmarks, [1] pythae.models.nn.benchmarks.celeba, [1], [2] pythae.models.nn.benchmarks.cifar, [1], [2] pythae.models.nn.benchmarks.mnist, [1], [2] pythae.models.normalizing_flows pythae.models.normalizing_flows.base pythae.models.normalizing_flows.iaf pythae.models.normalizing_flows.made pythae.models.normalizing_flows.maf pythae.models.normalizing_flows.pixelcnn pythae.models.normalizing_flows.planar_flow pythae.models.normalizing_flows.radial_flow pythae.models.piwae pythae.models.pvae pythae.models.rae_gp pythae.models.rae_l2 pythae.models.rhvae pythae.models.svae pythae.models.vae pythae.models.vae_gan pythae.models.vae_iaf pythae.models.vae_lin_nf pythae.models.vamp pythae.models.vq_vae pythae.models.wae_mmd pythae.pipelines pythae.samplers pythae.samplers.base pythae.samplers.gaussian_mixture pythae.samplers.hypersphere_uniform_sampler pythae.samplers.iaf_sampler pythae.samplers.maf_sampler pythae.samplers.manifold_sampler pythae.samplers.normal_sampling pythae.samplers.pixelcnn_sampler pythae.samplers.pvae_sampler pythae.samplers.two_stage_vae_sampler pythae.samplers.vamp_sampler pythae.trainers pythae.trainers.adversarial_trainer pythae.trainers.base_trainer pythae.trainers.coupled_optimizer_adversarial_trainer pythae.trainers.coupled_optimizer_trainer pythae.trainers.training_callbacks, [1]

MSSSIM_VAE (class in pythae.models) MSSSIM_VAEConfig (class in pythae.models)

N

NormalSampler (class in pythae.samplers)

O

on_epoch_begin() (pythae.trainers.training_callbacks.TrainingCallback method) on_epoch_end() (pythae.trainers.training_callbacks.ProgressBarCallback method) (pythae.trainers.training_callbacks.TrainingCallback method) on_eval_step_begin() (pythae.trainers.training_callbacks.ProgressBarCallback method) (pythae.trainers.training_callbacks.TrainingCallback method) on_eval_step_end() (pythae.trainers.training_callbacks.ProgressBarCallback method) (pythae.trainers.training_callbacks.TrainingCallback method) on_evaluate() (pythae.trainers.training_callbacks.TrainingCallback method) on_init_end() (pythae.trainers.training_callbacks.TrainingCallback method) on_log() (pythae.trainers.training_callbacks.CometCallback method) (pythae.trainers.training_callbacks.MetricConsolePrinterCallback method) (pythae.trainers.training_callbacks.MLFlowCallback method) (pythae.trainers.training_callbacks.TrainingCallback method) (pythae.trainers.training_callbacks.WandbCallback method) on_prediction_step() (pythae.trainers.training_callbacks.CometCallback method) (pythae.trainers.training_callbacks.TrainingCallback method) (pythae.trainers.training_callbacks.WandbCallback method)

on_save() (pythae.trainers.training_callbacks.TrainingCallback method) on_train_begin() (pythae.trainers.training_callbacks.CometCallback method) (pythae.trainers.training_callbacks.MLFlowCallback method) (pythae.trainers.training_callbacks.TrainingCallback method) (pythae.trainers.training_callbacks.WandbCallback method) on_train_end() (pythae.trainers.training_callbacks.CometCallback method) (pythae.trainers.training_callbacks.MLFlowCallback method) (pythae.trainers.training_callbacks.TrainingCallback method) (pythae.trainers.training_callbacks.WandbCallback method) on_train_step_begin() (pythae.trainers.training_callbacks.ProgressBarCallback method) (pythae.trainers.training_callbacks.TrainingCallback method) on_train_step_end() (pythae.trainers.training_callbacks.ProgressBarCallback method) (pythae.trainers.training_callbacks.TrainingCallback method)

P

PIWAE (class in pythae.models) PIWAEConfig (class in pythae.models) PixelCNN (class in pythae.models.normalizing_flows) PixelCNNConfig (class in pythae.models.normalizing_flows) PixelCNNSampler (class in pythae.samplers) PixelCNNSamplerConfig (class in pythae.samplers) PlanarFlow (class in pythae.models.normalizing_flows) PlanarFlowConfig (class in pythae.models.normalizing_flows) PoincareDiskSampler (class in pythae.samplers) PoincareVAE (class in pythae.models) PoincareVAEConfig (class in pythae.models) predict() (pythae.models.BaseAE method) (pythae.models.RHVAE method) prepare_training() (pythae.trainers.AdversarialTrainer method) (pythae.trainers.BaseTrainer method) (pythae.trainers.CoupledOptimizerAdversarialTrainer method) (pythae.trainers.CoupledOptimizerTrainer method) ProgressBarCallback (class in pythae.trainers.training_callbacks) push_to_hf_hub() (pythae.models.BaseAE method) pythae.models module pythae.models.adversarial_ae module pythae.models.ae module pythae.models.auto_model module pythae.models.base module pythae.models.beta_tc_vae module pythae.models.beta_vae module pythae.models.ciwae module pythae.models.disentangled_beta_vae module pythae.models.factor_vae module pythae.models.hvae module pythae.models.info_vae module pythae.models.iwae module pythae.models.miwae module pythae.models.msssim_vae module pythae.models.nn module pythae.models.nn.benchmarks module, [1] pythae.models.nn.benchmarks.celeba module, [1], [2] pythae.models.nn.benchmarks.cifar module, [1], [2] pythae.models.nn.benchmarks.mnist module, [1], [2] pythae.models.normalizing_flows module pythae.models.normalizing_flows.base module pythae.models.normalizing_flows.iaf module pythae.models.normalizing_flows.made module pythae.models.normalizing_flows.maf module

pythae.models.normalizing_flows.pixelcnn module pythae.models.normalizing_flows.planar_flow module pythae.models.normalizing_flows.radial_flow module pythae.models.piwae module pythae.models.pvae module pythae.models.rae_gp module pythae.models.rae_l2 module pythae.models.rhvae module pythae.models.svae module pythae.models.vae module pythae.models.vae_gan module pythae.models.vae_iaf module pythae.models.vae_lin_nf module pythae.models.vamp module pythae.models.vq_vae module pythae.models.wae_mmd module pythae.pipelines module pythae.samplers module pythae.samplers.base module pythae.samplers.gaussian_mixture module pythae.samplers.hypersphere_uniform_sampler module pythae.samplers.iaf_sampler module pythae.samplers.maf_sampler module pythae.samplers.manifold_sampler module pythae.samplers.normal_sampling module pythae.samplers.pixelcnn_sampler module pythae.samplers.pvae_sampler module pythae.samplers.two_stage_vae_sampler module pythae.samplers.vamp_sampler module pythae.trainers module pythae.trainers.adversarial_trainer module pythae.trainers.base_trainer module pythae.trainers.coupled_optimizer_adversarial_trainer module pythae.trainers.coupled_optimizer_trainer module pythae.trainers.training_callbacks module, [1]

R

RadialFlow (class in pythae.models.normalizing_flows) RadialFlowConfig (class in pythae.models.normalizing_flows) RAE_GP (class in pythae.models) RAE_GP_Config (class in pythae.models) RAE_L2 (class in pythae.models) RAE_L2_Config (class in pythae.models) rbf_kernel() (pythae.models.INFOVAE_MMD method) (pythae.models.WAE_MMD method)

reconstruct() (pythae.models.BaseAE method) (pythae.models.FactorVAE method) RHVAE (class in pythae.models) RHVAEConfig (class in pythae.models) RHVAESampler (class in pythae.samplers) RHVAESamplerConfig (class in pythae.samplers)

S

sample() (pythae.samplers.BaseSampler method) (pythae.samplers.GaussianMixtureSampler method) (pythae.samplers.HypersphereUniformSampler method) (pythae.samplers.IAFSampler method) (pythae.samplers.MAFSampler method) (pythae.samplers.NormalSampler method) (pythae.samplers.PixelCNNSampler method) (pythae.samplers.PoincareDiskSampler method) (pythae.samplers.RHVAESampler method) (pythae.samplers.TwoStageVAESampler method) (pythae.samplers.VAMPSampler method) save() (pythae.models.Adversarial_AE method) (pythae.models.BaseAE method) (pythae.models.normalizing_flows.BaseNF method) (pythae.models.RHVAE method) (pythae.models.VAEGAN method) (pythae.samplers.BaseSampler method)

save_checkpoint() (pythae.trainers.AdversarialTrainer method) (pythae.trainers.BaseTrainer method) (pythae.trainers.CoupledOptimizerAdversarialTrainer method) (pythae.trainers.CoupledOptimizerTrainer method) save_img() (pythae.samplers.BaseSampler method) save_model() (pythae.trainers.BaseTrainer method) set_decoder() (pythae.models.BaseAE method) set_discriminator() (pythae.models.Adversarial_AE method) (pythae.models.FactorVAE method) (pythae.models.VAEGAN method) set_encoder() (pythae.models.BaseAE method) set_metric() (pythae.models.RHVAE method) setup() (pythae.trainers.training_callbacks.CometCallback method) (pythae.trainers.training_callbacks.MLFlowCallback method) (pythae.trainers.training_callbacks.WandbCallback method) SVAE (class in pythae.models) SVAEConfig (class in pythae.models)

T

to_tuple() (pythae.models.base.base_utils.ModelOutput method) train() (pythae.trainers.AdversarialTrainer method) (pythae.trainers.BaseTrainer method) (pythae.trainers.CoupledOptimizerAdversarialTrainer method) (pythae.trainers.CoupledOptimizerTrainer method) train_step() (pythae.trainers.AdversarialTrainer method) (pythae.trainers.BaseTrainer method) (pythae.trainers.CoupledOptimizerAdversarialTrainer method) (pythae.trainers.CoupledOptimizerTrainer method)

TrainingCallback (class in pythae.trainers.training_callbacks) TrainingPipeline (class in pythae.pipelines) TwoStageVAESampler (class in pythae.samplers) TwoStageVAESamplerConfig (class in pythae.samplers)

U

update() (pythae.models.BaseAE method) (pythae.models.normalizing_flows.BaseNF method) (pythae.models.RHVAE method)

V

VAE (class in pythae.models) VAE_IAF (class in pythae.models) VAE_IAF_Config (class in pythae.models) VAE_LinNF (class in pythae.models) VAE_LinNF_Config (class in pythae.models) VAEConfig (class in pythae.models)

VAEGAN (class in pythae.models) VAEGANConfig (class in pythae.models) VAMP (class in pythae.models) VAMPConfig (class in pythae.models) VAMPSampler (class in pythae.samplers) VQVAE (class in pythae.models) VQVAEConfig (class in pythae.models)

W

WAE_MMD (class in pythae.models)

WAE_MMD_Config (class in pythae.models) WandbCallback (class in pythae.trainers.training_callbacks)