From e38716925f2052670edeb63bc81de47cb3cc247a Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 2 Oct 2020 09:00:18 -0600 Subject: [PATCH] Fix spsr8 class init --- codes/models/archs/SPSR_arch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/archs/SPSR_arch.py b/codes/models/archs/SPSR_arch.py index 30f7b01f..ff29f64b 100644 --- a/codes/models/archs/SPSR_arch.py +++ b/codes/models/archs/SPSR_arch.py @@ -598,7 +598,7 @@ class Spsr7(nn.Module): # Based on Spsr7 but swaps sw2 to the end of the chain. Also re-enables pretransform convs. class Spsr8(nn.Module): def __init__(self, in_nc, out_nc, nf, xforms=8, upscale=4, multiplexer_reductions=3, init_temperature=10): - super(Spsr7, self).__init__() + super(Spsr8, self).__init__() n_upscale = int(math.log(upscale, 2)) # processing the input embedding