ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 21:33:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #148 from NumesSanguis/master

Browse Source 
Fix PyTorch example, test_loader was using train batch_size instead of test_batch_size
This commit is contained in:
Allegro AI 2020-06-19 17:00:58 +03:00 committed by GitHub
commit c50574ca4d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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: