from local
This commit is contained in:
parent
a895d5b2f9
commit
a433f440bb
Binary file not shown.
Binary file not shown.
@ -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."""
|
"""Construct an ConvolutionModule object."""
|
||||||
super(ConvolutionModule, self).__init__()
|
super(ConvolutionModule, self).__init__()
|
||||||
# kernerl_size should be a odd number for 'SAME' padding
|
# kernerl_size should be a odd number for 'SAME' padding
|
||||||
|
|||||||
Reference in New Issue
Block a user