More fixes to corrupt_fea

This commit is contained in:
James Betker 2020-08-23 17:52:18 -06:00
parent 0005c56cd4
commit f9276007a8
2 changed files with 2 additions and 7 deletions

View File

@ -177,10 +177,7 @@ class SRGANModel(BaseModel):
self.use_corrupted_feature_input = train_opt['corrupted_feature_input'] if 'corrupted_feature_input' in train_opt.keys() else False
if self.use_corrupted_feature_input:
logger.info("Corrupting inputs into the feature network..")
self.feature_corruptor = GaussianBlur()
else:
logger.info("Using normal inputs into feature network..")
print(train_opt)
self.feature_corruptor = GaussianBlur().to(self.device)
self.netF = networks.define_F(use_bn=False).to(self.device)
self.lr_netF = None
if 'lr_fea_path' in train_opt.keys():
@ -502,8 +499,6 @@ class SRGANModel(BaseModel):
elif self.use_corrupted_feature_input:
cor_Pix = F.interpolate(self.feature_corruptor(pix), size=var_L.shape[2:])
real_fea = self.netF(cor_Pix).detach()
if step % 50 == 0:
utils.save_image(cor_Pix.detach().cpu(), "corrupted_pix.png")
else:
real_fea = self.netF(pix).detach()
if self.use_corrupted_feature_input:

View File

@ -32,7 +32,7 @@ def init_dist(backend='nccl', **kwargs):
def main():
#### options
parser = argparse.ArgumentParser()
parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_mi1_spsr_switched2.yml')
parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../experiments/finetune_imgset_spsr_switched2_xlbatch_limfeat.yml')
parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none',
help='job launcher')
parser.add_argument('--local_rank', type=int, default=0)