fix with k in gan

ragavvenkatesan · Mar 11, 2017 · 151fa79 · 151fa79
1 parent 7245bb5
commit 151fa79
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yann/special/gan.py b/yann/special/gan.py
@@ -810,7 +810,7 @@ def train ( self, verbose, **kwargs):
                     if self.softmax_head is True:
                         softmax_cost = self.mini_batch_train_softmax (minibatch, epoch_counter)
                     disc_cost = self.mini_batch_train_discriminator (minibatch, epoch_counter)
-                    if minibatch % k == 0:
+                    if (minibatch + 1) * (batch + 1 ) * (epoch_counter + 1) % k == 0:
                         gen_cost = self.mini_batch_train_generator (minibatch, epoch_counter)
 
                     if numpy.isnan(gen_cost) or \