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

hotfix/netural_network #29

Merged
s464843 merged 3 commits from hotfix/netural_network into master 2022-05-26 21:48:29 +02:00
Conversation 0 Commits 3 Files Changed 4 +1 -1
2 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 134e45de18 - Show all commits

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.