This model is finetuned version of ntu-spml/distilhubert for music detection.
This model was finetuned to distinguish between music and non-music audio.
Training hyperparameters
The following hyperparameters were used during training:
- metric_for_best_model = accuracy
- learning_rate = 5e-5
- seed = 42
- per_device_train_batch_size = 4
- per_device_eval_batch_size = 4
- gradient_accumulation_steps = 1
- warmup_ratio = 0.1
- fp16 = True
- adam_epsilon = 1e-08
- adam_beta1 = 0.9
- adam_beta2 = 0.999
- Downloads last month
- 417
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.