Spaces:
Running
Running
Commit
·
3927aba
1
Parent(s):
bba50e1
Update app.py
Browse files
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.
|
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.
|
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 |
|