Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

change config

This commit is contained in:
shahules786 2022-09-30 10:15:28 +05:30
parent 610c23a0eb
commit 7b0e7e2312
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cli/train_config/config.yaml
View File

@ -1,5 +1,5 @@
defaults: defaults:
- model : WaveUnet - model : Demucs
- dataset : Vctk - dataset : Vctk
- optimizer : Adam - optimizer : Adam
- hyperparameters : default - hyperparameters : default

2
cli/train_config/hyperparameters/default.yaml
View File

@ -1,4 +1,4 @@
loss : mse loss : mse
metric : mae metric : mae
lr : 0.001 lr : 0.0001
num_epochs : 100 num_epochs : 100

4
cli/train_config/trainer/default.yaml
View File

@ -9,7 +9,7 @@ benchmark: False
check_val_every_n_epoch: 1 check_val_every_n_epoch: 1
detect_anomaly: False detect_anomaly: False
deterministic: False deterministic: False
devices: auto devices: -1
enable_checkpointing: True enable_checkpointing: True
enable_model_summary: True enable_model_summary: True
enable_progress_bar: True enable_progress_bar: True
@ -22,7 +22,7 @@ limit_predict_batches: 1.0
limit_test_batches: 1.0 limit_test_batches: 1.0
limit_train_batches: 1.0 limit_train_batches: 1.0
limit_val_batches: 1.0 limit_val_batches: 1.0
log_every_n_steps: 50 log_every_n_steps: 10
max_epochs: 100 max_epochs: 100
max_steps: null max_steps: null
max_time: null max_time: null