提交 3f849780 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix not passing subsample to the gradients.

上级 0ad57e82
...@@ -272,8 +272,10 @@ class GpuDnnConv(GpuDnnConvBase): ...@@ -272,8 +272,10 @@ class GpuDnnConv(GpuDnnConvBase):
img, kerns = inp img, kerns = inp
top, = grads top, = grads
d_img = GpuDnnConvGradI(self.border_mode, self.conv_mode)(kerns, top) d_img = GpuDnnConvGradI(self.border_mode, self.subsample,
d_kerns = GpuDnnConvGradW(self.border_mode, self.conv_mode)(img, top) self.conv_mode)(kerns, top)
d_kerns = GpuDnnConvGradW(self.border_mode, self.subsample,
self.conv_mode)(img, top)
return d_img, d_kerns return d_img, d_kerns
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论