NikkeDoy/SISU
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

⚡ | Made training bit faster.

Browse Source
This commit is contained in:
NikkeDoy
2025-06-07 20:43:52 +03:00
parent 2ded03713d
commit 03fdc050cc
3 changed files with 27 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
data.py
View File

@ -11,7 +11,7 @@ import AudioUtils
class AudioDataset(Dataset):
audio_sample_rates = [11025]
def __init__(self, input_dir, device, clip_length = 256):
def __init__(self, input_dir, device, clip_length = 1024):
self.device = device
input_files = [os.path.join(root, f) for root, _, files in os.walk(input_dir) for f in files if f.endswith('.wav') or f.endswith('.mp3') or f.endswith('.flac')]