speechbrainteam commited on
Commit
43a1335
1 Parent(s): 279d3f3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -2
README.md CHANGED
@@ -57,12 +57,35 @@ torchaudio.save("source2hat.wav", est_sources[:, :, 1].detach().cpu(), 8000)
57
  ### Inference on GPU
58
  To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
59
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
60
  ### Limitations
61
  The SpeechBrain team does not provide any warranty on the performance achieved by this model when used on other datasets.
62
 
63
  #### Referencing SpeechBrain
64
 
65
- ```
66
  ```bibtex
67
  @misc{speechbrain,
68
  title={SpeechBrain: A General-Purpose Speech Toolkit},
@@ -73,7 +96,6 @@ The SpeechBrain team does not provide any warranty on the performance achieved b
73
  primaryClass={eess.AS}
74
  }
75
  ```
76
- ```
77
 
78
  #### Referencing SepFormer
79
  ```bibtex
 
57
  ### Inference on GPU
58
  To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
59
 
60
+ ### Training
61
+ The model was trained with SpeechBrain (e375cd13).
62
+ To train it from scratch follows these steps:
63
+ 1. Clone SpeechBrain:
64
+ ```bash
65
+ git clone https://github.com/speechbrain/speechbrain/
66
+ ```
67
+ 2. Install it:
68
+ ```
69
+ cd speechbrain
70
+ pip install -r requirements.txt
71
+ pip install -e .
72
+ ```
73
+
74
+ 3. Run Training:
75
+ ```
76
+ cd recipes/WHAMandWHAMR/separation
77
+ python train.py hparams/sepformer-wham.yaml --data_folder=your_data_folder
78
+ ```
79
+
80
+ You can find our training results (models, logs, etc) [here](https://drive.google.com/drive/folders/1dIAT8hZxvdJPZNUb8Zkk3BuN7GZ9-mZb?usp=sharing).
81
+
82
+
83
  ### Limitations
84
  The SpeechBrain team does not provide any warranty on the performance achieved by this model when used on other datasets.
85
 
86
  #### Referencing SpeechBrain
87
 
88
+
89
  ```bibtex
90
  @misc{speechbrain,
91
  title={SpeechBrain: A General-Purpose Speech Toolkit},
 
96
  primaryClass={eess.AS}
97
  }
98
  ```
 
99
 
100
  #### Referencing SepFormer
101
  ```bibtex