Skip to content

Commit

Permalink
remove channel code
Browse files Browse the repository at this point in the history
  • Loading branch information
Sean1572 authored Jul 15, 2024
1 parent 148c442 commit ea24ae6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pyha_analyzer/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit ea24ae6

Please sign in to comment.