Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/AntixK/PyTorch-VAE
/ functions
Functions
334 in github.com/AntixK/PyTorch-VAE
⨍
Functions
334
◇
Types & classes
53
Method
loss_function
Computes the VAE loss function. KL(N(\mu, \sigma), N(0, 1)) = \log \frac{1}{\sigma} + \frac{\sigma^2 + \mu^2}{2} - \frac{1}{2}
models/dip_vae.py:125
Method
loss_function
(self, *args, **kwargs)
models/wae_mmd.py:107
Method
loss_function
(self, *args, **kwargs)
models/beta_vae.py:129
Method
loss_function
(self, *args, **kwargs)
models/cvae.py:133
Method
loss_function
Computes the VAE loss function. KL(N(\mu, \sigma), N(0, 1)) = \log \frac{1}{\sigma} + \frac{\sigma^2 + \mu^2}{2} - \frac{1}{2}
models/joint_vae.py:175
Method
loss_function
Computes the VAE loss function. KL(N(\mu, \sigma), N(0, 1)) = \log \frac{1}{\sigma} + \frac{\sigma^2 + \mu^2}{2} - \frac{1}{2}
models/betatc_vae.py:144
Method
loss_function
Computes the VAE loss function. KL(N(\mu, \sigma), N(0, 1)) = \log \frac{1}{\sigma} + \frac{\sigma^2 + \mu^2}{2} - \frac{1}{2}
models/twostage_vae.py:148
Method
loss_function
Computes the VAE loss function. KL(N(\mu, \sigma), N(0, 1)) = \log \frac{1}{\sigma} + \frac{\sigma^2 + \mu^2}{2} - \frac{1}{2}
models/mssim_vae.py:131
Method
loss_function
(self, *args, **kwargs)
models/vampvae.py:123
Method
loss_function
Computes the VAE loss function. KL(N(\mu, \sigma), N(0, 1)) = \log \frac{1}{\sigma} + \frac{\sigma^2 + \mu^2}{2} - \frac{1}{2}
models/fvae.py:151
Method
loss_function
KL(N(\mu, \sigma), N(0, 1)) = \log \frac{1}{\sigma} + \frac{\sigma^2 + \mu^2}{2} - \frac{1}{2} :param args: :param kwargs:
models/iwae.py:129
Method
loss_function
(self, *args, **kwargs)
models/gamma_vae.py:184
Method
loss_function
:param args: :param kwargs: :return:
models/vq_vae.py:194
Method
on_validation_end
(self)
experiment.py:61
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/miwae.py:166
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/dfcvae.py:192
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/logcosh_vae.py:157
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/cat_vae.py:179
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/info_vae.py:231
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/vanilla_vae.py:148
Method
sample
(self, batch_size:int, current_device: int, **kwargs)
models/base.py:16
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/lvae.py:239
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/dip_vae.py:166
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/wae_mmd.py:206
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/beta_vae.py:154
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/cvae.py:149
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/joint_vae.py:236
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/betatc_vae.py:213
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/twostage_vae.py:172
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/mssim_vae.py:155
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/vampvae.py:170
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/fvae.py:203
Method
sample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/iwae.py:162
Method
sample
(self, num_samples: int, current_device: Union[int, str], **kwargs)
models/vq_vae.py:213
Method
setUp
(self)
tests/test_dipvae.py:9
Method
setUp
(self)
tests/bvae.py:9
Method
setUp
(self)
tests/test_swae.py:9
Method
setUp
(self)
tests/test_joint_Vae.py:9
Method
setUp
(self)
tests/test_wae.py:9
Method
setUp
(self)
tests/text_vamp.py:9
Method
setUp
(self)
tests/test_vae.py:9
Method
setUp
(self)
tests/test_lvae.py:9
Method
setUp
(self)
tests/test_vq_vae.py:9
Method
setUp
(self)
tests/test_logcosh.py:9
Method
setUp
(self)
tests/test_miwae.py:9
Method
setUp
(self)
tests/test_fvae.py:9
Method
setUp
(self)
tests/test_hvae.py:9
Method
setUp
(self)
tests/test_mssimvae.py:9
Method
setUp
(self)
tests/test_cat_vae.py:9
Method
setUp
(self)
tests/test_betatcvae.py:9
Method
setUp
(self)
tests/test_dfc.py:9
Method
setUp
(self)
tests/test_gvae.py:9
Method
setUp
(self)
tests/test_iwae.py:9
Method
setUp
(self)
tests/text_cvae.py:8
Method
test_forward
(self)
tests/test_dipvae.py:17
Method
test_forward
(self)
tests/bvae.py:17
Method
test_forward
(self)
tests/test_swae.py:16
Method
test_forward
(self)
tests/test_joint_Vae.py:17
Method
test_forward
(self)
tests/test_wae.py:16
Method
test_forward
(self)
tests/text_vamp.py:17
Method
test_forward
(self)
tests/test_vae.py:17
Method
test_forward
(self)
tests/test_lvae.py:17
Method
test_forward
(self)
tests/test_vq_vae.py:17
Method
test_forward
(self)
tests/test_logcosh.py:17
Method
test_forward
(self)
tests/test_miwae.py:17
Method
test_forward
(self)
tests/test_fvae.py:20
Method
test_forward
(self)
tests/test_hvae.py:17
Method
test_forward
(self)
tests/test_mssimvae.py:18
Method
test_forward
(self)
tests/test_cat_vae.py:17
Method
test_forward
(self)
tests/test_betatcvae.py:17
Method
test_forward
(self)
tests/test_dfc.py:18
Method
test_forward
(self)
tests/test_gvae.py:18
Method
test_forward
(self)
tests/test_iwae.py:17
Method
test_forward
(self)
tests/text_cvae.py:12
Method
test_generate
(self)
tests/test_dipvae.py:36
Method
test_generate
(self)
tests/test_vq_vae.py:36
Method
test_generate
(self)
tests/test_miwae.py:35
Method
test_generate
(self)
tests/test_betatcvae.py:36
Method
test_loss
(self)
tests/test_dipvae.py:24
Method
test_loss
(self)
tests/bvae.py:23
Method
test_loss
(self)
tests/test_swae.py:22
Method
test_loss
(self)
tests/test_joint_Vae.py:23
Method
test_loss
(self)
tests/test_wae.py:22
Method
test_loss
(self)
tests/text_vamp.py:23
Method
test_loss
(self)
tests/test_vae.py:23
Method
test_loss
(self)
tests/test_lvae.py:24
Method
test_loss
(self)
tests/test_vq_vae.py:24
Method
test_loss
(self)
tests/test_logcosh.py:23
Method
test_loss
(self)
tests/test_miwae.py:23
Method
test_loss
(self)
tests/test_fvae.py:27
Method
test_loss
(self)
tests/test_hvae.py:23
Method
test_loss
(self)
tests/test_mssimvae.py:25
Method
test_loss
(self)
tests/test_cat_vae.py:23
Method
test_loss
(self)
tests/test_betatcvae.py:24
Method
test_loss
(self)
tests/test_dfc.py:25
Method
test_loss
(self)
tests/test_gvae.py:25
Method
test_loss
(self)
tests/test_iwae.py:23
Method
test_loss
(self)
tests/text_cvae.py:19
Method
test_optim
(self)
tests/test_fvae.py:36
Method
test_sample
(self)
tests/test_dipvae.py:31
← previous
next →
201–300 of 334, ranked by callers