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.