from local

This commit is contained in:
dohe0342 2023-01-03 13:32:31 +09:00
parent a895d5b2f9
commit a433f440bb
3 changed files with 1 additions and 1 deletions

View File

@ -19,7 +19,7 @@ class ConvolutionModule(nn.Module):
"""
def __init__(self, channels, kernel_size, activation=nn.ReLU(), bias=True):
def __init__(self, channels, kernel_size, activation=nn.SiLU(), bias=True):
"""Construct an ConvolutionModule object."""
super(ConvolutionModule, self).__init__()
# kernerl_size should be a odd number for 'SAME' padding