plan-bind-d1b / bench_a5000_gpu.json
thanhkt's picture
Upload bench_a5000_gpu.json with huggingface_hub
f0eda3a verified
Raw History Blame Contribute Delete
4.35 kB
{
"device": "cuda",
"hardware": "NVIDIA RTX A5000",
"amp": false,
"fps": 30,
"torch": "2.11.0+cu128",
"policies": {
"bsp10": {
"path": "/home/25thanh.tk/lerobot-nexarm/experiments/261029-plan-bind-d1b-seeds/outputs/full/train_bsp10_s1000/checkpoints/last/pretrained_model",
"mode": "diffusion",
"num_inference_steps": 10,
"stages": {
"preprocess": {
"p50_ms": 2.7998340083286166,
"p95_ms": 2.9755288269370794,
"mean_ms": 2.7930569218005985,
"max_ms": 4.527234006673098
},
"encode": {
"p50_ms": 6.839177454821765,
"p95_ms": 7.612548471661285,
"mean_ms": 6.974750514491461,
"max_ms": 8.704309002496302
},
"denoise": {
"p50_ms": 70.37655444582924,
"p95_ms": 70.9241526434198,
"mean_ms": 70.40382484090514,
"max_ms": 71.88348297495395
},
"decode": {
"p50_ms": 0.47736900160089135,
"p95_ms": 0.6213226297404616,
"mean_ms": 0.5017320089973509,
"max_ms": 1.0303580202162266
}
},
"call": {
"p50_ms": 80.58626699494198,
"p95_ms": 81.57202625880018,
"mean_ms": 80.67336428619456,
"max_ms": 83.95737095270306
},
"latency_frames": 3,
"select_action": {
"p50_ms": 0.11398701462894678,
"p95_ms": 0.19491465063765592,
"mean_ms": 3.69553081585433,
"max_ms": 80.66570805385709
}
},
"onestep": {
"path": "/home/25thanh.tk/lerobot-nexarm/experiments/261029-plan-bind-d1b-seeds/outputs/full/train_onestep_s1000/checkpoints/last/pretrained_model",
"mode": "onestep",
"num_inference_steps": null,
"stages": {
"preprocess": {
"p50_ms": 2.9188795015215874,
"p95_ms": 4.729061241960151,
"mean_ms": 3.173999734572135,
"max_ms": 6.545683019794524
},
"encode": {
"p50_ms": 7.0853125071153045,
"p95_ms": 11.853664572117848,
"mean_ms": 7.544838071335107,
"max_ms": 12.780867982655764
},
"onestep": {
"p50_ms": 8.412808529101312,
"p95_ms": 10.002501198323444,
"mean_ms": 8.626534593640827,
"max_ms": 11.075549991801381
}
},
"call": {
"p50_ms": 18.603673379402608,
"p95_ms": 24.759409314719953,
"mean_ms": 19.34537239954807,
"max_ms": 27.342656976543367
},
"latency_frames": 1,
"select_action": {
"p50_ms": 0.2976925461553037,
"p95_ms": 17.047313024522737,
"mean_ms": 4.4017699774651025,
"max_ms": 22.707096999511123
}
},
"planbind": {
"path": "/home/25thanh.tk/lerobot-nexarm/experiments/261029-plan-bind-d1b-seeds/outputs/full/train_planbind_s1000/checkpoints/last/pretrained_model",
"mode": "planbind",
"num_inference_steps": null,
"stages": {
"preprocess": {
"p50_ms": 2.9204170568846166,
"p95_ms": 4.587121552322059,
"mean_ms": 3.131448646890931,
"max_ms": 6.867332034744322
},
"encode": {
"p50_ms": 7.030489505268633,
"p95_ms": 10.291650285944337,
"mean_ms": 7.428209136705846,
"max_ms": 19.658743985928595
},
"plan": {
"p50_ms": 8.009099983610213,
"p95_ms": 8.848073234548792,
"mean_ms": 8.12908087100368,
"max_ms": 11.86974800657481
},
"bind": {
"p50_ms": 8.146500447764993,
"p95_ms": 8.703204413177444,
"mean_ms": 8.269510077661835,
"max_ms": 11.68749202042818
}
},
"call": {
"p50_ms": 26.26900450559333,
"p95_ms": 31.512535951333117,
"mean_ms": 26.95824873226229,
"max_ms": 40.295892977155745
},
"latency_frames": 1,
"select_action": {
"p50_ms": 0.28910301625728607,
"p95_ms": 16.414664668263867,
"mean_ms": 4.391406258483502,
"max_ms": 23.49968603812158
},
"bind_only_call": {
"p50_ms": 18.2112890179269,
"p95_ms": 22.895342129049823,
"mean_ms": 18.82916786125861,
"max_ms": 31.269520055502653
}
}
}
}