m1ku2's picture
Upload from_pretrain/config.json with huggingface_hub
e3454b6 verified
Raw
History Blame Contribute Delete
1.19 kB
{
"model": "pi05",
"framework": "openpi-jax",
"training_type": "finetune_from_pi05_base",
"base_checkpoint": "/project/peilab/wam/Xqh_data/openpi/pretrained/pi05_base/params",
"config_name": "pi05_pack_3_objects_plus_hdf5_both_arms",
"dataset_id": "hdf5_pack_3_objects_plus_perfect_both_arms_pi05",
"exp_name": "pi05_pack_3_objects_plus_hdf5_both_arms",
"project_name": "pi0_real",
"task_prompt_fallback": "pack 3 objects",
"keep_arm": "both",
"model_variant": "pi05",
"batch_size": 32,
"num_workers": 8,
"num_train_steps": 10000,
"log_interval": 10,
"save_interval": 2500,
"keep_period": 5000,
"kept_checkpoints": [
5000,
9999
],
"final_checkpoint": 9999,
"checkpoint_format": "orbax",
"checkpoint_path_in_repo": "pi05_pack_3_objects_plus/from_pretrain/9999",
"weights_path_in_repo": "pi05_pack_3_objects_plus/from_pretrain/9999/params",
"train_state_path_in_repo": "pi05_pack_3_objects_plus/from_pretrain/9999/train_state",
"assets_path_in_repo": "pi05_pack_3_objects_plus/from_pretrain/9999/assets",
"wandb_project": "pi0_real",
"wandb_offline_run": "wandb/offline-run-20260531_022219-ejhdnzm5",
"slurm_job_id": 431204
}