hszhao / SAN

Exploring Self-attention for Image Recognition, CVPR2020.
MIT License
747 stars 133 forks source link

Enquiry regarding load_kernel #9

Open mckhang opened 4 years ago

mckhang commented 4 years ago

Hello Dr. Zhao, I was studying your paper and the code shared on the page but had difficulty understanding the "load_kernel()" and "f(block=...., grid=(...., args=[..., stream=...)" parts in subtraction/subtraction2/aggregate functions. How do they affect the subtraction relation function and aggregation function and will the results differ without them?

Any help would be appreciated. Thank you