SqueezeAttention-PyTorch icon indicating copy to clipboard operation
SqueezeAttention-PyTorch copied to clipboard

It has mine implementation of Squeeze and Attention Networks in PyTorch

SqueezeAttention-PyTorch

It has my implementation of Squeeze and Attention Networks in PyTorch which are defined in Zhong et al. https://arxiv.org/abs/1909.03402

Feel Free to open any issue if you see anything wrong or any improvement!