paul-corbalan/wasserstein-gan
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #39 from keskarnitish/patch-1

Browse Source 
Fixing wrong variable name
This commit is contained in:
Soumith Chintala 2017-04-13 19:26:34 -07:00 committed by GitHub
commit d92c503160
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/mlp.py
View File

@ -31,7 +31,7 @@ class MLP_G(nn.Module):
output = nn.parallel.data_parallel(self.main, input, range(self.ngpu)) output = nn.parallel.data_parallel(self.main, input, range(self.ngpu))
else: else:
output = self.main(input) 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): class MLP_D(nn.Module):