search

wkcn / AttentionSampler

Attention-based sampler in TASN (Trilinear Attention Sampling Network)
23 stars 0 forks source link

Question on AttSampler #2

Closed AlexHu123 closed 5 years ago

AlexHu123 commented 5 years ago

Hi, will you release the PyTorch version? And can your op support the backpropogation in the loss update ?

wkcn commented 5 years ago

The PyTorch version has been released. https://github.com/wkcn/AttentionSampler

AlexHu123 commented 5 years ago

Thanks for your kind support !