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

rmv inplace operation

This commit is contained in:
shahules786 2022-10-11 15:11:50 +05:30
parent c996798dd3
commit fdce8bb601
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
enhancer/models/demucs.py
View File

@ -226,7 +226,7 @@ class Demucs(Model):
x = x.permute(0, 2, 1) x = x.permute(0, 2, 1)
for decoder in self.decoder: for decoder in self.decoder:
skip_connection = encoder_outputs.pop(-1) skip_connection = encoder_outputs.pop(-1)
x += skip_connection[..., : x.shape[-1]] x = x + skip_connection[..., : x.shape[-1]]
x = decoder(x) x = decoder(x)
if self.hparams.resample > 1: if self.hparams.resample > 1:
@ -236,7 +236,8 @@ class Demucs(Model):
self.hparams.sampling_rate, self.hparams.sampling_rate,
) )
return x[..., :length] out = x[..., :length]
return out
def get_padding_length(self, input_length): def get_padding_length(self, input_length):