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

Disable debugging of comparable pingpong generations

Browse Source
This commit is contained in:
James Betker 2020-10-31 11:09:10 -06:00
parent 565517814e
commit c2866ad8d2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/steps/tecogan_losses.py
View File

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