changed output folder back

FluxML · Dec 1, 2020 · 483bd7a · 483bd7a
1 parent 6369e5c
commit 483bd7a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/vision/cdcgan_mnist/cGAN_mnist.jl b/vision/cdcgan_mnist/cGAN_mnist.jl
@@ -158,8 +158,8 @@ function train(; kws...)
     opt_dscr = ADAM(hparams.lr_dscr, (0.5, 0.99))
     opt_gen = ADAM(hparams.lr_gen, (0.5, 0.99))
 
-    # Check if the `results` directory exists or needed to be created
-    isdir("results")||mkdir("results")
+    # Check if the `output` directory exists or needed to be created
+    isdir("output")||mkdir("output")
 
     # Training
     train_steps = 0
@@ -173,15 +173,15 @@ function train(; kws...)
                 @info("Train step $(train_steps), Discriminator loss = $(loss["discr"]), Generator loss = $(loss["gen"])")
                 # Save generated fake image
                 output_image = create_output_image(gen, fixed_noise, fixed_labels, hparams)
-                save(@sprintf("results/cgan_steps_%06d.png", train_steps), output_image)
+                save(@sprintf("output/cgan_steps_%06d.png", train_steps), output_image)
             end
 
             train_steps += 1
         end
     end
 
     output_image = create_output_image(gen, fixed_noise, fixed_labels, hparams)
-    save(@sprintf("results/cgan_steps_%06d.png", train_steps), output_image)
+    save(@sprintf("output/cgan_steps_%06d.png", train_steps), output_image)
     return Flux.onecold.(cpu(fixed_labels))
 end