@@ -664,7 +664,7 @@ | |||
} | |||
], | |||
"source": [ | |||
"train_model(kyle, trainloader_train, 100, 10)" | |||
"train_model(kyle, trainloader, 100, 10)" | |||
] | |||
}, | |||
{ | |||
@@ -749,7 +749,7 @@ | |||
} | |||
], | |||
"source": [ | |||
"test_model(kyle, trainloader_test)" | |||
"test_model(kyle, testloader)" | |||
] | |||
}, | |||
{ |