def __init__(self):
    super(SE_HIPP_3D_Net, self).__init__()
    self.conv1 = nn.Conv2d(28, 32, kernel_size=4, stride=1, padding=1)
    self.bn1 = nn.BatchNorm2d(32)
    self.relu = nn.ReLU(inplace=True)
    self.conv2 = nn.Conv2d(32, 64, kernel_size=2, stride=1, padding=0)
    self.bn2 = nn.BatchNorm2d(64)
    self.fc1 = nn.Linear(((64 * 7) * 7), 120)
    self.dropout = nn.Dropout(0.5)
    self.fc2 = nn.Linear(120, 2)
def __init__(self):
    super(SE_HIPP_3D_Net, self).__init__()
    self.conv1 = nn.Conv2d(28, 32, kernel_size=4, stride=1, padding=1)
    self.bn1 = nn.BatchNorm2d(32)
    self.relu = nn.ReLU(inplace=True)
    self.conv2 = nn.Conv2d(32, 64, kernel_size=2, stride=1, padding=0)
    self.bn2 = nn.BatchNorm2d(64)
    self.fc1 = nn.Linear(((64 * 7) * 7), 120)
    self.dropout = nn.Dropout(0.5)
    self.fc2 = nn.Linear(120, 2)