titu1994 / keras-attention-augmented-convs

Keras implementation of Attention Augmented Convolutional Neural Networks
MIT License
120 stars 38 forks source link

compute_qkv convolution should be pointwise #3

Closed squarethecircle closed 5 years ago

squarethecircle commented 5 years ago

This is consistent with the paper, and necessary to achieve the desired parameter reduction.

titu1994 commented 5 years ago

Thanks for the fix !