I am trying to build a Variational Autoencoder from Scratch and stuck at a problem with dimensions.
Below Architecture:
image812×283 40 KB
Variational Autoencoder Class:
image1111×774 127 KB
Issue:
image1280×807 107 KB
I am not sure why decoder is expecting (64, 64, 3, 3) ?
Help me out