From 706a76cfea84f90042276681eb92b9e3880980d4 Mon Sep 17 00:00:00 2001 From: neverix Date: Wed, 3 Nov 2021 20:08:27 +0300 Subject: [PATCH] improve show() (#15) In Colab, calling show() multiple times in one cell won't display the pictures unless these two functions are called. --- rudalle/pipelines.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/rudalle/pipelines.py b/rudalle/pipelines.py index 884293b..4bde77e 100644 --- a/rudalle/pipelines.py +++ b/rudalle/pipelines.py @@ -94,3 +94,5 @@ def show(pil_images, nrow=4): img = torchvision.transforms.functional.to_pil_image(img) axs[0, i].imshow(np.asarray(img)) axs[0, i].set(xticklabels=[], yticklabels=[], xticks=[], yticks=[]) + fix.show() + plt.show()