From 174682626f21b4222a1cb294348e59ad3b260eb7 Mon Sep 17 00:00:00 2001 From: junyanz Date: Sat, 26 Aug 2017 19:04:12 -0700 Subject: fix the unaligned dataset --- models/networks.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'models/networks.py') diff --git a/models/networks.py b/models/networks.py index db36ac4..585b940 100644 --- a/models/networks.py +++ b/models/networks.py @@ -162,7 +162,7 @@ class ResnetGenerator(nn.Module): mult = 2**n_downsampling for i in range(n_blocks): - model += [ResnetBlock(ngf * mult, padding_type=padding_type, norm_layer=norm_layer, use_dropout=use_dropout)] + model += [ResnetBlock(ngf * mult, padding_type=padding_type, norm_layer=norm_layer, use_dropout=use_dropout, use_bias=use_bias)] for i in range(n_downsampling): mult = 2**(n_downsampling - i) @@ -189,9 +189,9 @@ class ResnetGenerator(nn.Module): class ResnetBlock(nn.Module): def __init__(self, dim, padding_type, norm_layer, use_dropout, use_bias): super(ResnetBlock, self).__init__() - self.conv_block = self.build_conv_block(dim, padding_type, norm_layer, use_dropout) + self.conv_block = self.build_conv_block(dim, padding_type, norm_layer, use_dropout, use_bias) - def build_conv_block(self, dim, padding_type, norm_layer, use_dropout): + def build_conv_block(self, dim, padding_type, norm_layer, use_dropout, use_bias): conv_block = [] p = 0 if padding_type == 'reflect': -- cgit v1.2.3-70-g09d2