eaglelandsonce commited on
Commit
37b97bc
·
verified ·
1 Parent(s): 189832d

Update pages/13_FFNN.py

Browse files
Files changed (1) hide show
  1. pages/13_FFNN.py +1 -1
pages/13_FFNN.py CHANGED
@@ -97,7 +97,7 @@ def imshow(img):
97
 
98
  if st.sidebar.button('Show Test Results'):
99
  dataiter = iter(testloader)
100
- images, labels = dataiter.next()
101
  imshow(torchvision.utils.make_grid(images))
102
  st.write('GroundTruth: ', ' '.join(f'{labels[j]}' for j in range(8)))
103
  outputs = net(images)
 
97
 
98
  if st.sidebar.button('Show Test Results'):
99
  dataiter = iter(testloader)
100
+ images, labels = next(dataiter) # Use next function
101
  imshow(torchvision.utils.make_grid(images))
102
  st.write('GroundTruth: ', ' '.join(f'{labels[j]}' for j in range(8)))
103
  outputs = net(images)