RoboDojo-pi05-checkpoints / metadata /checkpoints.json
cjgogo's picture
Add metadata/checkpoints.json
01e1b5e verified
Raw
History Blame Contribute Delete
1.93 kB
{
"repo_id": "cjgogo/RoboDojo-pi05-checkpoints",
"format": "JAX Orbax OCDBT",
"base_model": "Physical Intelligence pi05_base",
"checkpoints": [
{
"path": "checkpoints/sim-10task/9000",
"step": 9000,
"training_type": "full_finetune",
"config": "pi05_base_aloha_full_sim_arx-x5_seed_0",
"approx_size_gib": 42,
"epochs": 5.43,
"includes_train_state": true
},
{
"path": "checkpoints/sim-10task/14077",
"step": 14077,
"training_type": "full_finetune",
"config": "pi05_base_aloha_full_sim_arx-x5_seed_0",
"approx_size_gib": 42,
"epochs": 8.5,
"includes_train_state": true
},
{
"path": "checkpoints/sim-10task/15000",
"step": 15000,
"training_type": "full_finetune_resume",
"config": "pi05_base_aloha_full_sim_arx-x5_seed_0",
"approx_size_gib": 42,
"epochs": 9.06,
"includes_train_state": true
},
{
"path": "checkpoints/real-piper6-lora/3000",
"step": 3000,
"training_type": "jax_lora",
"config": "pi05_base_piper6_lora_real",
"approx_size_gib": 9,
"includes_train_state": true
},
{
"path": "checkpoints/real-piper6-lora/6000",
"step": 6000,
"training_type": "jax_lora",
"config": "pi05_base_piper6_lora_real",
"approx_size_gib": 9,
"includes_train_state": true
},
{
"path": "checkpoints/real-piper6-lora/7594",
"step": 7594,
"training_type": "jax_lora",
"config": "pi05_base_piper6_lora_real",
"approx_size_gib": 9,
"epochs": 8.0,
"includes_train_state": true
}
],
"excluded": [
{
"name": "RoboDojo-sim-arx_x5-joint-0/59999",
"reason": "pre-existing training base, not produced by these runs"
},
{
"name": "17077.orbax-checkpoint-tmp-4",
"reason": "incomplete Orbax temporary checkpoint"
}
]
}