From e9ea0d16956daed7ef792d4bfcb293a4db70fe30 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Mon, 26 Sep 2022 17:09:11 +0530 Subject: [PATCH] pass total frames --- enhancer/models/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/enhancer/models/model.py b/enhancer/models/model.py index 83355fd..65946a2 100644 --- a/enhancer/models/model.py +++ b/enhancer/models/model.py @@ -199,7 +199,7 @@ class Model(pl.LightningModule): window_size = round(duration * model_sampling_rate) batched_waveform = Inference.batchify(waveform,window_size,step_size=step_size) batch_prediction = self.infer(batched_waveform,batch_size=batch_size) - waveform = Inference.aggreagate(batch_prediction,window_size,step_size) + waveform = Inference.aggreagate(batch_prediction,window_size,waveform.shape[-1],step_size,) if save_output and isinstance(audio,(str,Path)): Inference.write_output(waveform,audio,model_sampling_rate)