| { |
| "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_stack_3_cups_hdf5_both_arms", |
| "dataset_id": "hdf5_stack_3_cups_perfect_both_arms_pi05", |
| "exp_name": "pi05_stack_3_cups_hdf5_both_arms", |
| "project_name": "pi0_real", |
| "task_prompt_fallback": "stack 3 cups", |
| "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": "from_pretrain/9999", |
| "weights_path_in_repo": "from_pretrain/9999/params", |
| "train_state_path_in_repo": "from_pretrain/9999/train_state", |
| "assets_path_in_repo": "from_pretrain/9999/assets", |
| "wandb_project": "pi0_real", |
| "slurm_job_id": 439479 |
| } |
|
|
