From b70911d7001d7aa0bfb82d5d6f0b5883ccc1066d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Kai=20Li=20=28=E6=9D=8E=E5=87=AF=29?= Date: Wed, 30 Oct 2024 17:28:30 +0800 Subject: [PATCH] Update musdb_moisesdb_datamodule.py --- look2hear/datas/musdb_moisesdb_datamodule.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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 + )