:albemic: | Tests.

This commit is contained in:
2025-03-25 19:50:51 +02:00
parent 7e1c7e935a
commit 54338e55a9
4 changed files with 21 additions and 17 deletions

View File

@ -9,7 +9,7 @@ import AudioUtils
class AudioDataset(Dataset):
audio_sample_rates = [11025]
MAX_LENGTH = 88200 # Define your desired maximum length here
MAX_LENGTH = 44100 # Define your desired maximum length here
def __init__(self, input_dir, device):
self.input_files = [os.path.join(root, f) for root, _, files in os.walk(input_dir) for f in files if f.endswith('.wav')]