speechbrainteam commited on
Commit
860e4a3
·
1 Parent(s): 4005852

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +22 -0
README.md CHANGED
@@ -57,6 +57,28 @@ 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
 
 
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-whamr.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
  ### Limitations
83
  The SpeechBrain team does not provide any warranty on the performance achieved by this model when used on other datasets.
84