seppl/pytorch_learn
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

edit 13_feedforward.py

Browse Source
This commit is contained in:
Joseph Hopfmüller
2023-01-01 19:45:09 +01:00
parent c5b5f2ef40
commit 7789767a7b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
13_feedforward.py
View File

@@ -45,7 +45,7 @@ print(samples.shape, labels.shape)
for i in range(6): for i in range(6):
plt.subplot(2, 3, i+1) plt.subplot(2, 3, i+1)
plt.imshow(samples[i][0], cmap='gray') plt.imshow(samples[i][0], cmap='gray')
# plt.show() plt.show()
class NeuralNet(nn.Module): class NeuralNet(nn.Module):
def __init__(self, input_size, hidden_size, num_classes): def __init__(self, input_size, hidden_size, num_classes):