VAE

Scripts to use Variational Autoencoders

---

source

AbstractVAE

 AbstractVAE (seq_len:int, feat_dim:int, latent_dim:int)

Helper class that provides a standard way to create an ABC using inheritance.

---

source

BetaVAE

 BetaVAE (encoder:torch.nn.modules.module.Module,
          decoder:torch.nn.modules.module.Module, beta:float=1.0,
          loss_fn=None, optimizer_cls=<class 'torch.optim.adam.Adam'>,
          lr:float=0.001, **kwargs)

Hooks to be used in LightningModule.

---

source

cBetaVAE

 cBetaVAE (encoder:torch.nn.modules.module.Module,
           decoder:torch.nn.modules.module.Module, beta:float=1.0,
           loss_fn=None, optimizer_cls=<class 'torch.optim.adam.Adam'>,
           lr:float=0.001, **kwargs)

Hooks to be used in LightningModule.