Merge pull request #39 from keskarnitish/patch-1

Fixing wrong variable name
2017-04-13 19:26:34 -07:00 · 2017-04-13 19:26:34 -07:00 · d92c503160
parent f00c4d1831 c085ca07c5
commit d92c503160
1 changed files with 1 additions and 1 deletions
--- a/models/mlp.py
+++ b/models/mlp.py
@ -31,7 +31,7 @@ class MLP_G(nn.Module):
            output = nn.parallel.data_parallel(self.main, input, range(self.ngpu))
        else:
            output = self.main(input)
-        return output.view(out.size(0), self.nc, self.isize, self.isize)
+        return output.view(output.size(0), self.nc, self.isize, self.isize)


 class MLP_D(nn.Module):