remove broken channel adjustment
This commit is contained in:
parent
477747cced
commit
11e643bcb5
|
@ -88,10 +88,6 @@ class RRDBNet(nn.Module):
|
||||||
):
|
):
|
||||||
super(RRDBNet, self).__init__()
|
super(RRDBNet, self).__init__()
|
||||||
self.scale = scale
|
self.scale = scale
|
||||||
if scale == 2:
|
|
||||||
num_in_ch = num_in_ch * 4
|
|
||||||
elif scale == 1:
|
|
||||||
num_in_ch = num_in_ch * 16
|
|
||||||
|
|
||||||
RRDB_block_f = functools.partial(RRDB, nf=num_feat, gc=num_grow_ch)
|
RRDB_block_f = functools.partial(RRDB, nf=num_feat, gc=num_grow_ch)
|
||||||
print([num_in_ch, num_out_ch, num_feat, num_block, num_grow_ch, scale])
|
print([num_in_ch, num_out_ch, num_feat, num_block, num_grow_ch, scale])
|
||||||
|
|
Loading…
Reference in New Issue