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

fix10

Browse Source
This commit is contained in:
Jakub Henyk 2023-05-07 18:27:09 +02:00
parent 42ec57f385
commit a0dc9ef9aa
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

@ -76,7 +76,7 @@ if __name__ == '__main__':
epochs = sys.argv[1]
trainNet(trainloader, criterion, optimizer, int(epochs))
trainNet(trainloader, criterion, optimizer, int(float(epochs)))
PATH = './cifar_net.pth'
torch.save(net.state_dict(), PATH)