clara/pytorch-ai
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

Implemented Net based AI

Browse source
This commit is contained in:
Clemens Dautermann 2019-12-26 23:02:08 +01:00
parent afb48a38a1
commit 9350b1e4ba
15 changed files with 68 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

2
mnist_classifier.py
View file

@ -14,7 +14,7 @@ test = datasets.MNIST('./datasets', train=False, download=True,
transforms.ToTensor()
]))
trainset = torch.utils.data.DataLoader(train, batch_size=10, shuffle=True)
trainset = torch.utils.data.DataLoader(train, batch_size=15, shuffle=True)
testset = torch.utils.data.DataLoader(test, batch_size=10, shuffle=False)