fix8

2023-05-07 18:23:15 +02:00 · 2023-05-07 18:23:15 +02:00 · 0cc0810ad7
commit 0cc0810ad7
parent 3e2e907dd6
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -74,7 +74,7 @@ if __name__ == '__main__':
    criterion = nn.CrossEntropyLoss()
    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)