corentin/torch_utils
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Several improvement

Browse source
This commit is contained in:
Corentin Risselin 2020-04-13 17:30:35 +09:00
commit 12afc7cc93
2 changed files with 16 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
layers.py
View file

@ -48,7 +48,7 @@ class Layer(nn.Module):
class Conv2d(Layer):
def __init__(self, in_channels: int, out_channels: int, kernel_size: int, stride: int = 1,
def __init__(self, in_channels: int, out_channels: int, kernel_size: int = 3, stride: int = 1,
activation=0, batch_norm=None, **kwargs):
super(Conv2d, self).__init__(activation, batch_norm)