Open stepjam opened 7 years ago
Ah, turns out I don't actually need the Gaussian posterior, but instead a deterministic one. So this isn't a problem for me anymore. But I guess it still handy for you to know about the problem for future users.
Edit: Also, out of interest, have you tried your solution for AAE with rgb images (e.g. cifar10) and a conv-net for the encoder?
@stepjam Thanks! I wouldn't have noticed that for a while. I need to change that to K.shape(p[0])
instead of p[0].shape
and it'll work on both backends.
What are you trying to build? Do you mean discrete? Not sure what a deterministic posterior would look like.
Haven't written an AAE CNN example yet but I assume it would have some better performance. Just need to swap out the models in the AAE example.
Cheers
Just for comparison when you get around to it, here is a sample my trained autoencoder output after 100 epochs on the cifar10 dataset:
(Input image on the left) This was using a conv-net for the encoder, and a de-conv for the generator/decoder.
@stepjam fixed the example and added an example for AAE cifar10. It runs but I need to pick some better parameters to make it look good.
It is trivially easy to make an AAE that autoencodes well. An autoencoder does as much. The harder part is making the generated samples also look good. I'll let you know when I clean up the example.
Hi again,
When running the
example_aae.py
, I get the following error: I am running tensorflow as the backend