def forward(self, x):
    outs = []
    s0 = s1 = self.stem(x)
    for (i, cell) in enumerate(self.cells):
        (s0, s1) = (s1, cell(s0, s1))
        if (i in self.out_indices):
            outs.append(s1)
        if ((i == self.auxliary_indice) and self.training):
            aux_feature = self.auxliary_module(s1)
            outs.insert(0, aux_feature)
    return tuple(outs)
def forward(self, x):
    outs = []
    s0 = s1 = self.stem(x)
    for (i, cell) in enumerate(self.cells):
        (s0, s1) = (s1, cell(s0, s1))
        if (i in self.out_indices):
            outs.append(s1)
        if ((i == self.auxliary_indice) and self.training):
            aux_feature = self.auxliary_module(s1)
            outs.insert(0, aux_feature)
    return tuple(outs)