cinematic-music-descriptor-v2-all-checkpoints

suyashnpande/cinematic-music-descriptor-v2-all-checkpoints

cinematic-music-descriptor-v2-all-checkpoints is a machine learning model from suyashnpande. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.

This repository consolidates all training checkpoints for the Cinematic Music Descriptor v2 pipeline (Modules 1, 2, 3).

Downloads · 30 days

Access

Public

Updated Apr 10, 2026

Repo size

2.5 GB

Likes

Public

Hugging Face

Repo makeup

Click a slice to open those files.

.pt2.5 GB · 100%

At a glance

License: apache-2.0
Access: Public
Created: Apr 10, 2026
Updated: Apr 10, 2026
SHA: 9fe6f2c7

Model card

From the Hugging Face model README

Cinematic Music Descriptor v2 — All Checkpoints

This repository consolidates all training checkpoints for the Cinematic Music Descriptor v2 pipeline (Modules 1, 2, 3).

Repository Layout

module1/
    module1_mlm_final.pt          ← Phase 2a: MLM pre-training
    module1_finetune_final.pt     ← Phase 2b: Supervised fine-tuning
    module1_regression_final.pt   ← Phase 2c: Regression heads
    module1_e2e_final.pt          ← Phase 6:  After E2E joint training

module2/
    module2_pretrain_final.pt     ← Phase 3a: Masked scene pre-training
    module2_finetune_final.pt     ← Phase 3b: Supervised fine-tuning
    module2_e2e_best.pt           ← Phase 5:  E2E best checkpoint
    module2_e2e_final.pt          ← Phase 6:  Final E2E checkpoint

module3/
    module3_m3_final.pt           ← Phase 4:  M3 standalone training
    module3_e2e_best.pt           ← Phase 5:  E2E best checkpoint  ★ recommended
    module3_e2e_final.pt          ← Phase 6:  Final E2E checkpoint

Architecture Summary

Module	Role	Backbone
Module 1	Scene-level encoding & classification	RoBERTa-base + task heads
Module 2	Cross-scene narrative context	Transformer encoder (4L × 8H)
Module 3	Music descriptor prediction	Gated fusion (M1 + M2) + heads

Recommended Checkpoint Combination

For best end-to-end performance, load:

module1/module1_e2e_final.pt
module2/module2_e2e_final.pt
module3/module3_e2e_best.pt ← or module3_e2e_final.pt

Source Repositories

Originally spread across:

suyashnpande/cinematic-music-descriptor-v2-module1
suyashnpande/cinematic-music-descriptor-v2-module2
suyashnpande/cinematic-music-descriptor-v2-module3

License: apache-2.0
Created: Apr 10, 2026
Updated: Apr 10, 2026