Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Porting to Tensorflow / Keras. Help Needed #18

Open
SarthakYadav opened this issue Jul 20, 2017 · 2 comments
Open

Porting to Tensorflow / Keras. Help Needed #18

SarthakYadav opened this issue Jul 20, 2017 · 2 comments

Comments

@SarthakYadav
Copy link

Hello

I am trying to reproduce the results of this paper using Keras. I have implemented the loss function as seen in the paper, but the accuracy of the classifier never goes above 10%.
To test the loss implementation, I am using a tiny CNN trained from scratch. The model structure is as seen in the image attached. I have also tried building on top of a pre-trained VGG16, but the results are the same.
model
As seen from the name, the second last layer called "latent_48bit" is the latent layer with the sigmoid activation. The final layer is the softmax classifier.

Is the model architecture correct?

Also, can you help me with the loss function? I believe that's where I am going wrong.

@kevinlin311tw
Copy link
Owner

kevinlin311tw commented Jul 25, 2017 via email

@SarthakYadav
Copy link
Author

Trains fine with Softmax!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants