def __init__(self, num_branches, blocks, num_blocks, num_inchannels, num_channels, fuse_method, multi_scale_output=True):
    super(HighResolutionModule, self).__init__()
    self._check_branches(num_branches, blocks, num_blocks, num_inchannels, num_channels)
    self.num_inchannels = num_inchannels
    self.fuse_method = fuse_method
    self.num_branches = num_branches
    self.multi_scale_output = multi_scale_output
    self.branches = self._make_branches(num_branches, blocks, num_blocks, num_channels)
    self.fuse_layers = self._make_fuse_layers()
    self.relu = nn.ReLU(False)
def __init__(self, num_branches, blocks, num_blocks, num_inchannels, num_channels, fuse_method, multi_scale_output=True):
    super(HighResolutionModule, self).__init__()
    self._check_branches(num_branches, blocks, num_blocks, num_inchannels, num_channels)
    self.num_inchannels = num_inchannels
    self.fuse_method = fuse_method
    self.num_branches = num_branches
    self.multi_scale_output = multi_scale_output
    self.branches = self._make_branches(num_branches, blocks, num_blocks, num_channels)
    self.fuse_layers = self._make_fuse_layers()
    self.relu = nn.ReLU(False)