def forward(self, x):
    x = self.conv1(x)
    x = self.bn1(x)
    x = self.relu(x)
    x = self.conv2(x)
    x = self.bn2(x)
    x = self.relu(x)
    x = self.layer1(x)
    x_list = []
    for i in range(self.stage2_cfg['num_branches']):
        if (self.transition1[i] is not None):
            x_list.append(self.transition1[i](x))
        else:
            x_list.append(x)
    y_list = self.stage2(x_list)
    x_list = []
    for i in range(self.stage3_cfg['num_branches']):
        if (self.transition2[i] is not None):
            x_list.append(self.transition2[i](y_list[(- 1)]))
        else:
            x_list.append(y_list[i])
    y_list = self.stage3(x_list)
    x_list = []
    for i in range(self.stage4_cfg['num_branches']):
        if (self.transition3[i] is not None):
            x_list.append(self.transition3[i](y_list[(- 1)]))
        else:
            x_list.append(y_list[i])
    y_list = self.stage4(x_list)
    return y_list
def forward(self, x):
    x = self.conv1(x)
    x = self.bn1(x)
    x = self.relu(x)
    x = self.conv2(x)
    x = self.bn2(x)
    x = self.relu(x)
    x = self.layer1(x)
    x_list = []
    for i in range(self.stage2_cfg['num_branches']):
        if (self.transition1[i] is not None):
            x_list.append(self.transition1[i](x))
        else:
            x_list.append(x)
    y_list = self.stage2(x_list)
    x_list = []
    for i in range(self.stage3_cfg['num_branches']):
        if (self.transition2[i] is not None):
            x_list.append(self.transition2[i](y_list[(- 1)]))
        else:
            x_list.append(y_list[i])
    y_list = self.stage3(x_list)
    x_list = []
    for i in range(self.stage4_cfg['num_branches']):
        if (self.transition3[i] is not None):
            x_list.append(self.transition3[i](y_list[(- 1)]))
        else:
            x_list.append(y_list[i])
    y_list = self.stage4(x_list)
    return y_list