cli

enhancer/cli/train.py

@@ -1,11 +1,12 @@
 import os
 from types import MethodType
+
 import hydra
 from hydra.utils import instantiate
 from omegaconf import DictConfig
-from torch.optim.lr_scheduler import ReduceLROnPlateau
+from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint
-from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping
 from pytorch_lightning.loggers import MLFlowLogger
+from torch.optim.lr_scheduler import ReduceLROnPlateau
 
 os.environ["HYDRA_FULL_ERROR"] = "1"
 JOB_ID = os.environ.get("SLURM_JOBID", "0")

enhancer/cli/train_config/dataset/DNS-2020.yaml

@@ -10,4 +10,3 @@ files:
   test_clean : clean_test_wav
   train_noisy : clean_test_wav
   test_noisy : clean_test_wav
-

enhancer/cli/train_config/dataset/Vctk.yaml

@@ -10,6 +10,3 @@ files:
   test_clean : clean_testset_wav
   train_noisy : noisy_trainset_28spk_wav
   test_noisy : noisy_testset_wav
-
-
-

enhancer/cli/train_config/hyperparameters/default.yaml

@@ -5,4 +5,3 @@ ReduceLr_patience : 5
 ReduceLr_factor : 0.1
 min_lr : 0.000001
 EarlyStopping_factor : 10
-

enhancer/cli/train_config/model/Demucs.yaml

@@ -14,5 +14,3 @@ encoder_decoder:
 lstm:
   bidirectional: False
   num_layers: 2
-
-