diff options
| -rw-r--r-- | hyperparameters.py | 2 | ||||
| -rw-r--r-- | main.py | 5 | ||||
| -rw-r--r-- | 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 @@ -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) |
