I’m trying to apply single convolution layer on MNIST dataset, the way it’s been shown in CNN python video. But the plot showing bit different for me, i don’t think it’s plotting correctly. It should be some blurred image not lines like shown below. Please let me know how can I get correct image here.
FirstCNN class is same as described in the video.
super(FirstCNN, self).\_\_init\_\_() self.conv1 = nn.Conv2d(1,16,3)
x = self.conv1(x) return x