examples/dcgan/dcgan.rb in ruby-dnn-1.2.3 vs examples/dcgan/dcgan.rb in ruby-dnn-1.3.0

@@ -119,10 +119,10 @@
     @dis.disable_training
     x = @dis.(x)
     x
   end
 
-  def train_step(x_batch, y_batch)
+  def train_step(x_batch, y_batch, need_accuracy: false)
     batch_size = x_batch.shape[0]
     noise = Numo::SFloat.new(batch_size, 20).rand(-1, 1)
     images = @gen.predict(noise)
     y_real = Numo::SFloat.ones(batch_size, 1)
     y_fake = Numo::SFloat.zeros(batch_size, 1)