class_docstr: Basic Pytorch CNN for MNIST-like data.
class_docstr_tok: ["Basic","Pytorch","CNN","for","MNIST-like","data","."]
code: def __init__(self):
      super(SimpleNet, self).__init__()
      self.conv1 = nn.Conv2d(1, 10, kernel_size=5)
      self.conv2 = nn.Conv2d(10, 20, kernel_size=5)
      self.conv2_drop = nn.Dropout2d()
      self.fc1 = nn.Linear(320, 50)
      self.fc2 = nn.Linear(50, 10)
  
imports: from sklearn.base import BaseEstimator
  import torch
  import torch.nn
  import torch.nn.functional
  import torch.optim
  from torchvision import datasets, transforms
  from torch.autograd import Variable
  from torch.utils.data.sampler import SubsetRandomSampler
  import numpy
label_desc: Use image node_id 1 to create a new SimpleNet object from inherited base classes: nn with example: obj = SimpleNet()