search

Spijkervet / CLMR

Official PyTorch implementation of Contrastive Learning of Musical Representations
https://arxiv.org/abs/2103.09410
Apache License 2.0
309 stars 48 forks source link

train by raw dataset #4

Closed Breeze-Zero closed 3 years ago

Breeze-Zero commented 3 years ago

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

Spijkervet commented 3 years ago

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 :)