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

feamod fix

Browse Source
This commit is contained in:
James Betker 2020-08-30 08:08:49 -06:00
parent a56e906f9c
commit b2091cb698
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/feature_model.py
View File

@ -74,7 +74,7 @@ class FeatureModel(BaseModel):
self.optimizer_G.zero_grad()
# grey out the LR image but keep 3 channels.
lr = torch.mean(self.var_H, dim=1, keepdim=True)
lr = torch.mean(self.real_H, dim=1, keepdim=True)
lr = lr.repeat(1, 3, 1, 1)
self.fake_H = self.fea_train(lr, interpolate_factor=1)