atom-ai / docs /artifacts /batched_training /bench_cpu_1thread.json
thefinalboss's picture
ATOM (atom-ai): code snapshot (100fae0 + batched-train), README with measured numbers, curated checkpoints
9f1076a verified
Raw History Blame Contribute Delete
2.86 kB
[
{
"B": 1,
"legacy": true,
"device": "cpu",
"threads": 1,
"K": 64,
"d_model": 32,
"ticks": 64,
"repeat": 3,
"seconds_all": [
5.868604458999471,
5.509067286999198,
1.4524448690062854
],
"ticks_per_s": 44.063634610645614,
"bytes_per_s": 44.063634610645614,
"bytes_per_s_median": 11.617211528897654,
"cpu_seconds_all": [
0.5674520940000001,
0.5674846260000002,
0.532370089,
0.53237274,
0.40030953100000044,
0.40031202599999993
],
"bytes_per_cpu_s": 159.87628333535713,
"peak_rss_mb": 346.66796875
},
{
"B": 1,
"legacy": false,
"device": "cpu",
"threads": 1,
"K": 64,
"d_model": 32,
"ticks": 64,
"repeat": 3,
"seconds_all": [
0.221989019002649,
0.23102366499369964,
0.21758131100796163
],
"ticks_per_s": 294.14291008504006,
"bytes_per_s": 294.14291008504006,
"bytes_per_s_median": 288.30254887173624,
"cpu_seconds_all": [
0.21286655600000004,
0.21518449499999992,
0.21053944599999985
],
"bytes_per_cpu_s": 303.9810411584347,
"peak_rss_mb": 344.921875
},
{
"B": 8,
"legacy": false,
"device": "cpu",
"threads": 1,
"K": 64,
"d_model": 32,
"ticks": 64,
"repeat": 3,
"seconds_all": [
0.3558585329883499,
0.31862772699969355,
0.3167166780040134
],
"ticks_per_s": 202.0733496048762,
"bytes_per_s": 1616.5867968390096,
"bytes_per_s_median": 1606.890915681335,
"cpu_seconds_all": [
0.33745186500000024,
0.3063102510000002,
0.3027095759999998
],
"bytes_per_cpu_s": 1691.3901659985818,
"peak_rss_mb": 424.00390625
},
{
"B": 32,
"legacy": false,
"device": "cpu",
"threads": 1,
"K": 64,
"d_model": 32,
"ticks": 64,
"repeat": 3,
"seconds_all": [
7.107633487001294,
6.839838102998328,
7.253203468993888
],
"ticks_per_s": 9.356946617193293,
"bytes_per_s": 299.42229175018537,
"bytes_per_s_median": 288.1409126885142,
"cpu_seconds_all": [
0.6856104250000001,
0.6589649569999998,
0.6974502499999997
],
"bytes_per_cpu_s": 3107.9042644751753,
"peak_rss_mb": 715.79296875
},
{
"B": 128,
"legacy": false,
"device": "cpu",
"threads": 1,
"K": 64,
"d_model": 32,
"ticks": 64,
"repeat": 3,
"seconds_all": [
4.8719542829931015,
1.3274299780023284,
1.4088259730051504
],
"ticks_per_s": 48.213465915780105,
"bytes_per_s": 6171.3236372198535,
"bytes_per_s_median": 5814.770707645132,
"cpu_seconds_all": [
1.3365111499999998,
1.2526937870000001,
1.2874514720000008
],
"bytes_per_cpu_s": 6539.507168482507,
"peak_rss_mb": 1826.109375
}
]