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):
def backward(ctx, grad_output):
grad_input=Variable(grad_output.data.new())
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:
grad_bias=None
else:
......
......@@ -127,4 +127,6 @@ class Deconvolution(Module):
return s
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