examples/vae.rb in ruby-dnn-1.1.6 vs examples/vae.rb in ruby-dnn-1.2.0

@@ -95,10 +95,10 @@
 
 model = VAE.new
 dec = model.dec
 model.setup(Adam.new, VAELoss.new)
 
-model.train(x_train, x_train, 10, batch_size: 100)
+model.train(x_train, x_train, 10, batch_size: 128)
 
 images = []
 10.times do |i|
   10.times do |j|
     z1 = (i / 4.5) - 1