Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

include lr

This commit is contained in:
shahules786 2022-09-08 11:34:36 +05:30
parent 05d75c0485
commit 8230061b7b
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
enhancer/models/demucs.py
View File

@ -51,11 +51,12 @@ class Demucs(Model):
num_channels:int=1, num_channels:int=1,
resample:int=4, resample:int=4,
sampling_rate = 16000, sampling_rate = 16000,
lr:float=1e-3,
dataset:Optional[EnhancerDataset]=None, dataset:Optional[EnhancerDataset]=None,
): ):
super().__init__(num_channels=num_channels, 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) encoder_decoder = merge_dict(self.ED_DEFAULTS,encoder_decoder)
lstm = merge_dict(self.LSTM_DEFAULTS,lstm) lstm = merge_dict(self.LSTM_DEFAULTS,lstm)