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

Another fix..

Browse Source
This commit is contained in:
James Betker 2020-10-04 21:08:00 -06:00
parent a890e3a9c0
commit e760658fdb
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/steps/losses.py
View File

@ -268,7 +268,7 @@ class GeometricSimilarityGeneratorLoss(ConfigurableLoss):
else:
upsampled_altered = net(*altered)
if self.gen_output_to_use:
if self.gen_output_to_use is not None:
upsampled_altered = upsampled_altered[self.gen_output_to_use]
# Undo alteration on HR image