mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix module name mistake, start layers sub-package

Browse Source
This commit is contained in:
Ross Wightman 2020-02-09 12:44:26 -08:00
parent 7011cd0902
commit 4defbbbaa8
3 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
timm/models/layers/__init__.py Normal file
View File

@ -0,0 +1 @@
from .eca import EcaModule, CecaModule

0
timm/models/EcaModule.py → timm/models/layers/eca.py
View File

4
timm/models/resnet.py
View File

@ -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