examples/pix2pix/train.rb in ruby-dnn-0.16.0 vs examples/pix2pix/train.rb in ruby-dnn-0.16.1
- old
+ new
@@ -36,9 +36,10 @@
num_batchs.times do |index|
x_in, x_out = iter1.next_batch(batch_size)
images = gen.predict(x_in)
y_real = Numo::SFloat.ones(batch_size, 1)
y_fake = Numo::SFloat.zeros(batch_size, 1)
+ dis.enable_training
dis_loss = dis.train_on_batch([x_in, x_out], y_real)
dis_loss += dis.train_on_batch([x_in, images], y_fake)
x_in, x_out = iter2.next_batch(batch_size)
dcgan_loss = dcgan.train_on_batch(x_in, y_real)