Asteroid model

Imported from this Zenodo model page.

Description:

This model was trained by Brij Mohan using the Librimix/ConvTasNet recipe in Asteroid. It was trained on the enh_single task of the Libri3Mix dataset.

Training config:

data:
    n_src: 1
    sample_rate: 8000
    segment: 3
    task: enh_single
    train_dir: data/wav8k/min/train-360
    valid_dir: data/wav8k/min/dev
filterbank:
    kernel_size: 16
    n_filters: 512
    stride: 8
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
training:
    batch_size: 24
    early_stop: True
    epochs: 200
    half_lr: True

Results:

si_sdr: 14.783675142685572
si_sdr_imp: 11.464625198953202
sdr: 15.497505907983102
sdr_imp: 12.07230150154914
sar: 15.497505907983102
sar_imp: 12.07230150154914
stoi: 0.9270030254700518
stoi_imp: 0.1320547197597893 

License notice:

This work "ConvTasNet_Libri1Mix_enhsingle_8k" is a derivative of LibriSpeech ASR corpus by Vassil Panayotov, used under CC BY 4.0. "ConvTasNet_Libri1Mix_enhsingle_8k" is licensed under Attribution-ShareAlike 3.0 Unported by Manuel Pariente.

Downloads last month
22
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no pipeline_tag.