From 5d31ad51490766acd5d4f8be1cc8b1fbdddd27be Mon Sep 17 00:00:00 2001 From: Michael Foiani Date: Sat, 7 May 2022 17:58:34 -0400 Subject: mege with pycache --- hyperparameters.py | 2 +- main.py | 5 ++++- save.jpg | Bin 19949 -> 135003 bytes 3 files changed, 5 insertions(+), 2 deletions(-) diff --git a/hyperparameters.py b/hyperparameters.py index 75528742..ba471691 100644 --- a/hyperparameters.py +++ b/hyperparameters.py @@ -9,7 +9,7 @@ Number of epochs. If you experiment with more complex networks you might need to increase this. Likewise if you add regularization that slows training. """ -num_epochs = 500 +num_epochs = 5 """ A critical parameter that can dramatically affect whether training diff --git a/main.py b/main.py index 4497cce0..193d7c7a 100644 --- a/main.py +++ b/main.py @@ -70,7 +70,10 @@ def main(): final_image = tf.squeeze(my_model.x) - plt.imshow(final_image).astype('uint8') + # convert image to uint8 + final_image = tf.cast(final_image, tf.uint8) + + plt.imshow(final_image) imsave(ARGS.savefile, final_image) diff --git a/save.jpg b/save.jpg index e35f2cbf..a42a41dd 100644 Binary files a/save.jpg and b/save.jpg differ -- cgit v1.2.3-70-g09d2