Update Utilities/visualize.py

Utilities/visualize.py

@@ -19,10 +19,10 @@ def plot_incorrect_preds(incorrect, classes, num_imgs):
         # unnormalize = T.Normalize((-mean / std).tolist(), (1.0 / std).tolist())
         unnormalized = transforms.Normalize(
             (-1.98947368, -1.98436214, -1.71072797), (4.048583, 4.11522634, 3.83141762)
-        )(cur_incorrect[i][0])
+        )(cur_incorrect[0])
         plt.imshow(transforms.ToPILImage()(unnormalized))
         plt.title(
-            f"{classes[cur_incorrect[i][1].item()]}|{classes[cur_incorrect[i][2].item()]}",
+            f"{classes[cur_incorrect[1].item()]}|{classes[cur_incorrect[2].item()]}",
             # fontsize=8,
         )
         plt.xticks([])