Can you please hepl me with this code:
import torch import torch.nn as nn import torch.nn.functional as f import torch.optim as optim from torchvision import datasets, transforms from torch.autograd import Variable kwargs = {} train_data = torch.utils.data.DataLoader(datasets.MNIST('data', train=True, download=True, transform=transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.1307,), (0.3081,))])), batch_size=64, shuffle=True, **kwargs) test_data = torch.utils.data.DataLoader(datasets.MNIST('data', train=False, transform=transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.1307,), (0.3081,))])), batch_size=64, shuffle=True, **kwargs) class Netz(nn.Module): def __init__(self): super(Netz, self).__init__() self.conv1 = nn.Conv2d(1, 10, kernel_size=5) self.conv2 = nn.Conv2d(10, 20, kernel_size=5) self.conf_dropout = nn.Dropout2d() self.fc1 = nn.Linear(320, 60) self.fc2 = nn.Linear(60, 10) def forward(self, x): x = self.conv1(x) x = f.max_pool2d(x, 2) x = f.relu(x) x = self.conv2(x) x = self.conv.dropout(x) x = f.max_pool2d(x, 2) x = f.relu(x) print(x.size()) exit() model = Netz() optimizer = optim.SGD(model.parameters(), lr=0.1, momentum=0.8) def train(epoch): model.train() for batch_id, (data, target) in enumerate(train_data): data = Variable(data) target = Variable(target) optimizer.zero_grad() out = model(data) criterion = F.nll_loss loss = criterion(out, target) loss.backward() optimizer.step() print('Train Epoch: {} [{}/{} ({:.0f}%)]\tLoss: {:.6f}'.format(epoch, batch_id * len(data), len(train_data.dataset), 100. * batch_id / len(train_data), loss.data[0])) for epoch in range(1, 30): train(epoch)I'm getting these Errors:
Error:Traceback (most recent call last):
File "/Applications/PyCharm CE.app/Contents/helpers/pydev/pydevd.py", line 1741, in <module>
main()
File "/Applications/PyCharm CE.app/Contents/helpers/pydev/pydevd.py", line 1735, in main
globals = debugger.run(setup['file'], None, None, is_module)
File "/Applications/PyCharm CE.app/Contents/helpers/pydev/pydevd.py", line 1135, in run
pydev_imports.execfile(file, globals, locals) # execute the script
File "/Applications/PyCharm CE.app/Contents/helpers/pydev/_pydev_imps/_pydev_execfile.py", line 18, in execfile
exec(compile(contents+"\n", file, 'exec'), glob, loc)
File "/Users/***/PycharmProjects/Pytorch/venv/mnist.py", line 51, in <module>
train(epoch)
File "/Users/***/PycharmProjects/Pytorch/venv/mnist.py", line 41, in train
out = model(data)
File "/Users/***/PycharmProjects/Pytorch/venv/lib/python3.7/site-packages/torch/nn/modules/module.py", line 489, in __call__
result = self.forward(*input, **kwargs)
File "/Users/***/PycharmProjects/Pytorch/venv/mnist.py", line 26, in forward
x = self.conv.dropout(x)
File "/Users/***/PycharmProjects/Pytorch/venv/lib/python3.7/site-packages/torch/nn/modules/module.py", line 535, in __getattr__
type(self).__name__, name))
AttributeError: 'Netz' object has no attribute 'conv'