General
- Python (verified on 3.9)
- CUDA (verified on 11.8)
Python Packages
- refer to requirements.txt
conda create -n envs python=3.9
conda activate envs
pip install -r requirements.txtData structure:
datasets
|
├── sentence-level
│ ├── PHEE
│ │ ├── train
│ │ │ ├── train-0.wav
│ │ │ ├── ...
│ │ ├── dev
│ │ │ ├── dev-0.wav
│ │ │ ├── ...
│ │ ├── test
│ │ │ ├── test-0.wav
│ │ │ ├── ...
|
├── document-level ...
|
├── dialogue-level ...
The command for training model is as follows:
bash run.bash -d 0 -f tree -m t5-large --label_smoothing 0 -l 5e-5 --lr_scheduler linear --warmup_steps 2000 -b 16 -i PHEE-drefers to the GPU device id.-m t5-largerefers to using T5-large as the textual decoder.-irefers to the dataset.