Commit b87ba095 authored by Benjamin Thomas Graham's avatar Benjamin Thomas Graham
Browse files

Fixes for Issue #31

parent 11274575
...@@ -54,10 +54,6 @@ class ConvolutionFunction(Function): ...@@ -54,10 +54,6 @@ class ConvolutionFunction(Function):
def backward(ctx, grad_output): def backward(ctx, grad_output):
grad_input=Variable(grad_output.data.new()) grad_input=Variable(grad_output.data.new())
grad_weight=Variable(grad_output.data.new().resize_as_(ctx.weight).zero_()) grad_weight=Variable(grad_output.data.new().resize_as_(ctx.weight).zero_())
if ctx.bias is None:
grad_bias=None
else:
grad_bias = Variable(grad_output.data.new().resize_as_(bias).zero_())
if ctx.bias is None: if ctx.bias is None:
grad_bias=None grad_bias=None
else: else:
......
...@@ -127,4 +127,6 @@ class Deconvolution(Module): ...@@ -127,4 +127,6 @@ class Deconvolution(Module):
return s return s
def input_spatial_size(self, out_size): def input_spatial_size(self, out_size):
return (out_size - 1) * self.filter_stride + self.filter_size in_size = (out_size - self.filter_size) / self.filter_stride + 1
assert ((in_size - 1) * self.filter_stride + self.filter_size == out_size).all()
return in_size
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