ncm-mlx-tinyworld / generator.json
janmusic's picture
Add TinyWorld codec, generator, AR baseline and model card
8ed7b11 verified
Raw History Blame Contribute Delete
4.75 kB
{
"stage": "generator",
"steps": 3000,
"best_step": 1500,
"train_seconds": 266.11,
"history": [
{
"step": 250,
"train_loss": 0.4347,
"val_weighted_ce": 0.4357,
"ce_per_level": [
4.8339,
2.1513,
0.7181,
0.3265,
0.2223,
0.2898
],
"acc_per_level": [
0.024,
0.2745,
0.7007,
0.8765,
0.9264,
0.8988
]
},
{
"step": 500,
"train_loss": 0.3446,
"val_weighted_ce": 0.3768,
"ce_per_level": [
4.822,
2.1152,
0.6384,
0.2698,
0.1767,
0.2234
],
"acc_per_level": [
0.024,
0.2885,
0.728,
0.8917,
0.9391,
0.9212
]
},
{
"step": 750,
"train_loss": 0.2984,
"val_weighted_ce": 0.3549,
"ce_per_level": [
4.812,
2.0659,
0.6166,
0.2465,
0.1576,
0.2018
],
"acc_per_level": [
0.021,
0.2915,
0.7297,
0.9016,
0.9442,
0.93
]
},
{
"step": 1000,
"train_loss": 0.2932,
"val_weighted_ce": 0.349,
"ce_per_level": [
4.8004,
2.0642,
0.5901,
0.2355,
0.1539,
0.1985
],
"acc_per_level": [
0.024,
0.2905,
0.7297,
0.9035,
0.9488,
0.9313
]
},
{
"step": 1250,
"train_loss": 0.2581,
"val_weighted_ce": 0.3501,
"ce_per_level": [
4.8017,
2.072,
0.5819,
0.2336,
0.1582,
0.1995
],
"acc_per_level": [
0.024,
0.289,
0.7315,
0.9044,
0.9473,
0.9344
]
},
{
"step": 1500,
"train_loss": 0.2341,
"val_weighted_ce": 0.3454,
"ce_per_level": [
4.8075,
2.0351,
0.5787,
0.23,
0.152,
0.1968
],
"acc_per_level": [
0.024,
0.296,
0.729,
0.9035,
0.9514,
0.9362
]
},
{
"step": 1750,
"train_loss": 0.2297,
"val_weighted_ce": 0.3487,
"ce_per_level": [
4.7977,
2.0335,
0.575,
0.23,
0.1583,
0.201
],
"acc_per_level": [
0.024,
0.2975,
0.7225,
0.9022,
0.9519,
0.9383
]
},
{
"step": 2000,
"train_loss": 0.2242,
"val_weighted_ce": 0.3545,
"ce_per_level": [
4.8037,
2.0176,
0.5667,
0.2335,
0.1621,
0.2114
],
"acc_per_level": [
0.024,
0.3005,
0.7375,
0.9024,
0.9544,
0.939
]
},
{
"step": 2250,
"train_loss": 0.2063,
"val_weighted_ce": 0.3587,
"ce_per_level": [
4.7992,
2.0185,
0.5751,
0.2345,
0.1659,
0.2166
],
"acc_per_level": [
0.024,
0.295,
0.7315,
0.9009,
0.9536,
0.9408
]
},
{
"step": 2500,
"train_loss": 0.2111,
"val_weighted_ce": 0.3625,
"ce_per_level": [
4.7993,
2.0127,
0.5736,
0.2358,
0.1702,
0.2222
],
"acc_per_level": [
0.024,
0.3055,
0.7215,
0.9005,
0.9541,
0.9408
]
},
{
"step": 2750,
"train_loss": 0.2049,
"val_weighted_ce": 0.3649,
"ce_per_level": [
4.7984,
2.0105,
0.5747,
0.2359,
0.1722,
0.2259
],
"acc_per_level": [
0.024,
0.3035,
0.718,
0.901,
0.9543,
0.9411
]
},
{
"step": 3000,
"train_loss": 0.2027,
"val_weighted_ce": 0.3689,
"ce_per_level": [
4.7987,
2.0092,
0.5774,
0.2376,
0.1753,
0.2315
],
"acc_per_level": [
0.024,
0.304,
0.7202,
0.8986,
0.955,
0.9415
]
}
],
"train": {
"ce_per_level": [
4.8224,
1.8823,
0.4374,
0.1278,
0.0369,
0.046
],
"acc_per_level": [
0.022,
0.3345,
0.7897,
0.9413,
0.9871,
0.9844
]
},
"val": {
"ce_per_level": [
4.8075,
2.0351,
0.5787,
0.23,
0.152,
0.1968
],
"acc_per_level": [
0.024,
0.296,
0.729,
0.9035,
0.9514,
0.9362
]
}
}