Merge branch 'main' of https://github.com/DanielHalachev/ImageColorizer

DanielHalachev · DanielHalachev · commit aed1cfaa9294 · 2024-01-28T10:10:33.000+02:00
diff --git a/utils.py b/utils.py
@@ -126,7 +126,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')
