This commit is contained in:
Colombo 2020-07-16 23:39:56 +04:00
parent 6c7bb74ad4
commit dd21880ecd

View file

@ -21,7 +21,7 @@ class Conv2D(nn.LayerBase):
if isinstance(padding, str):
if padding == "SAME":
padding = int( ( (kernel_size - 1) * dilations + 1 ) / 2.5 )
padding = ( (kernel_size - 1) * dilations + 1 ) // 2
elif padding == "VALID":
padding = 0
else: