tiny_diff.models.vae.gan.VAEGANAdapt

class tiny_diff.models.vae.gan.VAEGANAdapt(**kwargs)

Bases: VAEGAN

VAEGAN with an adaptative weight.

__init__(**kwargs)

Methods

__init__(**kwargs)
adaptative_weight(recon_l, disc_l, layer[, ...])	Adaptative weight used in Taming Transformers paper.
disc_loss_weight(loss, epoch)	Value used to weight the gan loss.
gen_loss(x[, epoch])	Generator loss.
loss(x[, epoch, mode])	Computes the VAEGAN loss.