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

dbg

Browse Source
This commit is contained in:
James Betker 2020-08-23 17:43:03 -06:00
parent 4bb5b3c981
commit 0005c56cd4
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/models/SRGAN_model.py
View File

@ -178,6 +178,9 @@ class SRGANModel(BaseModel):
if self.use_corrupted_feature_input: if self.use_corrupted_feature_input:
logger.info("Corrupting inputs into the feature network..") logger.info("Corrupting inputs into the feature network..")
self.feature_corruptor = GaussianBlur() self.feature_corruptor = GaussianBlur()
else:
logger.info("Using normal inputs into feature network..")
print(train_opt)
self.netF = networks.define_F(use_bn=False).to(self.device) self.netF = networks.define_F(use_bn=False).to(self.device)
self.lr_netF = None self.lr_netF = None
if 'lr_fea_path' in train_opt.keys(): if 'lr_fea_path' in train_opt.keys():