kevinzakka / recurrent-visual-attention

A PyTorch Implementation of "Recurrent Models of Visual Attention"
MIT License
468 stars 123 forks source link

nan loss whn locations are provided #36

Open Deccan12 opened 3 years ago

Deccan12 commented 3 years ago

I am providing locations of the important landmarks for image classification problem I am solving, but when these 2 dimensions are passed for each image, instead of loss being decreased, loss is becoming nan along with accuracy. Is it expected to normalise the location pixels?