mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Turn BN off in SRG1

Browse Source 
This wont work well but just testing if GAN performance comes back
This commit is contained in:
James Betker 2020-07-04 14:51:27 -06:00
parent 0ee39d419b
commit 726e946e79
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/archs/SRG1_arch.py
View File

@ -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