wavenet trainig

2022-09-30 10:27:11 +05:30 · 2022-09-30 10:27:11 +05:30 · 6aa502c3dc
parent 7b0e7e2312
commit 6aa502c3dc
2 changed files with 2 additions and 2 deletions
--- a/cli/train_config/config.yaml
+++ b/cli/train_config/config.yaml
@ -1,5 +1,5 @@
 defaults:
-  - model : Demucs
+  - model : WaveUnet
  - dataset : Vctk
  - optimizer : Adam
  - hyperparameters : default
--- a/cli/train_config/trainer/default.yaml
+++ b/cli/train_config/trainer/default.yaml
@ -2,7 +2,7 @@ _target_: pytorch_lightning.Trainer
 accelerator: auto
 accumulate_grad_batches: 1
 amp_backend: native
-auto_lr_find: False
+auto_lr_find: True
 auto_scale_batch_size: False
 auto_select_gpus: True
 benchmark: False