tiny_diff.losses.vae_loss

Classes

BetaPolicy([a, b, offset, slope])	Function representing a Beta Policy.
VAELoss([recon_l, kl_w_policy, kl_weight])	Loss for VAE models.