balthazarneveu / gyraudio

A novel method for audio source separation

1 stars 0 forks source link

Setup AudioSep #5

Closed balthazarneveu closed 6 months ago

balthazarneveu commented 6 months ago

Global framework

Setup a convenient training environment

[x] Torchaudio
[x] Dataloader
[x] Dataloader SNR filter
[x] Training loop
[x] Weights and biases
[x] Experiment tracking
[x] Test training loop: CPU :+1: GPU :+1:
[x] Store checkpoints
[x] Inference / Validation code

Improve trainings scripts

[x] Generate mixed data on the fly
[x] Report receptive field

Improve inference script

[x] Improve inference : pick number of samples
[x] Improve inference : pick SNR
[x] Report metrics per SNR #11

Pytests

[x] Dataloader pytest
[x] Training script pytest (lauch exp_0)
[x] Add Models instantiation unit-tests

Dataloader pytest

SNR: 1 - TRAIN SIZE 506 TEST SIZE 209
SNR: 2 - TRAIN SIZE 485 TEST SIZE 171
SNR: 3 - TRAIN SIZE 509 TEST SIZE 206
SNR: 4 - TRAIN SIZE 520 TEST SIZE 200
SNR: -1 - TRAIN SIZE 506 TEST SIZE 201
SNR: -2 - TRAIN SIZE 507 TEST SIZE 205
SNR: -4 - TRAIN SIZE 516 TEST SIZE 206
SNR: -3 - TRAIN SIZE 450 TEST SIZE 207