From 6945d7480a94135f89c646aeae18ff9ec4bc142f Mon Sep 17 00:00:00 2001 From: Rohan Banerjee Date: Thu, 23 Jan 2020 12:13:01 -0500 Subject: [PATCH] Update dataset.py --- dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dataset.py b/dataset.py index 61c4ac3..8203d6b 100644 --- a/dataset.py +++ b/dataset.py @@ -7,7 +7,7 @@ def view_samples(epoch, samples, nrows, ncols, figsize=(5,5)): for ax, img in zip(axes.flatten(), samples[epoch]): ax.axis('off') img = ((img - img.min())*255 / (img.max() - img.min())).astype(np.uint8) - ax.set_adjustable('box-forced') + ax.set_adjustable('box') im = ax.imshow(img, aspect='equal') plt.subplots_adjust(wspace=0, hspace=0)