Spijkervet / CLMR

Official PyTorch implementation of Contrastive Learning of Musical Representations

Home Page:https://arxiv.org/abs/2103.09410

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

train by raw dataset

Breeze-Zero opened this issue · comments

Hi,I want to use this method to implement a feature of my graduation project. But how can i train a model using a raw dataset such as Universal Music? Looking forward to your reply

commented

Hi there!

You can follow the example in one of the unit tests of this repository:
https://github.com/Spijkervet/CLMR/blob/master/tests/test_audioset.py

You can use from clmr.datasets import AUDIO as a generic audio PyTorch dataloader for your music library :)