Showing images now doesn't block.

DanielHalachev · DanielHalachev · commit d7b502e6d2d3 · 2023-12-23T11:12:57.000+02:00
diff --git a/utils.py b/utils.py
@@ -83,7 +83,7 @@ def visualize(model, data, save=True):
         ax = plt.subplot(3, 5, i + 1 + 10)
         ax.imshow(real_imgs[i])
         ax.axis("off")
-    plt.show()
+    plt.show(block=False)
     if save:
         base_directory = os.getcwd()
         save_folder = os.path.join(base_directory, 'data', 'stages')
