Hugging Face's logo

Spaces:
eaglelandsonce
/
pytorch
Running

App Files Community
eaglelandsonce commited on
Commit
b8d9743
·
verified ·
1 Parent(s): 491f419

Update pages/15_CNN.py

Browse files
Files changed (1) hide show
  1. pages/15_CNN.py +1 -1
pages/15_CNN.py CHANGED
@@ -40,7 +40,7 @@ test_loader = DataLoader(test_dataset, batch_size=batch_size, shuffle=False)
40
 
41
  # Display some sample images
42
  st.write("## Sample Images from CIFAR-10 Dataset")
43
- sample_images, sample_labels = iter(train_loader).next()
44
  fig, axes = plt.subplots(1, 6, figsize=(15, 5))
45
  for i in range(6):
46
  axes[i].imshow(np.transpose(sample_images[i].numpy(), (1, 2, 0)))
 
40
 
41
  # Display some sample images
42
  st.write("## Sample Images from CIFAR-10 Dataset")
43
+ sample_images, sample_labels = next(iter(train_loader))
44
  fig, axes = plt.subplots(1, 6, figsize=(15, 5))
45
  for i in range(6):
46
  axes[i].imshow(np.transpose(sample_images[i].numpy(), (1, 2, 0)))