Merge remote-tracking branch 'origin/gan_lab' into gan_lab

This commit is contained in:
James Betker 2020-10-31 11:09:32 -06:00
commit eb7df63592

View File

@ -347,8 +347,8 @@ class PingPongLoss(ConfigurableLoss):
early = fake[:, i]
late = fake[:, -(i+1)]
l_total += self.criterion(early, late)
if self.env['step'] % 50 == 0:
self.produce_teco_visual_debugs2(early, late, i)
#if self.env['step'] % 50 == 0:
# self.produce_teco_visual_debugs2(early, late, i)
if self.env['step'] % 50 == 0:
self.produce_teco_visual_debugs(fake)