Varitational Autoencoder

[mayurat22 (Mayur jain)]

I am trying to build a Variational Autoencoder from Scratch and stuck at a problem with dimensions.

Below Architecture:

Variational Autoencoder Class:

Issue:

I am not sure why decoder is expecting (64, 64, 3, 3) ?
Help me out

[Usama_Hasan]

Hy @mayurat22 Can you post the code of your training section also.

[suraj.pt (Suraj)]

Can you try reshaping with batch size = 1? Like
x = x.reshape((1,64,7,7))

[mayurat22 (Mayur jain)]

[mayurat22 (Mayur jain)]

It worked. Thank you.