Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -63,7 +63,7 @@ class Generator(nn.Module):
|
|
| 63 |
norm_layer(out_features),
|
| 64 |
nn.ReLU(inplace=True) ]
|
| 65 |
in_features = out_features
|
| 66 |
-
out_features = in_features
|
| 67 |
self.model3 = nn.Sequential(*model3)
|
| 68 |
|
| 69 |
# Output layer
|
|
|
|
| 63 |
norm_layer(out_features),
|
| 64 |
nn.ReLU(inplace=True) ]
|
| 65 |
in_features = out_features
|
| 66 |
+
out_features = in_features*3
|
| 67 |
self.model3 = nn.Sequential(*model3)
|
| 68 |
|
| 69 |
# Output layer
|