From 726e946e7978e3f5c69f15c833833cc1f7479ee8 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sat, 4 Jul 2020 14:51:27 -0600 Subject: [PATCH] Turn BN off in SRG1 This wont work well but just testing if GAN performance comes back --- codes/models/archs/SRG1_arch.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/codes/models/archs/SRG1_arch.py b/codes/models/archs/SRG1_arch.py index e30a0181..79ebe608 100644 --- a/codes/models/archs/SRG1_arch.py +++ b/codes/models/archs/SRG1_arch.py @@ -58,7 +58,7 @@ class HalvingProcessingBlock(nn.Module): def __init__(self, filters): super(HalvingProcessingBlock, self).__init__() self.bnconv1 = ConvBnLelu(filters, filters * 2, stride=2, bn=False) - self.bnconv2 = ConvBnLelu(filters * 2, filters * 2, bn=True) + self.bnconv2 = ConvBnLelu(filters * 2, filters * 2, bn=False) def forward(self, x): x = self.bnconv1(x) @@ -71,7 +71,7 @@ def create_sequential_growing_processing_block(filters_init, filter_growth, num_ convs = [] current_filters = filters_init for i in range(num_convs): - convs.append(ConvBnLelu(current_filters, current_filters + filter_growth, bn=True)) + convs.append(ConvBnLelu(current_filters, current_filters + filter_growth, bn=False)) current_filters += filter_growth return nn.Sequential(*convs), current_filters