From dab68de260b2eb8ccc44914a393ea599f4ad76a8 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Sun, 16 Oct 2022 12:04:43 +0530 Subject: [PATCH] config --- enhancer/cli/train_config/dataset/DNS-2020.yaml | 2 +- enhancer/cli/train_config/trainer/default.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/enhancer/cli/train_config/dataset/DNS-2020.yaml b/enhancer/cli/train_config/dataset/DNS-2020.yaml index 0dbf4d1..4c6bfee 100644 --- a/enhancer/cli/train_config/dataset/DNS-2020.yaml +++ b/enhancer/cli/train_config/dataset/DNS-2020.yaml @@ -4,7 +4,7 @@ name : dns-2020 duration : 5.0 sampling_rate: 16000 batch_size: 64 -valid_size: 0.05 +valid_size: 0.025 files: train_clean : CleanSpeech_training test_clean : CleanSpeech_testing diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index 0842ad4..d4917e8 100644 --- a/enhancer/cli/train_config/trainer/default.yaml +++ b/enhancer/cli/train_config/trainer/default.yaml @@ -23,7 +23,7 @@ limit_test_batches: 1.0 limit_train_batches: 1.0 limit_val_batches: 1.0 log_every_n_steps: 100 -max_epochs: 100 +max_epochs: 250 max_time: 00:47:00:00 min_epochs: 1 min_steps: null