From ad734c614a4cc6ebb644326f4966f51392d8e7ee Mon Sep 17 00:00:00 2001
From: Luis Roberto Mercado Diaz <luis.mercado_diaz@uconn.edu>
Date: Mon, 27 Nov 2023 12:35:27 -0500
Subject: [PATCH] Just update in the plotting method

---
 project_NVIL.py | 15 ---------------
 1 file changed, 15 deletions(-)

diff --git a/project_NVIL.py b/project_NVIL.py
index 7996ed7..2654939 100644
--- a/project_NVIL.py
+++ b/project_NVIL.py
@@ -216,21 +216,6 @@ def imshow(image, ax=None, title=None, normalize=True):
 
 
 
-
-fig, axes = plt.subplots(figsize=(20,15), ncols=6)
-
-for i, batch in enumerate(train_loader):
-    images, label, names = batch["data"], batch["label"], batch["segment_name"]
-    ax = axes[i]
-#     helper.imshow(images[ii], ax=ax, normalize=False)
-    imshow(images[i], ax=ax, normalize=False)
-    if i==6:
-        break
-
-
-
-
-
 class PriorNet(nn.Module):
     """
     An NN that parameterises a prior distribution.