[egs] Add readme and pretrained model to FaSNet recipe (asteroid-team…

…#561)
darius522 · Oct 24, 2021 · 9d59ecb · 9d59ecb
1 parent 64e10e9
commit 9d59ecb
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 3 deletions.
diff --git a/egs/TAC/README.md b/egs/TAC/README.md
@@ -0,0 +1,11 @@
+This model was trained using the [dataset](https://github.com/yluo42/TAC/tree/master/data) provided by Yi Luo, author of
+["End-to-end Microphone Permutation and Number Invariant Multi-channel Speech Separation"](https://arxiv.org/abs/1910.14104).
+
+### Results
+
+|      |   task    | dataset type | SI-SNRi(dB) |
+|:----:|:---------:|:----------:|:----------:|
+| Paper| sep_clean |   adhoc    |   ?        |
+| Here | sep_clean |     adhoc    |   9.9    |
+
+The pretrained model is available [here](https://huggingface.co/JorisCos/FasNet)
diff --git a/egs/TAC/eval.py b/egs/TAC/eval.py
@@ -24,7 +24,7 @@
 )
 parser.add_argument("--exp_dir", default="exp/tmp", help="Experiment root")
 parser.add_argument(
-    "--n_save_ex", type=int, default=50, help="Number of audio examples to save, -1 means all"
+    "--n_save_ex", type=int, default=10, help="Number of audio examples to save, -1 means all"
 )
 
 compute_metrics = ["si_sdr"]  # , "sdr", "sir", "sar", "stoi"]

diff --git a/egs/TAC/local/conf.yml b/egs/TAC/local/conf.yml
@@ -19,10 +19,10 @@ optim:
   weight_decay: !!float 1e-5
 training:
   epochs: 200
-  batch_size: 4
+  batch_size: 1
   gradient_clipping: 5
   accumulate_batches: 1
-  save_top_k: 10
+  save_top_k: 5
   num_workers: 8
   patience: 30
   half_lr: true