def forward(self, x):
    x = self.conv1(x)
    x = F.max_pool2d(x, kernel_size=3, stride=2, padding=0)
    x = self.bn1(x)
    x = self.relu(x)
    x = self.conv2(x)
    x = F.max_pool2d(x, kernel_size=2, stride=2, padding=1)
    x = self.bn2(x)
    x = self.relu(x)
    x = x.view((- 1), self.num_flat_features(x))
    x = self.dropout(x)
    x = F.relu(self.fc1(x))
    x = self.fc2(x)
    return x
def forward(self, x):
    x = self.conv1(x)
    x = F.max_pool2d(x, kernel_size=3, stride=2, padding=0)
    x = self.bn1(x)
    x = self.relu(x)
    x = self.conv2(x)
    x = F.max_pool2d(x, kernel_size=2, stride=2, padding=1)
    x = self.bn2(x)
    x = self.relu(x)
    x = x.view((- 1), self.num_flat_features(x))
    x = self.dropout(x)
    x = F.relu(self.fc1(x))
    x = self.fc2(x)
    return x