Adapt to convolutional networks

[JakobKHAndersen]

Hi

Thank for making your code availiable. I'm trying to adapt the model into a convolutional network but i'm failing due to shape mismatching. Have you tried to do this?

BR.

[divamgupta]

Hi,

Could you provide more details on what you are trying to do?
I have not tested the model with a CNN but it should work.

[JakobKHAndersen]

Basically, I'm trying to replace the fully connected (Dense) layers in the encoder with Conv2D layers in order to classify (28,28,1) input volumes as opposed to (784,) inputs. Hope it makes sense.