eschmidbauer/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix spsr8 class init

Browse Source
This commit is contained in:
James Betker 2020-10-02 09:00:18 -06:00
parent efbf6b737b
commit e38716925f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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. # Based on Spsr7 but swaps sw2 to the end of the chain. Also re-enables pretransform convs.
class Spsr8(nn.Module): class Spsr8(nn.Module):
def __init__(self, in_nc, out_nc, nf, xforms=8, upscale=4, multiplexer_reductions=3, init_temperature=10): 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)) n_upscale = int(math.log(upscale, 2))
# processing the input embedding # processing the input embedding