Closed dipayandas97 closed 4 years ago
The issue was solved. The arguments of Flatten(11128) needs to be adjusted according to input image size.
Hello @dipayandas97,
I am also trying to use this for a binary classification. As I am very new to deep learning I am a bit confused and was hoping you should share what you changed the arguments of flatten to?
Thanks in advance
Congratulations on this wonderful achievement of the paper. I am trying to implement the Bayesian CNN for a project of mine, a simple binary classification. Here is the code to define the model:
However, when i input a random input like,
the ouput has a shape (4,2) whereas it should be (1,2). The peculiarity seems like the model outputs (1,2) only when it gets an input of shape (1,32,32), like the MNIST dataset. However I could not find where this fixation of 32x32 image size is done.
Looking forward to your support! Thanks