bug: test_loader was using train batch_size instead of test_batch_size

This commit is contained in:
stef-ubuntu 2020-06-19 16:55:15 +09:00
parent 9a7850b23d
commit 6bf448f43b
2 changed files with 2 additions and 2 deletions

View File

@ -118,7 +118,7 @@ def main():
transform=transforms.Compose([
transforms.ToTensor(),
transforms.Normalize((0.1307,), (0.3081,))])),
batch_size=args.batch_size, shuffle=True, **kwargs)
batch_size=args.test_batch_size, shuffle=True, **kwargs)
model = Net()
if args.cuda:

View File

@ -118,7 +118,7 @@ def main():
transform=transforms.Compose([
transforms.ToTensor(),
transforms.Normalize((0.1307,), (0.3081,))])),
batch_size=args.batch_size, shuffle=True, **kwargs)
batch_size=args.test_batch_size, shuffle=True, **kwargs)
model = Net()
if args.cuda: