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