From ea24ae6c8997fa6addce3ca91246cbd6ab36638e Mon Sep 17 00:00:00 2001 From: Sean Perry Date: Mon, 15 Jul 2024 16:36:47 -0700 Subject: [PATCH] remove channel code --- pyha_analyzer/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyha_analyzer/dataset.py b/pyha_analyzer/dataset.py index 54f1d26..d9397e0 100644 --- a/pyha_analyzer/dataset.py +++ b/pyha_analyzer/dataset.py @@ -256,7 +256,7 @@ def to_image(self, audio): # Sigmoid to get 0 to 1 scaling (0.5 becomes mean) mel = torch.sigmoid(mel) - return mel.unsqueeze(0) #torch.stack([mel, mel, mel]) + return torch.stack([mel, mel, mel]) def __getitem__(self, index): #-> Any: """ Takes an index and returns tuple of spectrogram image with corresponding label