Caroline Mai Chan
commited on
Commit
·
641f158
1
Parent(s):
379e978
norm layer fix
Browse files
app.py
CHANGED
|
@@ -7,6 +7,7 @@ class Generator(nn.Module):
|
|
| 7 |
def __init__(self, input_nc, output_nc, n_residual_blocks=9, sigmoid=True):
|
| 8 |
super(Generator, self).__init__()
|
| 9 |
|
|
|
|
| 10 |
# Initial convolution block
|
| 11 |
model0 = [ nn.ReflectionPad2d(3),
|
| 12 |
nn.Conv2d(input_nc, 64, 7),
|
|
|
|
| 7 |
def __init__(self, input_nc, output_nc, n_residual_blocks=9, sigmoid=True):
|
| 8 |
super(Generator, self).__init__()
|
| 9 |
|
| 10 |
+
norm_layer = nn.InstanceNorm2d
|
| 11 |
# Initial convolution block
|
| 12 |
model0 = [ nn.ReflectionPad2d(3),
|
| 13 |
nn.Conv2d(input_nc, 64, 7),
|