SHI-Labs/Neighborhood-Attention-Transformer

Tiny Bug in nattencuda.py

z-jiaming opened this issue · 1 comments

z-jiaming commented

Great Work!

Well, I found a small bug in nattencuda.py.

Neighborhood-Attention-Transformer/natten/nattencuda.py

Line 117 in a9a7580

pad_r = max(0, self.window_size - W)

it should be self.kernel_size during padding if feature size is small than kernel_size

alihassanijr commented

Thank you for your interest, and for bringing this to our attention.