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`.
This commit is contained in:
Nitish Shirish Keskar 2017-04-13 15:34:29 -05:00 committed by GitHub
parent f00c4d1831
commit c085ca07c5
1 changed files with 1 additions and 1 deletions

View File

@ -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):