AlekseyKorshuk commited on
Commit
3927aba
·
1 Parent(s): bba50e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,7 @@ from torchvision.utils import save_image
14
  class Generator(nn.Module):
15
  def __init__(self, nc=4, nz=100, ngf=64):
16
  super(Generator, self).__init__()
17
- self.network = nn.Sequential(
18
  nn.ConvTranspose2d(nz, ngf * 4, 3, 1, 0, bias=False),
19
  nn.BatchNorm2d(ngf * 4),
20
  nn.ReLU(True),
@@ -29,7 +29,7 @@ class Generator(nn.Module):
29
  )
30
 
31
  def forward(self, input):
32
- output = self.network(input)
33
  return output
34
 
35
 
 
14
  class Generator(nn.Module):
15
  def __init__(self, nc=4, nz=100, ngf=64):
16
  super(Generator, self).__init__()
17
+ self.model = nn.Sequential(
18
  nn.ConvTranspose2d(nz, ngf * 4, 3, 1, 0, bias=False),
19
  nn.BatchNorm2d(ngf * 4),
20
  nn.ReLU(True),
 
29
  )
30
 
31
  def forward(self, input):
32
+ output = self.model(input)
33
  return output
34
 
35