@@ -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))
186186end
187187
0 commit comments