search

AntixK / PyTorch-VAE

A Collection of Variational Autoencoders (VAE) in PyTorch.
Apache License 2.0
6.46k stars 1.05k forks source link

Dynamic batch size in Beta-tcvae #36

Open OneDirection9 opened 3 years ago

OneDirection9 commented 3 years ago

Hi nice works,

I use dynamic batch size in my training, is it ok to use dynamic batch size to train beta-tcvae, as start_weight calculating depends on batch_size https://github.com/AntixK/PyTorch-VAE/blob/8700d245a9735640dda458db4cf40708caf2e77f/models/betatc_vae.py#L177