s452627/ium_452627
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix8

Browse Source
This commit is contained in:
Jakub Henyk 2023-05-07 18:23:15 +02:00
parent 3e2e907dd6
commit 0cc0810ad7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -74,7 +74,7 @@ if __name__ == '__main__':
criterion = nn.CrossEntropyLoss() criterion = nn.CrossEntropyLoss()
optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9) optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9)
epochs = sys.argv[1] epochs = int(sys.argv[1])
trainNet(trainloader, criterion, optimizer, epochs) trainNet(trainloader, criterion, optimizer, epochs)