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

implemented data loading and normalization

Browse source
This commit is contained in:
Clemens Dautermann 2019-12-11 22:27:51 +01:00
parent 01d07da8f3
commit afb48a38a1
26 changed files with 986456 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

1
mnist_classifier.py
View file

@ -17,6 +17,7 @@ test = datasets.MNIST('./datasets', train=False, download=True,
trainset = torch.utils.data.DataLoader(train, batch_size=10, shuffle=True)
testset = torch.utils.data.DataLoader(test, batch_size=10, shuffle=False)
class Net(nn.Module):
def __init__(self):
super().__init__()