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([])