From 05fe7ec317efd2be43697815244ccd151a2205c9 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Sat, 10 Sep 2022 14:24:41 +0530 Subject: [PATCH] fix arguments --- 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 6172714..ce9b8c5 100644 --- a/enhancer/models/demucs.py +++ b/enhancer/models/demucs.py @@ -57,7 +57,8 @@ class Demucs(Model): ): super().__init__(num_channels=num_channels, - sampling_rate=sampling_rate,lr=lr,dataset=dataset,loss) + sampling_rate=sampling_rate,lr=lr, + dataset=dataset,loss=loss) encoder_decoder = merge_dict(self.ED_DEFAULTS,encoder_decoder) lstm = merge_dict(self.LSTM_DEFAULTS,lstm)