fix arguments

This commit is contained in:
shahules786 2022-09-10 14:24:41 +05:30
parent 619d0be8ce
commit 05fe7ec317
1 changed files with 2 additions and 1 deletions

View File

@ -57,7 +57,8 @@ class Demucs(Model):
): ):
super().__init__(num_channels=num_channels, 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) encoder_decoder = merge_dict(self.ED_DEFAULTS,encoder_decoder)
lstm = merge_dict(self.LSTM_DEFAULTS,lstm) lstm = merge_dict(self.LSTM_DEFAULTS,lstm)