From 8230061b7bd60d2cd9d0b648ef453efc2c367e2c Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Thu, 8 Sep 2022 11:34:36 +0530
Subject: [PATCH] include lr

---
 enhancer/models/demucs.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/enhancer/models/demucs.py b/enhancer/models/demucs.py
index ed7e38f..bcc8214 100644
--- a/enhancer/models/demucs.py
+++ b/enhancer/models/demucs.py
@@ -51,11 +51,12 @@ class Demucs(Model):
         num_channels:int=1,
         resample:int=4,
         sampling_rate = 16000,
+        lr:float=1e-3,
         dataset:Optional[EnhancerDataset]=None,
 
     ):
         super().__init__(num_channels=num_channels,
-                            sampling_rate=sampling_rate,dataset=dataset)
+                            sampling_rate=sampling_rate,lr=lr,dataset=dataset)
         
         encoder_decoder = merge_dict(self.ED_DEFAULTS,encoder_decoder)
         lstm = merge_dict(self.LSTM_DEFAULTS,lstm)