Thank you for your response.
Do you think I should increase the number of channels or have additional convolutions.
Also, why should mu and logvar be generated with linear layers and not convolutions?