add trained NN on 10 epoch

2022-05-26 20:55:18 +02:00 · 2022-05-26 20:55:18 +02:00 · 134e45de18
commit 134e45de18
parent 8029b42ed3
2 changed files with 1 additions and 1 deletions
--- a/NeuralNetwork/train_nn.py
+++ b/NeuralNetwork/train_nn.py
@ -20,7 +20,7 @@ def trainNeuralNetwork():
    criterion = nn.CrossEntropyLoss()
    optimizer = optim.SGD(neural_net.parameters(), lr=0.001, momentum=0.9)

-    epoch_num = 4 # najlepiej 10, dla lepszej wiarygodności
+    epoch_num = 10 # najlepiej 10, dla lepszej wiarygodności
    for epoch in range(epoch_num):
        measure_loss = 0.0
        for i, data in enumerate(trainloader, 0):
--- a/resources/trained_nn.pth
+++ b/resources/trained_nn.pth