def forward(self, x):
    x = self.conv1(x)
    x = self.stages[0](x)
    x = self.stages[1](x)
    out3 = self.stages[2](x)
    out4 = self.stages[3](out3)
    out5 = self.stages[4](out4)
    return (out3, out4, out5)
def forward(self, x):
    x = self.conv1(x)
    x = self.stages[0](x)
    x = self.stages[1](x)
    out3 = self.stages[2](x)
    out4 = self.stages[3](out3)
    out5 = self.stages[4](out4)
    return (out3, out4, out5)