search

AntixK / PyTorch-VAE

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

Generalise hard coded values #52

Open rishabh-ranjan opened 2 years ago

rishabh-ranjan commented 2 years ago

The code fails for general values of hidden dims, image sizes, etc. due to shape mismatches. Since image size and hidden dims are parameters anyway, can you please increase the flexibility in other parts of the code? This will allow it to be used as it is on other datasets and different architectures.

PratikStar commented 2 years ago

I can try to do this when I get some time :)

clementchadebec commented 1 year ago

Not sure it is still relevant to you but I have put together some Implementations of VAE variants that allow such flexibility here. Maybe this can help.