forked from mrq/DL-Art-School
Another fix..
This commit is contained in:
parent
a890e3a9c0
commit
e760658fdb
|
@ -268,7 +268,7 @@ class GeometricSimilarityGeneratorLoss(ConfigurableLoss):
|
||||||
else:
|
else:
|
||||||
upsampled_altered = net(*altered)
|
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]
|
upsampled_altered = upsampled_altered[self.gen_output_to_use]
|
||||||
|
|
||||||
# Undo alteration on HR image
|
# Undo alteration on HR image
|
||||||
|
|
Loading…
Reference in New Issue
Block a user