From a39ad02f487b59b2906b00b4111b2a10df4eae69 Mon Sep 17 00:00:00 2001 From: Drew Arad Hudson Date: Sat, 7 Mar 2020 07:36:58 -0800 Subject: [PATCH] Update training_loop.py For the gradient accumulation case, fix Discriminator regularization to use a different batch of real images every round. --- training/training_loop.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/training/training_loop.py b/training/training_loop.py index c2d88cf0..2389ace1 100755 --- a/training/training_loop.py +++ b/training/training_loop.py @@ -304,8 +304,7 @@ def training_loop( for _round in rounds: tflib.run(data_fetch_op, feed_dict) tflib.run(D_train_op, feed_dict) - if run_D_reg: - for _round in rounds: + if run_D_reg: tflib.run(D_reg_op, feed_dict) # Perform maintenance tasks once per tick.