Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
yuguochencuc authored Jan 11, 2022
1 parent fc604f6 commit 3c93d56
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,10 @@ Abstract:Curriculum learning begins to thrive in the speech enhancement area,
The code for network architecture is privoided now, while the trainer is released soon. You can use dual_aia_trans_merge_crm() in aia_trans.py for dual-branch SE, while aia_complex_trans_mag() and aia_complex_trans_ri() are single-branch aprroaches.
The trained weights on VB dataset is also provided. You can directly perform inference or finetune the model by using vb_aia_merge_new.pth.tar. The code for inference will be released soon.

### Inference:
The trained weights vb_aia_merge_new.pth.tar on VB dataset is also provided in BEST_MODEL.
you can run python enhance.py to enhance the noisy speech samples.

### requirements:

CUDA 10.1
Expand Down

0 comments on commit 3c93d56

Please sign in to comment.