|
|
@ -31,10 +31,8 @@ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
|
|
SOFTWARE.
|
|
|
|
SOFTWARE.
|
|
|
|
'''
|
|
|
|
'''
|
|
|
|
import torch
|
|
|
|
|
|
|
|
from torch import nn
|
|
|
|
from torch import nn
|
|
|
|
from torch.nn.parameter import Parameter
|
|
|
|
import torch.nn.functional as F
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class EcaModule(nn.Module):
|
|
|
|
class EcaModule(nn.Module):
|
|
|
|
"""Constructs a ECA module.
|
|
|
|
"""Constructs a ECA module.
|
|
|
|