Spaces:
Sleeping
Sleeping
primepake
commited on
Commit
·
0cae0b4
1
Parent(s):
3ba181f
update readme dac
Browse files- dac-vae/README.md +2 -6
dac-vae/README.md
CHANGED
|
@@ -107,10 +107,7 @@ dataset/
|
|
| 107 |
```bash
|
| 108 |
torchrun --nnodes=1 --nproc_per_node=4 train.py \
|
| 109 |
--run_id my_experiment \
|
| 110 |
-
--config configs/configx2.yml
|
| 111 |
-
--dataset_path /path/to/dataset \
|
| 112 |
-
--num_epochs 200 \
|
| 113 |
-
--batch_size 32
|
| 114 |
```
|
| 115 |
|
| 116 |
## Evaluation
|
|
@@ -120,8 +117,7 @@ Evaluate model performance using:
|
|
| 120 |
```bash
|
| 121 |
python3 evaluate.py \
|
| 122 |
--checkpoint checkpoint.pt \
|
| 123 |
-
--test_dir /path/to/test/audio
|
| 124 |
-
--metrics pesq stoi
|
| 125 |
```
|
| 126 |
|
| 127 |
## Pretrained Models
|
|
|
|
| 107 |
```bash
|
| 108 |
torchrun --nnodes=1 --nproc_per_node=4 train.py \
|
| 109 |
--run_id my_experiment \
|
| 110 |
+
--config configs/configx2.yml
|
|
|
|
|
|
|
|
|
|
| 111 |
```
|
| 112 |
|
| 113 |
## Evaluation
|
|
|
|
| 117 |
```bash
|
| 118 |
python3 evaluate.py \
|
| 119 |
--checkpoint checkpoint.pt \
|
| 120 |
+
--test_dir /path/to/test/audio
|
|
|
|
| 121 |
```
|
| 122 |
|
| 123 |
## Pretrained Models
|