diff --git a/look2hear/datas/musdb_moisesdb_datamodule.py b/look2hear/datas/musdb_moisesdb_datamodule.py index ef3267e..85c9527 100644 --- a/look2hear/datas/musdb_moisesdb_datamodule.py +++ b/look2hear/datas/musdb_moisesdb_datamodule.py @@ -120,7 +120,7 @@ def __getitem__(self, idx: int) -> Tuple[torch.Tensor, torch.Tensor]: start = random.randint(0, music_wav.shape[-1] - self.segments) ori_wav = music_wav[:, start:start+self.segments] - codec_wav = codec_simu(ori_wav, sr=self.sr, codec_type=self.codec_type, options=self.codec_options) + codec_wav = codec_simu(ori_wav, sr=self.sr, options=self.codec_options) max_scale = max(ori_wav.abs().max(), codec_wav.abs().max()) @@ -212,4 +212,4 @@ def val_dataloader(self) -> DataLoader: num_workers=self.hparams.num_workers, shuffle=False, pin_memory=True, - ) \ No newline at end of file + )