mege with pycache

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
--- a/save.jpg
+++ b/save.jpg