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
SOFTWARE.
'''
import torch
from torch import nn
from torch.nn.parameter import Parameter
import torch.nn.functional as F
class EcaModule(nn.Module):
"""Constructs a ECA module.
@ -100,7 +98,7 @@ class CecaModule(nn.Module):
#pytorch circular padding mode is bugged as of pytorch 1.4
# see https://github.com/pytorch/pytorch/pull/17240
#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.sigmoid = nn.Sigmoid()
@ -109,7 +107,7 @@ class CecaModule(nn.Module):
y = self.avg_pool(x)
#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
y = self.conv(y)

Loading…
Cancel
Save