Method forward

networks/module.py:347–349

(self, x)

        self.cosR_small=CostRegNet_part(in_channels, base_channels,stage=0)
        self.cosR_huge=CostRegNet_part(in_channels, base_channels,stage=0)
    def forward(self, x):
        results=torch.cat((self.cosR_small(x),self.cosR_huge(x)),axis=1)
        return results
class CostRegNet_refine(nn.Module):
    def __init__(self, in_channels, base_channels,stage=0):
        super(CostRegNet_refine, self).__init__()