@ -21,6 +21,7 @@ test_data = datasets.MNIST(
transform=ToTensor(),
)
batch_size = 100
train_loader = DataLoader(training_data, batch_size=batch_size)