Attention-Augmented-Conv2d
Attention-Augmented-Conv2d copied to clipboard
device typo
typo in readme
also I get error
RuntimeError: shape '[-1, 64, 16, 31]' is invalid for input of size 6451200
when I use AugmentedConv
also is it possible to add pip install?