Dynamic batch size in Beta-tcvae

[OneDirection9]

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