dataloader_imageview.py

figure = plt.figure(figsize=(12,12))
cols, rows = 4,8

for batchSize, (X, y) in enumerate(train_dataloader):
#     X = X.transpose()
    batch = y.shape[0]
    for i in range(batch):
        img, label = X[i], y[i]
#         print(img.shape)
        img.to(device)
#         img = img.transpose(3,2)
        figure.add_subplot(rows, cols, i+1)
    #     print(labels.iloc[label].char)
        plt.title(f" {label}")
        plt.axis("off")
        plt.imshow(img.squeeze(), cmap="gray")
#     for i 
    break
plt.show()

pytorch dataloader