def __init__(self):
    super(Stage5, self).__init__()
    self.layer1 = torch.nn.Conv2d(64, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
    self._initialize_weights()
def __init__(self):
    super(Stage5, self).__init__()
    self.layer1 = torch.nn.Conv2d(64, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
    self._initialize_weights()