s462072/Trashmaster
s462072/Trashmaster
5
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add trained NN on 10 epoch

Browse Source
This commit is contained in:
happy531 2022-05-26 20:55:18 +02:00
parent 8029b42ed3
commit 134e45de18
2 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
NeuralNetwork/train_nn.py
View File

@ -20,7 +20,7 @@ def trainNeuralNetwork():
criterion = nn.CrossEntropyLoss() criterion = nn.CrossEntropyLoss()
optimizer = optim.SGD(neural_net.parameters(), lr=0.001, momentum=0.9) 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): for epoch in range(epoch_num):
measure_loss = 0.0 measure_loss = 0.0
for i, data in enumerate(trainloader, 0): for i, data in enumerate(trainloader, 0):

BIN
resources/trained_nn.pth → trained_nn.pth
View File

Binary file not shown.