Spaces:
Running
Running
Update pages/13_FFNN.py
Browse files- pages/13_FFNN.py +3 -0
pages/13_FFNN.py
CHANGED
@@ -80,6 +80,9 @@ net = FeedforwardNeuralNetwork(784, hidden1_size, hidden2_size, hidden3_size, 10
|
|
80 |
criterion = nn.CrossEntropyLoss()
|
81 |
optimizer = optim.SGD(net.parameters(), lr=learning_rate, momentum=momentum)
|
82 |
|
|
|
|
|
|
|
83 |
# Train the network
|
84 |
if st.sidebar.button('Train Network'):
|
85 |
train_network(net, trainloader, criterion, optimizer, epochs)
|
|
|
80 |
criterion = nn.CrossEntropyLoss()
|
81 |
optimizer = optim.SGD(net.parameters(), lr=learning_rate, momentum=momentum)
|
82 |
|
83 |
+
# Add vertical space
|
84 |
+
st.write('\n' * 10)
|
85 |
+
|
86 |
# Train the network
|
87 |
if st.sidebar.button('Train Network'):
|
88 |
train_network(net, trainloader, criterion, optimizer, epochs)
|