Edit model card

Asteroid model mhu-coder/ConvTasNet_Libri1Mix_enhsingle

Imported from Zenodo

Description:

This model was trained by Mathieu Hu using the librimix/ConvTasNet recipe in Asteroid. It was trained on the enh_single task of the Libri1Mix dataset.

Training config:

data:
    n_src: 1
    sample_rate: 16000
    segment: 3
    task: enh_single
    train_dir: data/wav16k/min/train-100
    valid_dir: data/wav16k/min/dev
filterbank:
    kernel_size: 16
    n_filters: 512
    stride: 8
main_args:
    exp_dir: exp/train_convtasnet_f34664b9
    help: None
masknet:
    bn_chan: 128
    hid_chan: 512
    mask_act: relu
    n_blocks: 8
    n_repeats: 3
    n_src: 1
    skip_chan: 128
optim:
    lr: 0.001
    optimizer: adam
    weight_decay: 0.0
positional arguments:
training:
    batch_size: 2
    early_stop: True
    epochs: 200
    half_lr: True
    num_workers: 4

Results:

si_sdr: 13.938355526049932
si_sdr_imp: 10.488574220190232
sdr: 14.567380104207393
sdr_imp: 11.064717304994337
sir: inf
sir_imp: nan
sar: 14.567380104207393
sar_imp: 11.064717304994337
stoi: 0.9201010933251715
stoi_imp: 0.1241812697846321 

License notice:

This work "ConvTasNet_Libri1Mx_enhsingle" is a derivative of CSR-I (WSJ0) Complete by LDC, used under LDC User Agreement for Non-Members (Research only). "ConvTasNet_Libri1Mix_enhsingle" is licensed under Attribution-ShareAlike 3.0 Unported by Mathieu Hu.

Downloads last month
8
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.