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

Revert "spsr7 adjustments"

Browse Source 
This reverts commit 9fee1cec71.
This commit is contained in:
James Betker 2020-09-29 18:30:41 -06:00
parent 9fee1cec71
commit 896b4f5be2
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
codes/models/archs/SPSR_arch.py
View File

@ -528,16 +528,16 @@ class Spsr7(nn.Module):
transform_count=self.transformation_counts, init_temp=init_temperature,
add_scalable_noise_to_transforms=False, feed_transforms_into_multiplexer=True)
self.sw1_out = nn.Sequential(ConvGnLelu(nf, nf, kernel_size=3, norm=False, activation=True),
ConvGnLelu(nf, 3, kernel_size=1, norm=False, activation=False))
ConvGnLelu(nf, 3, kernel_size=1, norm=False, activation=False, bias=True))
self.sw2 = ConfigurableSwitchComputer(transformation_filters, multiplx_fn,
pre_transform_block=None, transform_block=transform_fn,
attention_norm=True,
transform_count=self.transformation_counts, init_temp=init_temperature,
add_scalable_noise_to_transforms=False, feed_transforms_into_multiplexer=True)
self.feature_lr_conv = ConvGnLelu(nf, nf, kernel_size=3, norm=True, activation=False, bias=False)
self.feature_lr_conv = ConvGnLelu(nf, nf, kernel_size=3, norm=True, activation=False)
self.feature_lr_conv2 = ConvGnLelu(nf, nf, kernel_size=3, norm=False, activation=False, bias=False)
self.sw2_out = nn.Sequential(ConvGnLelu(nf, nf, kernel_size=3, norm=False, activation=True),
ConvGnLelu(nf, 3, kernel_size=1, norm=False, activation=False))
ConvGnLelu(nf, 3, kernel_size=1, norm=False, activation=False, bias=True))
# Grad branch. Note - groupnorm on this branch is REALLY bad. Avoid it like the plague.
self.get_g_nopadding = ImageGradientNoPadding()
@ -587,7 +587,6 @@ class Spsr7(nn.Module):
x2 = x1
x2, a2 = self.sw2(x2, True, identity=x1, att_in=(x2, ref_embedding))
x2 = self.feature_lr_conv2(self.feature_lr_conv(x2))
s2out = self.sw2_out(x2)
x_grad = self.grad_conv(x_grad)