mirror of
https://github.com/aladdinpersson/Machine-Learning-Collection.git
synced 2026-04-10 12:33:44 +00:00
small revisions to code examples
This commit is contained in:
@@ -94,6 +94,7 @@ for epoch in range(num_epochs):
|
||||
# gradient descent or adam step
|
||||
optimizer.step()
|
||||
|
||||
|
||||
# Check accuracy on training & test to see how good our model
|
||||
def check_accuracy(loader, model):
|
||||
num_correct = 0
|
||||
|
||||
Reference in New Issue
Block a user