Views
No views yet
.pt bundles the model config together with the weights, and is loaded by ModelBase.from_pretrained in the codebase.| File | Model | Role |
|---|---|---|
stage_one_pretrained.pt | 1D harmony skeleton model (12 layers, hidden 768, 87M) | Samples harmony skeletons |
stage_two_pretrained.pt | 3D symphony model (33 layers, hidden 512, 124M, 2-stream cross-attention) | Pretrained on the SymphonyNet Dataset; starting point for GRPO / finetuning |
grpo_clamp_epoch_10.pt | 3D symphony model ("reinforced") | GRPO with the pure CLaMP 3 audio reward (paper round-1 subjective test) |
grpo_clamp+track_epoch_6.pt | 3D symphony model ("reinforced+track") | GRPO with the composite CLaMP 3 + track density reward (paper round-2 subjective test) |
(λ_hn, λ_nn) = (1, 10); use register decay (--register_decay 1) with reinforced+track and --register_decay 0 with reinforced.1git clone https://github.com/symphonygen/symphonygen
2cd symphonygen && pip install -r requirements.txt
3export PYTHONPATH=$(pwd)
4
5# Download the checkpoints into $ASSET_DIR (default: asset/)
6hf download SymphonyGen/SymphonyGen --local-dir asset/
7
8# Generate a harmony skeleton, then orchestrate it
9python arch/harmo/generator.py asset/stage_one_pretrained.pt --batch_size 4 --save_dir harmony_out
10python arch/symph/generator.py asset/grpo_clamp+track_epoch_6.pt harmony_out \
11 --group_size 1 --save_dir songs --forbid_piano
12
13# Or re-orchestrate any MIDI by analyzing its harmony skeleton
14python arch/symph/generator.py asset/grpo_clamp+track_epoch_6.pt your_song.mid \
15 --analyze_harmo --save_dir songs--analyze_harmo.DIFF.md for implementation details that go beyond the paper's description.Results/ in the codebase.Reproducibility disclaimer: the objective metrics depend on the exact implementation of the harmony analysis and of the harmony-skeleton filters (seeDIFF.mdin the code repository). As a result, you might see reproduced objective evaluation numbers that differ from the paper's tables. The trends, however, should be the same.
1@inproceedings{symphonygen2026,
2 title = {SymphonyGen: 3D Hierarchical Orchestral Generation with Controllable Harmony Skeleton},
3 author = {He, Xuzheng and Nan, Nan and Wang, Zhilin and Kang, Ziyue and Mo, Zhuoru and Li, Ao and Pan, Yu and Li, Xiaobing and Yu, Feng and Guan, Xiaohong},
4 booktitle = {Proceedings of the 27th International Society for Music Information Retrieval Conference (ISMIR)},
5 year = {2026}
6}