From 4defbbbaa89049f1805951cda9db15f832c32695 Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Sun, 9 Feb 2020 12:44:26 -0800
Subject: [PATCH] Fix module name mistake, start layers sub-package

---
 timm/models/layers/__init__.py              | 1 +
 timm/models/{EcaModule.py => layers/eca.py} | 0
 timm/models/resnet.py                       | 4 ++--
 3 files changed, 3 insertions(+), 2 deletions(-)
 create mode 100644 timm/models/layers/__init__.py
 rename timm/models/{EcaModule.py => layers/eca.py} (100%)

diff --git a/timm/models/layers/__init__.py b/timm/models/layers/__init__.py
new file mode 100644
index 00000000..325516e9
--- /dev/null
+++ b/timm/models/layers/__init__.py
@@ -0,0 +1 @@
+from .eca import EcaModule, CecaModule
diff --git a/timm/models/EcaModule.py b/timm/models/layers/eca.py
similarity index 100%
rename from timm/models/EcaModule.py
rename to timm/models/layers/eca.py
diff --git a/timm/models/resnet.py b/timm/models/resnet.py
index da755373..893350ef 100644
--- a/timm/models/resnet.py
+++ b/timm/models/resnet.py
@@ -14,7 +14,7 @@ import torch.nn.functional as F
 from .registry import register_model
 from .helpers import load_pretrained
 from .adaptive_avgmax_pool import SelectAdaptivePool2d
-from .EcaModule import EcaModule
+from .layers import EcaModule
 from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
 
 
@@ -212,7 +212,7 @@ class Bottleneck(nn.Module):
         self.bn3 = norm_layer(outplanes)
 
         self.se = SEModule(outplanes, planes // 4) if use_se else None
-        self.eca = Eca_Module(outplanes) if use_eca else None
+        self.eca = EcaModule(outplanes) if use_eca else None
     
         self.act3 = act_layer(inplace=True)
         self.downsample = downsample