Pytorch CNN NotImplementedError

Question

Here is my network class:

class network(nn.Module):
def __init__(self):
    super(network, self).__init__()
    self.conv1 = nn.Conv2d(3, 32,