speechbrainteam commited on
Commit
c1f4fa3
1 Parent(s): 5dd18c0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -1
README.md CHANGED
@@ -60,6 +60,29 @@ torchaudio.save("source2hat.wav", est_sources[:, :, 1].detach().cpu(), 8000)
60
  ### Inference on GPU
61
  To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
62
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
63
  ### Limitations
64
  The SpeechBrain team does not provide any warranty on the performance achieved by this model when used on other datasets.
65
 
@@ -72,7 +95,7 @@ The SpeechBrain team does not provide any warranty on the performance achieved b
72
  year = {2021},
73
  publisher = {GitHub},
74
  journal = {GitHub repository},
75
- howpublished = {\url{https://github.com/speechbrain/speechbrain}},
76
  }
77
  ```
78
 
 
60
  ### Inference on GPU
61
  To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
62
 
63
+ ### Training
64
+ The model was trained with SpeechBrain (d0accc8).
65
+ To train it from scratch follows these steps:
66
+ 1. Clone SpeechBrain:
67
+ ```bash
68
+ git clone https://github.com/speechbrain/speechbrain/
69
+ ```
70
+ 2. Install it:
71
+ ```
72
+ cd speechbrain
73
+ pip install -r requirements.txt
74
+ pip install -e .
75
+ ```
76
+
77
+ 3. Run Training:
78
+ ```
79
+ cd recipes/WSJ0Mix/separation
80
+ python train.py hparams/sepformer.yaml --data_folder=your_data_folder
81
+
82
+ https://drive.google.com/drive/folders/1fcVP52gHgoMX9diNN1JxX_My5KaRNZWs?usp=sharing
83
+
84
+ You can find our training results (models, logs, etc) [here](https://drive.google.com/drive/folders/1suvbKScf3VbkxRjZlpi1Q4hKU9yTdBVM?usp=sharing)
85
+
86
  ### Limitations
87
  The SpeechBrain team does not provide any warranty on the performance achieved by this model when used on other datasets.
88
 
 
95
  year = {2021},
96
  publisher = {GitHub},
97
  journal = {GitHub repository},
98
+ howpublished = {\\url{https://github.com/speechbrain/speechbrain}},
99
  }
100
  ```
101