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

Support training RRDB encoder [srflow]

Browse Source
This commit is contained in:
James Betker 2020-11-20 10:03:06 -07:00
parent 5ccdbcefe3
commit 6c8c35ac47
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
codes/models/archs/srflow_orig/SRFlowNet_arch.py
View File

@ -25,10 +25,7 @@ class SRFlowNet(nn.Module):
hidden_channels = opt_get(opt, ['networks', 'generator','flow', 'hidden_channels']) hidden_channels = opt_get(opt, ['networks', 'generator','flow', 'hidden_channels'])
hidden_channels = hidden_channels or 64 hidden_channels = hidden_channels or 64
self.RRDB_training = True # Default is true self.RRDB_training = opt_get(self.opt, ['networks', 'generator','train_RRDB'], default=False)
train_RRDB_delay = opt_get(self.opt, ['networks', 'generator','train_RRDB_delay'])
self.RRDB_training = False
self.flowUpsamplerNet = \ self.flowUpsamplerNet = \
FlowUpsamplerNet((160, 160, 3), hidden_channels, K, FlowUpsamplerNet((160, 160, 3), hidden_channels, K,