eaglelandsonce commited on
Commit
9ce62f1
·
verified ·
1 Parent(s): 37b97bc

Update pages/13_FFNN.py

Browse files
Files changed (1) hide show
  1. 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)