Skip to content

Commit

Permalink
fixed padding
Browse files Browse the repository at this point in the history
fixed issue with padding not making it through (which was only present when training a single net)
  • Loading branch information
mmirman committed Jul 30, 2018
1 parent 460c919 commit 409a3cc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion components.py
Expand Up @@ -231,7 +231,7 @@ def transfer(tp):
return tp
if isinstance(tp[0], str):
return MaxPool2D(*tp[1:])
return Conv(out_channels = tp[0], kernel_size = tp[1], stride = tp[-1] if len(tp) == 4 else 1, bias=bias, normal=normal)
return Conv(out_channels = tp[0], kernel_size = tp[1], stride = tp[-1] if len(tp) == 4 else 1, bias=bias, normal=normal, **kargs)

return Seq(*([transfer(s) for s in conv_layers] + [FFNN(ly, **kargs, bias=bias)]))

Expand Down

0 comments on commit 409a3cc

Please sign in to comment.