def _make_branches(self, num_branches, block, num_blocks, num_channels):
    branches = []
    for i in range(num_branches):
        branches.append(self._make_one_branch(i, block, num_blocks, num_channels))
    return nn.ModuleList(branches)
def _make_branches(self, num_branches, block, num_blocks, num_channels):
    branches = []
    for i in range(num_branches):
        branches.append(self._make_one_branch(i, block, num_blocks, num_channels))
    return nn.ModuleList(branches)