training_config_summary.json

{
  "repo_name": "pour_pi05_base",
  "hf_repo_id": "ColinSkywalker/pour_pi05_base",
  "source_slurm_job_id": 9753605,
  "train_config_name": "pi05_ur5_pour_pytorch_baseline",
  "exp_name": "pi05_ur5_pour_pytorch_baseline_tandon_2gpu_b16",
  "dataset_path": "/scratch/yz11445/real_robot_data/ur5_place_and_pour_nuts_camera_shifts",
  "dataset_repo_id": "ur5_place_and_pour_nuts_camera_shifts",
  "norm_stats_source": "/scratch/yz11445/pi05_ur5_real_robot/ur5_place_and_pour_nuts_camera_shifts",
  "checkpoint_base_dir": "/scratch/yz11445/tmp/openpi_cam/checkpoints",
  "release_checkpoint_source_dir": "/scratch/yz11445/tmp/openpi_cam/checkpoints/pi05_ur5_pour_pytorch_baseline/pi05_ur5_pour_pytorch_baseline_tandon_2gpu_b16",
  "base_model_path": "/scratch/yz11445/pi05_base",
  "num_gpus": 2,
  "batch_size": 16,
  "num_train_steps": 30000,
  "save_interval": 5000,
  "keep_period": 5000,
  "completed_checkpoint_steps": [
    20000,
    25000,
    30000
  ],
  "available_source_checkpoint_steps_at_packaging_time": [
    5000,
    10000,
    15000,
    20000,
    25000,
    30000
  ],
  "reserved_release_checkpoint_steps": [],
  "base_camera_key": "observation.images.context_left_rgb",
  "wrist_camera_key": "observation.images.wrist_right_rgb",
  "prompt_from_task": true,
  "delta_actions_first_6_joints_only": true,
  "pose_enc_type": "null",
  "ray_enc_type": false,
  "view_enc_type": false,
  "cross_view_type": "none",
  "uses_gt_point_targets": false,
  "aux_point_head_enabled": false,
  "wandb_project": "openpi_cam_real_robot",
  "wandb_entity": "NYU-robotics",
  "wandb_id": "kq34yc5t",
  "serve_command_example": "uv run scripts/serve_policy.py policy:checkpoint --policy.config=pi05_ur5_pour_pytorch_baseline --policy.dir=/path/to/pour_pi05_base/checkpoints/30000",
  "train_command_reference": "uv run python scripts/train_pytorch.py pi05_ur5_pour_pytorch_baseline --exp_name pi05_ur5_pour_pytorch_baseline_tandon_2gpu_b16",
  "norm_stats_command_reference": "uv run scripts/compute_norm_stats.py --config-name pi05_ur5_pour_pytorch_baseline",
  "run_status": "completed"
}