def forward(self, x):
    x = self.net1.base(x)
    for stage_id in range(self.num_stages):
        x = self.net1.stages[stage_id](x)
    out1 = self.net1.head(x)
    out2 = self.net2.head(x)
    return AttrDict({'out1': out1, 'out2': out2})
def forward(self, x):
    x = self.net1.base(x)
    for stage_id in range(self.num_stages):
        x = self.net1.stages[stage_id](x)
    out1 = self.net1.head(x)
    out2 = self.net2.head(x)
    return AttrDict({'out1': out1, 'out2': out2})