Commit 02624b4a authored by Julian Treiber's avatar Julian Treiber

added "groups" parameter for conv and transconv

parent cfd3bbd8
......@@ -7,6 +7,7 @@
self.${element.name} = gluon.nn.Conv2D(channels=${element.channels?c},
kernel_size=(${tc.join(element.kernel, ",")}),
strides=(${tc.join(element.stride, ",")}),
groups=${element.groups?c},
use_bias=${element.noBias?string("False", "True")})
<#include "OutputShape.ftl">
<#elseif mode == "FORWARD_FUNCTION">
......
......@@ -8,6 +8,7 @@
kernel_size=(${tc.join(element.kernel, ",")}),
strides=(${tc.join(element.stride, ",")}),
padding=self.${element.name}padding,
groups=${element.groups?c},
use_bias=${element.noBias?string("False", "True")})
<#include "OutputShape.ftl">
<#elseif mode == "FORWARD_FUNCTION">
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment