Function load_residual_unit

src/models/heartmula/codec.cpp:393–420

    const int64_t first_channels = config.init_channel * pow2_i64(config.upsample_factors.size());
    weights.input_conv = load_weight_norm_conv1d(
        store,
        source,
        "scalar_model.decoder.0",
        storage_type,
        first_channels,
        config.latent_hidden_dim,
        config.delay_kernel_size);
    weights.blocks.reserve(config.upsample_factors.size());
    int64_t in_channels = first_channels;
    for (size_t block_index = 0; block_index < config.upsample_factors.size(); ++block_index) {
        const int64_t out_channels = in_channels / 2;
        const std::string prefix = "scalar_model.decoder." + std::to_string(block_index + 1);
        HeartCodecDecoderBlockWeights block;
        block.up_conv = load_weight_norm_conv_transpose1d(
            store,
            source,
            prefix + ".up_conv.layer",
            storage_type,
            in_channels,
            out_channels,
            config.upsample_kernel_sizes[block_index]);
        block.residual_units.reserve(5);
        for (int64_t residual = 0; residual < 5; ++residual) {
            block.residual_units.push_back(load_residual_unit(
                store,
                source,
                prefix + ".convs." + std::to_string(residual),
                storage_type,
                out_channels,
                config.res_kernel_size));
        }