Skip to content

Commit 483bd7a

Browse files
committed
changed output folder back
1 parent 6369e5c commit 483bd7a

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

vision/cdcgan_mnist/cGAN_mnist.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -158,8 +158,8 @@ function train(; kws...)
158158
opt_dscr = ADAM(hparams.lr_dscr, (0.5, 0.99))
159159
opt_gen = ADAM(hparams.lr_gen, (0.5, 0.99))
160160

161-
# Check if the `results` directory exists or needed to be created
162-
isdir("results")||mkdir("results")
161+
# Check if the `output` directory exists or needed to be created
162+
isdir("output")||mkdir("output")
163163

164164
# Training
165165
train_steps = 0
@@ -173,15 +173,15 @@ function train(; kws...)
173173
@info("Train step $(train_steps), Discriminator loss = $(loss["discr"]), Generator loss = $(loss["gen"])")
174174
# Save generated fake image
175175
output_image = create_output_image(gen, fixed_noise, fixed_labels, hparams)
176-
save(@sprintf("results/cgan_steps_%06d.png", train_steps), output_image)
176+
save(@sprintf("output/cgan_steps_%06d.png", train_steps), output_image)
177177
end
178178

179179
train_steps += 1
180180
end
181181
end
182182

183183
output_image = create_output_image(gen, fixed_noise, fixed_labels, hparams)
184-
save(@sprintf("results/cgan_steps_%06d.png", train_steps), output_image)
184+
save(@sprintf("output/cgan_steps_%06d.png", train_steps), output_image)
185185
return Flux.onecold.(cpu(fixed_labels))
186186
end
187187

0 commit comments

Comments
 (0)