ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

bug: test_loader was using train batch_size instead of test_batch_size

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/frameworks/pytorch/pytorch_tensorboard.py
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:

2
examples/frameworks/tensorboardx/pytorch_tensorboardX.py
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: