Skip to content

Commit

Permalink
refactor(config): change default log dir from logs to logs_and_checkp…
Browse files Browse the repository at this point in the history
…oints

fixes #39
  • Loading branch information
roedoejet committed Sep 7, 2023
1 parent c5d9ff7 commit be74a8c
Show file tree
Hide file tree
Showing 15 changed files with 23 additions and 18 deletions.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
__pycache__
.coverage
logs
logs_and_checkpoints
*.ckpt
.vscode
*egg-info
Expand Down
4 changes: 2 additions & 2 deletions docs/guides/custom.md
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,7 @@ Which would use the GPU accelerator and specify 1 device/chip.

To generate audio when you train your feature prediction network, you need to add your vocoder checkpoint to the `config/feature_prediction.yaml`

At the bottom of that file you'll find a key called vocoder_path. Add the absolute path to your trained vocder (here it would be `/path/to/test/logs/VocoderExperiment/base/checkpoints/last.ckpt` where `/path/to` would be the actual path to it on your computer.)
At the bottom of that file you'll find a key called vocoder_path. Add the absolute path to your trained vocder (here it would be `/path/to/test/logs_and_checkpoints/VocoderExperiment/base/checkpoints/last.ckpt` where `/path/to` would be the actual path to it on your computer.)

Once you've replaced the vocoder_path key, you can train your feature prediction network:

Expand All @@ -93,7 +93,7 @@ everyvoice fs2 train -p config/feature_prediction.yaml
You can synthesize by pointing the CLI to your trained feature prediction network and passing in the text. You can export to wav, npy, or pt files.

```bash
everyvoice fs2 synthesize logs/FeaturePredictionExperiment/base/checkpoints/last.ckpt -t "මෙදා සැරේ සාකච්ඡාවක් විදියට නෙවෙයි නේද පල කරල තියෙන්නෙ" -a gpu -d 1 -O wav
everyvoice fs2 synthesize logs_and_checkpoints/FeaturePredictionExperiment/base/checkpoints/last.ckpt -t "මෙදා සැරේ සාකච්ඡාවක් විදියට නෙවෙයි නේද පල කරල තියෙන්නෙ" -a gpu -d 1 -O wav
```

<!-- % Step 10 (optional): Finetune your vocoder
Expand Down
6 changes: 3 additions & 3 deletions everyvoice/config/base/base_shared.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ aligner:
filelist_loader: "everyvoice.utils.generic_dict_loader"
logger:
name: "BaseExperiment"
save_dir: "./logs"
save_dir: "./logs_and_checkpoints"
sub_dir: "everyvoice.utils.get_current_time"
version: "base"
val_data_workers: 0
Expand Down Expand Up @@ -135,7 +135,7 @@ feature_prediction:
filelist_loader: "everyvoice.utils.generic_dict_loader"
logger:
name: "BaseExperiment"
save_dir: "./logs"
save_dir: "./logs_and_checkpoints"
sub_dir: "everyvoice.utils.get_current_time"
version: "base"
val_data_workers: 0
Expand Down Expand Up @@ -176,7 +176,7 @@ vocoder:
filelist_loader: "everyvoice.utils.generic_dict_loader"
logger:
name: "BaseExperiment"
save_dir: "./logs"
save_dir: "./logs_and_checkpoints"
sub_dir: "everyvoice.utils.get_current_time"
version: "base"
val_data_workers: 0
Expand Down
2 changes: 1 addition & 1 deletion everyvoice/config/default/aligner.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ training:
filelist_loader: "everyvoice.utils.generic_dict_loader"
logger:
name: "BaseExperiment"
save_dir: "./logs"
save_dir: "./logs_and_checkpoints"
sub_dir: "everyvoice.utils.get_current_time"
version: "base"
val_data_workers: 0
Expand Down
2 changes: 1 addition & 1 deletion everyvoice/config/default/fastspeech2.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -102,7 +102,7 @@ training:
filelist_loader: "everyvoice.utils.generic_dict_loader"
logger:
name: "BaseExperiment"
save_dir: "./logs"
save_dir: "./logs_and_checkpoints"
sub_dir: "everyvoice.utils.get_current_time"
version: "base"
val_data_workers: 0
Expand Down
2 changes: 1 addition & 1 deletion everyvoice/config/default/hifigan.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ training:
filelist_loader: "everyvoice.utils.generic_dict_loader"
logger:
name: "BaseExperiment"
save_dir: "./logs"
save_dir: "./logs_and_checkpoints"
sub_dir: "everyvoice.utils.get_current_time"
version: "base"
val_data_workers: 0
Expand Down
2 changes: 1 addition & 1 deletion everyvoice/config/default/istft.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ training:
filelist_loader: "everyvoice.utils.generic_dict_loader"
logger:
name: "Base-iSTFT-Experiment"
save_dir: "./logs"
save_dir: "./logs_and_checkpoints"
sub_dir: "everyvoice.utils.get_current_time"
version: "base"
val_data_workers: 0
Expand Down
2 changes: 1 addition & 1 deletion everyvoice/config/shared_types.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,7 @@ def __init__(self, **data) -> None:

class LoggerConfig(ConfigModel):
name: str = "BaseExperiment"
save_dir: DirectoryPath = Path("./logs")
save_dir: DirectoryPath = Path("./logs_and_checkpoints")
sub_dir: str = "everyvoice.utils.get_current_time"
version: str = "base"

Expand Down
2 changes: 1 addition & 1 deletion everyvoice/model/aligner/DeepForcedAligner
6 changes: 3 additions & 3 deletions everyvoice/model/e2e/config/lj_training.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -5,13 +5,13 @@ ckpt_epochs: 1
max_epochs: 1000
seed: 1234
finetune_checkpoint: null
vocoder_checkpoint: "./logs/LJ/base-vocoder/checkpoints/last.ckpt"
feature_prediction_checkpoint: "./logs/LJ/fastspeech2-base/checkpoints/last.ckpt"
vocoder_checkpoint: "./logs_and_checkpoints/LJ/base-vocoder/checkpoints/last.ckpt"
feature_prediction_checkpoint: "./logs_and_checkpoints/LJ/fastspeech2-base/checkpoints/last.ckpt"
filelist: "./preprocessed/LJ/processed_filelist.psv"
filelist_loader: "everyvoice.utils.generic_dict_loader"
logger:
name: "LJ"
save_dir: "./logs"
save_dir: "./logs_and_checkpoints"
sub_dir: "everyvoice.utils.get_current_time"
version: "e2e"
val_data_workers: 0
Expand Down
2 changes: 1 addition & 1 deletion everyvoice/model/e2e/config/training.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ filelist: "./preprocessed/LJ/processed_filelist.psv"
filelist_loader: "everyvoice.utils.generic_dict_loader"
logger:
name: "BaseExperiment"
save_dir: "./logs"
save_dir: "./logs_and_checkpoints"
sub_dir: "everyvoice.utils.get_current_time"
version: "e2e"
val_data_workers: 0
Expand Down
3 changes: 3 additions & 0 deletions everyvoice/tests/test_wizard.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,9 @@ def test_config_format_effect(self):
]
config_step.state["dataset_test"]["sox_effects"] = []
config_step.effect()
self.assertTrue(
(Path(tmpdirname) / config_step.name / "logs_and_checkpoints").exists()
)

def test_access_response(self):
root_step = Step(
Expand Down
3 changes: 2 additions & 1 deletion everyvoice/wizard/basic.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,8 @@ def effect(self):
config_dir = (output_path / "config").absolute()
config_dir.mkdir(exist_ok=True, parents=True)
# log dir
log_dir = (output_path / "logs").absolute()
log_dir = LoggerConfig().save_dir.stem
log_dir = (output_path / log_dir).absolute()
log_dir.mkdir(parents=True, exist_ok=True)
datasets = []
# Text Configuration
Expand Down

0 comments on commit be74a8c

Please sign in to comment.