Difference between MNIST_CNN.h5 and MNIST_CNN_tanh.h5
herrkun opened this issue · 1 comments
herrkun commented
What's the mainly difference between MNIST_CNN.h5 and MNIST_CNN_tanh.h5 ? just finally activation layer using tanh or others ? It will be perfect as showing the source code of these two models.
aadeshnpn commented
@herrkun . The only difference is the use of ReLU and tanh activation in the final layers. I tried finding the source code for the models I trained. As far as I remember, I was using the basic architecture called LeCun Net (the network from the MNIST paper). I have updated the repo with steps to use the existing code with your own model and data. Hope it helps.