fix arguments

2022-09-10 14:24:41 +05:30 · 2022-09-10 14:24:41 +05:30 · 05fe7ec317
parent 619d0be8ce
commit 05fe7ec317
1 changed files with 2 additions and 1 deletions
--- 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)