Tiny Bug in nattencuda.py

[Jiaming Zhang]

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

[Ali Hassani]

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