@@ -158,8 +158,8 @@ function train(; kws...)
158
158
opt_dscr = ADAM (hparams. lr_dscr, (0.5 , 0.99 ))
159
159
opt_gen = ADAM (hparams. lr_gen, (0.5 , 0.99 ))
160
160
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 " )
163
163
164
164
# Training
165
165
train_steps = 0
@@ -173,15 +173,15 @@ function train(; kws...)
173
173
@info (" Train step $(train_steps) , Discriminator loss = $(loss[" discr" ]) , Generator loss = $(loss[" gen" ]) " )
174
174
# Save generated fake image
175
175
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)
177
177
end
178
178
179
179
train_steps += 1
180
180
end
181
181
end
182
182
183
183
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)
185
185
return Flux. onecold .(cpu (fixed_labels))
186
186
end
187
187
0 commit comments