I would like to know how to use parts 0-3 of the mnist dataset so that the output neurons of the fully connected neural network end up at 4。
here is my code
trans = transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.1307,), (0.3081,))])
train_set = datasets.MNIST(‘…/data’, train=True, transform=trans, download=True)
test_set = datasets.MNIST(‘…/data’, train=False, transform=trans, download=True)
train_loader = torch.utils.data.DataLoader(train_set, batch_size=batch_size, shuffle=True)
test_loader = torch.utils.data.DataLoader(test_set, batch_size=batch_size, shuffle=True)
Please help me fix my code