[Testing on attsampler on pytorch]

wkcn / AttentionSampler

Attention-based sampler in TASN (Trilinear Attention Sampling Network)

23 stars 0 forks source link

[Testing on attsampler on pytorch] #4

Closed AlexHu123 closed 5 years ago

AlexHu123 commented 5 years ago

Hi, when I test on the attention sampler using Pytorch, I found some bugs when it creates gird for sampling ( it has 'nan' in grid ). Would it be possible for you to fix it out? Thanks in advance!

wkcn commented 5 years ago

It is not the bug of AttSampler Operator. I have tested the AttSampler in official MXNet implementation, and got 87 accuracy on CUB_200_2011.

AlexHu123 commented 5 years ago

Thanks for your reply, I will check it