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

fix9

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

4
train.py
View File

@ -74,9 +74,9 @@ if __name__ == '__main__':
criterion = nn.CrossEntropyLoss()
optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9)
epochs = int(sys.argv[1])
epochs = sys.argv[1]
trainNet(trainloader, criterion, optimizer, epochs)
trainNet(trainloader, criterion, optimizer, int(epochs))
PATH = './cifar_net.pth'
torch.save(net.state_dict(), PATH)