diff --git a/13_feedforward.py b/13_feedforward.py index eb5d580..6515fcc 100644 --- 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):