MCPcopy Create free account

hub / github.com/AntixK/PyTorch-VAE / functions

Functions334 in github.com/AntixK/PyTorch-VAE

Methodloss_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
Methodloss_function
(self, *args, **kwargs)
models/wae_mmd.py:107
Methodloss_function
(self, *args, **kwargs)
models/beta_vae.py:129
Methodloss_function
(self, *args, **kwargs)
models/cvae.py:133
Methodloss_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
Methodloss_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
Methodloss_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
Methodloss_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
Methodloss_function
(self, *args, **kwargs)
models/vampvae.py:123
Methodloss_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
Methodloss_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
Methodloss_function
(self, *args, **kwargs)
models/gamma_vae.py:184
Methodloss_function
:param args: :param kwargs: :return:
models/vq_vae.py:194
Methodon_validation_end
(self)
experiment.py:61
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/miwae.py:166
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/dfcvae.py:192
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/logcosh_vae.py:157
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/cat_vae.py:179
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/info_vae.py:231
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/vanilla_vae.py:148
Methodsample
(self, batch_size:int, current_device: int, **kwargs)
models/base.py:16
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/lvae.py:239
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/dip_vae.py:166
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/wae_mmd.py:206
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/beta_vae.py:154
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/cvae.py:149
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/joint_vae.py:236
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/betatc_vae.py:213
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/twostage_vae.py:172
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/mssim_vae.py:155
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/vampvae.py:170
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/fvae.py:203
Methodsample
Samples from the latent space and return the corresponding image space map. :param num_samples: (Int) Number of samples
models/iwae.py:162
Methodsample
(self, num_samples: int, current_device: Union[int, str], **kwargs)
models/vq_vae.py:213
MethodsetUp
(self)
tests/test_dipvae.py:9
MethodsetUp
(self)
tests/bvae.py:9
MethodsetUp
(self)
tests/test_swae.py:9
MethodsetUp
(self)
tests/test_joint_Vae.py:9
MethodsetUp
(self)
tests/test_wae.py:9
MethodsetUp
(self)
tests/text_vamp.py:9
MethodsetUp
(self)
tests/test_vae.py:9
MethodsetUp
(self)
tests/test_lvae.py:9
MethodsetUp
(self)
tests/test_vq_vae.py:9
MethodsetUp
(self)
tests/test_logcosh.py:9
MethodsetUp
(self)
tests/test_miwae.py:9
MethodsetUp
(self)
tests/test_fvae.py:9
MethodsetUp
(self)
tests/test_hvae.py:9
MethodsetUp
(self)
tests/test_mssimvae.py:9
MethodsetUp
(self)
tests/test_cat_vae.py:9
MethodsetUp
(self)
tests/test_betatcvae.py:9
MethodsetUp
(self)
tests/test_dfc.py:9
MethodsetUp
(self)
tests/test_gvae.py:9
MethodsetUp
(self)
tests/test_iwae.py:9
MethodsetUp
(self)
tests/text_cvae.py:8
Methodtest_forward
(self)
tests/test_dipvae.py:17
Methodtest_forward
(self)
tests/bvae.py:17
Methodtest_forward
(self)
tests/test_swae.py:16
Methodtest_forward
(self)
tests/test_joint_Vae.py:17
Methodtest_forward
(self)
tests/test_wae.py:16
Methodtest_forward
(self)
tests/text_vamp.py:17
Methodtest_forward
(self)
tests/test_vae.py:17
Methodtest_forward
(self)
tests/test_lvae.py:17
Methodtest_forward
(self)
tests/test_vq_vae.py:17
Methodtest_forward
(self)
tests/test_logcosh.py:17
Methodtest_forward
(self)
tests/test_miwae.py:17
Methodtest_forward
(self)
tests/test_fvae.py:20
Methodtest_forward
(self)
tests/test_hvae.py:17
Methodtest_forward
(self)
tests/test_mssimvae.py:18
Methodtest_forward
(self)
tests/test_cat_vae.py:17
Methodtest_forward
(self)
tests/test_betatcvae.py:17
Methodtest_forward
(self)
tests/test_dfc.py:18
Methodtest_forward
(self)
tests/test_gvae.py:18
Methodtest_forward
(self)
tests/test_iwae.py:17
Methodtest_forward
(self)
tests/text_cvae.py:12
Methodtest_generate
(self)
tests/test_dipvae.py:36
Methodtest_generate
(self)
tests/test_vq_vae.py:36
Methodtest_generate
(self)
tests/test_miwae.py:35
Methodtest_generate
(self)
tests/test_betatcvae.py:36
Methodtest_loss
(self)
tests/test_dipvae.py:24
Methodtest_loss
(self)
tests/bvae.py:23
Methodtest_loss
(self)
tests/test_swae.py:22
Methodtest_loss
(self)
tests/test_joint_Vae.py:23
Methodtest_loss
(self)
tests/test_wae.py:22
Methodtest_loss
(self)
tests/text_vamp.py:23
Methodtest_loss
(self)
tests/test_vae.py:23
Methodtest_loss
(self)
tests/test_lvae.py:24
Methodtest_loss
(self)
tests/test_vq_vae.py:24
Methodtest_loss
(self)
tests/test_logcosh.py:23
Methodtest_loss
(self)
tests/test_miwae.py:23
Methodtest_loss
(self)
tests/test_fvae.py:27
Methodtest_loss
(self)
tests/test_hvae.py:23
Methodtest_loss
(self)
tests/test_mssimvae.py:25
Methodtest_loss
(self)
tests/test_cat_vae.py:23
Methodtest_loss
(self)
tests/test_betatcvae.py:24
Methodtest_loss
(self)
tests/test_dfc.py:25
Methodtest_loss
(self)
tests/test_gvae.py:25
Methodtest_loss
(self)
tests/test_iwae.py:23
Methodtest_loss
(self)
tests/text_cvae.py:19
Methodtest_optim
(self)
tests/test_fvae.py:36
Methodtest_sample
(self)
tests/test_dipvae.py:31
← previousnext →201–300 of 334, ranked by callers