edit 13_feedforward.py

2023-01-01 19:45:09 +01:00
parent c5b5f2ef40
commit 7789767a7b
1 changed files with 1 additions and 1 deletions
--- a/13_feedforward.py
+++ b/13_feedforward.py
@@ -45,7 +45,7 @@ print(samples.shape, labels.shape)
 for i in range(6):
    plt.subplot(2, 3, i+1)
    plt.imshow(samples[i][0], cmap='gray')
-# plt.show()
+plt.show()

 class NeuralNet(nn.Module):
    def __init__(self, input_size, hidden_size, num_classes):