I get nan while training

kuanghuei / SCAN

PyTorch source code for "Stacked Cross Attention for Image-Text Matching" (ECCV 2018)

Apache License 2.0

549 stars 113 forks source link

Closed dfatimazohra closed 5 years ago

dfatimazohra commented 5 years ago

Hello, I am training the model without changing any parameters. but I get "nan" while trainig: error_SCAN

Do you know what is causing that? Thank you.

kuanghuei commented 5 years ago

Hi, I am not exactly sure what is the cause. I personally never encounter this problem. Would you mind to provide more information?

dfatimazohra commented 5 years ago

Thanks for your reply. I think Pytorch version was causing that. I'm not getting this issue anymore.

kuanghuei commented 5 years ago

Good to hear that it's solved. Wondering which version of Pytorch you were using?

dfatimazohra commented 5 years ago

I tried using both Pytorch 0.4 and 1.1

kuanghuei commented 5 years ago

If I remember correctly, pytorch 0.4 also worked for me.