urfm-vega-preview / planner.json
JacobLinCool's picture
URFM Vega Preview: weights, model card and license
2497c45 verified
Raw History Blame Contribute Delete
1.15 kB
{
"config": {
"beat_dim": 256,
"beat_pool": "mean",
"cond_dim": 7,
"d_model": 160,
"dropout": 0.1,
"ff_mult": 4,
"games": [
"maimai_dx",
"osu_catch",
"osu_mania",
"osu_std",
"osu_taiko",
"sm_bm_double5",
"sm_bm_double7",
"sm_bm_single5",
"sm_bm_single7",
"sm_bm_versus5",
"sm_bm_versus7",
"sm_dance_couple",
"sm_dance_double",
"sm_dance_routine",
"sm_dance_single",
"sm_dance_solo",
"sm_dance_threepanel",
"sm_ds3ddx_single",
"sm_ez2_double",
"sm_ez2_real",
"sm_ez2_single",
"sm_kb7_single",
"sm_lights_cabinet",
"sm_maniax_double",
"sm_maniax_single",
"sm_para_single",
"sm_pnm_five",
"sm_pnm_nine",
"sm_pump_couple",
"sm_pump_double",
"sm_pump_halfdouble",
"sm_pump_routine",
"sm_pump_single",
"sm_techno_double4",
"sm_techno_double5",
"sm_techno_double8",
"sm_techno_single4",
"sm_techno_single5",
"sm_techno_single8",
"taiko"
],
"head_sizes": [
16,
5,
5,
5,
11,
16,
4,
6,
5,
5
],
"in_dim": 490,
"layers": 4,
"max_sections": 384,
"n_heads": 2,
"z_dim": 16
}
}