From c085ca07c5c0b30b879111d4abf1d5e0b6941891 Mon Sep 17 00:00:00 2001 From: Nitish Shirish Keskar Date: Thu, 13 Apr 2017 15:34:29 -0500 Subject: [PATCH] Fixing wrong variable name There is no variable called `out`; the code produces an error if you try to run it. Given the history of the file, it seems that the variable should be `output`. --- models/mlp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/mlp.py b/models/mlp.py index 4701026..27ed5d4 100644 --- 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):