We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0e4bd20 commit 7bdd1efCopy full SHA for 7bdd1ef
channel_attention/pyramid_squeeze_attention.py
@@ -0,0 +1,17 @@
1
+import torch
2
+from torch import nn
3
+
4
5
+class PyramidSqueezeAttention(nn.Module):
6
+ """
7
+ The Pyramid Squeeze Attention Module for Time Series (1D) or Image (2D) Analysis.
8
+ This module implements a pyramid squeeze-and-excitation attention mechanism.
9
10
+ References: "EPSANet: An Efficient Pyramid Squeeze Attention Block on Convolutional Neural Network" by Hu Zhang, et al.
11
12
+ URL: https://arxiv.org/abs/2105.14447
13
14
15
+ def __init__(self, *args, **kwargs):
16
+ super().__init__(*args, **kwargs)
17
+ pass
0 commit comments