Update EcaModule.py

Make pylint happy
(commas, unused imports, missed imports)
pull/82/head
Chris Ha 4 years ago
parent db91ba053b
commit 904c618040

@ -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.
@ -100,7 +98,7 @@ class CecaModule(nn.Module):
#pytorch circular padding mode is bugged as of pytorch 1.4 #pytorch circular padding mode is bugged as of pytorch 1.4
# see https://github.com/pytorch/pytorch/pull/17240 # see https://github.com/pytorch/pytorch/pull/17240
#implement manual circular padding #implement manual circular padding
self.conv = nn.Conv1d(1, 1, kernel_size=k_size, padding = 0, bias=False) self.conv = nn.Conv1d(1, 1, kernel_size=k_size, padding=0, bias=False)
self.padding = (k_size - 1) // 2 self.padding = (k_size - 1) // 2
self.sigmoid = nn.Sigmoid() self.sigmoid = nn.Sigmoid()
@ -109,7 +107,7 @@ class CecaModule(nn.Module):
y = self.avg_pool(x) y = self.avg_pool(x)
#manually implement circular padding, F.pad does not seemed to be bugged #manually implement circular padding, F.pad does not seemed to be bugged
y = F.pad(y.view(x.shape[0],1,-1),(self.padding,self.padding),mode='circular') y = F.pad(y.view(x.shape[0], 1, -1), (self.padding, self.padding), mode='circular')
# Two different branches of ECA module # Two different branches of ECA module
y = self.conv(y) y = self.conv(y)

Loading…
Cancel
Save