2021-02-15 09:26:21 +01:00

13 lines
392 B
Python

from pathlib import Path
import variables as v
from datasets.primates_librosa_datamodule import PrimatesLibrosaDatamodule
data_root = Path() / 'data'
if __name__ == '__main__':
dataset = PrimatesLibrosaDatamodule(data_root, batch_size=25, num_worker=6,
sr=v.sr, n_mels=64, n_fft=512, hop_length=256)
dataset.prepare_data()
print('done')