abhimanyudubey / confusion

Code for the ECCV 2018 paper "Pairwise Confusion for Fine-Grained Visual Classification"
https://arxiv.org/abs/1705.08016
200 stars 39 forks source link

How to avoid log zero in EntropicConfusion ? #12

Open QtacierP opened 5 years ago

QtacierP commented 5 years ago

Thanks for your great paper and related codes. However, I still met some problems in my own implementation with Keras. How to initialize weights better? I mean that if I use my default setting from your PyTorch codes, the feature map in EntropicConfusion hasn't been normalized, and the Log will get a NAN problem. I think I should add an extra normalization?