diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml
index 9c464ac..77877c6 100644
--- a/enhancer/cli/train_config/config.yaml
+++ b/enhancer/cli/train_config/config.yaml
@@ -1,6 +1,6 @@
 defaults:
-  - model : Demucs
-  - dataset : DNS-2020
+  - model : DCCRN
+  - dataset : Vctk
   - optimizer : Adam
   - hyperparameters : default
   - trainer : default
diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml
index 356181c..80fb033 100644
--- a/enhancer/cli/train_config/hyperparameters/default.yaml
+++ b/enhancer/cli/train_config/hyperparameters/default.yaml
@@ -1,6 +1,6 @@
 loss : si-snr
 metric : [stoi,pesq]
-lr : 0.0003
+lr : 0.001
 ReduceLr_patience : 10
 Early_stop : False
 ReduceLr_factor : 0.2
diff --git a/enhancer/cli/train_config/mlflow/experiment.yaml b/enhancer/cli/train_config/mlflow/experiment.yaml
index 87e7895..da0150e 100644
--- a/enhancer/cli/train_config/mlflow/experiment.yaml
+++ b/enhancer/cli/train_config/mlflow/experiment.yaml
@@ -1,2 +1,2 @@
 experiment_name : shahules/enhancer
-run_name : Demucs-DNS30
+run_name : DCCRN-Vctk28
diff --git a/enhancer/utils/transforms.py b/enhancer/utils/transforms.py
index f8e4b50..ffd2bc8 100644
--- a/enhancer/utils/transforms.py
+++ b/enhancer/utils/transforms.py
@@ -86,6 +86,7 @@ class ConviSTFT(ConvFFT):
         out = F.conv_transpose1d(input, self.weight, stride=self.hop_size)
         coeff = self.window.unsqueeze(1).repeat(1, 1, input.size(-1)) ** 2
         coeff.to(input.device)
+        self.enframe.to(input.device)
         coeff = F.conv_transpose1d(coeff, self.enframe, stride=self.hop_size)
         out = out / (coeff + 1e-8)
         pad = self.window_len - self.hop_size