def _make_stage(self, layer_config, num_inchannels, multi_scale_output=True):
    num_modules = layer_config['num_modules']
    num_branches = layer_config['num_branches']
    num_blocks = layer_config['num_blocks']
    num_channels = layer_config['num_channels']
    block = blocks_dict[layer_config['block']]
    fuse_method = layer_config['fuse_method']
    modules = []
    for i in range(num_modules):
        if ((not multi_scale_output) and (i == (num_modules - 1))):
            reset_multi_scale_output = False
        else:
            reset_multi_scale_output = True
        modules.append(HighResolutionModule(num_branches, block, num_blocks, num_inchannels, num_channels, fuse_method, reset_multi_scale_output))
        num_inchannels = modules[(- 1)].get_num_inchannels()
    return (nn.Sequential(*modules), num_inchannels)
def _make_stage(self, layer_config, num_inchannels, multi_scale_output=True):
    num_modules = layer_config['num_modules']
    num_branches = layer_config['num_branches']
    num_blocks = layer_config['num_blocks']
    num_channels = layer_config['num_channels']
    block = blocks_dict[layer_config['block']]
    fuse_method = layer_config['fuse_method']
    modules = []
    for i in range(num_modules):
        if ((not multi_scale_output) and (i == (num_modules - 1))):
            reset_multi_scale_output = False
        else:
            reset_multi_scale_output = True
        modules.append(HighResolutionModule(num_branches, block, num_blocks, num_inchannels, num_channels, fuse_method, reset_multi_scale_output))
        num_inchannels = modules[(- 1)].get_num_inchannels()
    return (nn.Sequential(*modules), num_inchannels)