diff --git a/004000/pretrained_model/README.md b/004000/pretrained_model/README.md new file mode 100644 index 0000000000000000000000000000000000000000..6565338d7b6577ddbee192786d397d22079b8338 --- /dev/null +++ b/004000/pretrained_model/README.md @@ -0,0 +1,204 @@ +--- +library_name: peft +tags: +- base_model:adapter:lerobot/pi0_base +- lora +--- + +# Model Card for Model ID + + + + + +## Model Details + +### Model Description + + + + + +- **Developed by:** [More Information Needed] +- **Funded by [optional]:** [More Information Needed] +- **Shared by [optional]:** [More Information Needed] +- **Model type:** [More Information Needed] +- **Language(s) (NLP):** [More Information Needed] +- **License:** [More Information Needed] +- **Finetuned from model [optional]:** [More Information Needed] + +### Model Sources [optional] + + + +- **Repository:** [More Information Needed] +- **Paper [optional]:** [More Information Needed] +- **Demo [optional]:** [More Information Needed] + +## Uses + + + +### Direct Use + + + +[More Information Needed] + +### Downstream Use [optional] + + + +[More Information Needed] + +### Out-of-Scope Use + + + +[More Information Needed] + +## Bias, Risks, and Limitations + + + +[More Information Needed] + +### Recommendations + + + +Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations. + +## How to Get Started with the Model + +Use the code below to get started with the model. + +[More Information Needed] + +## Training Details + +### Training Data + + + +[More Information Needed] + +### Training Procedure + + + +#### Preprocessing [optional] + +[More Information Needed] + + +#### Training Hyperparameters + +- **Training regime:** [More Information Needed] + +#### Speeds, Sizes, Times [optional] + + + +[More Information Needed] + +## Evaluation + + + +### Testing Data, Factors & Metrics + +#### Testing Data + + + +[More Information Needed] + +#### Factors + + + +[More Information Needed] + +#### Metrics + + + +[More Information Needed] + +### Results + +[More Information Needed] + +#### Summary + + + +## Model Examination [optional] + + + +[More Information Needed] + +## Environmental Impact + + + +Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700). + +- **Hardware Type:** [More Information Needed] +- **Hours used:** [More Information Needed] +- **Cloud Provider:** [More Information Needed] +- **Compute Region:** [More Information Needed] +- **Carbon Emitted:** [More Information Needed] + +## Technical Specifications [optional] + +### Model Architecture and Objective + +[More Information Needed] + +### Compute Infrastructure + +[More Information Needed] + +#### Hardware + +[More Information Needed] + +#### Software + +[More Information Needed] + +## Citation [optional] + + + +**BibTeX:** + +[More Information Needed] + +**APA:** + +[More Information Needed] + +## Glossary [optional] + + + +[More Information Needed] + +## More Information [optional] + +[More Information Needed] + +## Model Card Authors [optional] + +[More Information Needed] + +## Model Card Contact + +[More Information Needed] +### Framework versions + +- PEFT 0.18.1 \ No newline at end of file diff --git a/004000/pretrained_model/adapter_config.json b/004000/pretrained_model/adapter_config.json new file mode 100644 index 0000000000000000000000000000000000000000..3c00acefa2527617e9a485e421d89b9166fed77b --- /dev/null +++ b/004000/pretrained_model/adapter_config.json @@ -0,0 +1,41 @@ +{ + "alora_invocation_tokens": null, + "alpha_pattern": {}, + "arrow_config": null, + "auto_mapping": { + "base_model_class": "PI0Policy", + "parent_library": "lerobot.policies.pi0.modeling_pi0" + }, + "base_model_name_or_path": "lerobot/pi0_base", + "bias": "none", + "corda_config": null, + "ensure_weight_tying": false, + "eva_config": null, + "exclude_modules": null, + "fan_in_fan_out": false, + "inference_mode": true, + "init_lora_weights": true, + "layer_replication": null, + "layers_pattern": null, + "layers_to_transform": null, + "loftq_config": {}, + "lora_alpha": 8, + "lora_bias": false, + "lora_dropout": 0.0, + "megatron_config": null, + "megatron_core": "megatron.core", + "modules_to_save": [], + "peft_type": "LORA", + "peft_version": "0.18.1", + "qalora_group_size": 16, + "r": 32, + "rank_pattern": {}, + "revision": null, + "target_modules": "(.*\\.gemma_expert\\..*\\.self_attn\\.(q|v)_proj|model\\.(state_proj|action_in_proj|action_out_proj|action_time_mlp_in|action_time_mlp_out))", + "target_parameters": null, + "task_type": null, + "trainable_token_indices": null, + "use_dora": false, + "use_qalora": false, + "use_rslora": false +} \ No newline at end of file diff --git a/004000/pretrained_model/adapter_model.safetensors b/004000/pretrained_model/adapter_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..5b4cf59609ed0f5ddb93e633c8da3bcf395d0386 --- /dev/null +++ b/004000/pretrained_model/adapter_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d3b10f26df22c335e6b20da23524f5fcb6fbda167ad095b59ebdc474e8dfb487 +size 11101592 diff --git a/004000/pretrained_model/config.json b/004000/pretrained_model/config.json new file mode 100644 index 0000000000000000000000000000000000000000..5bded5dca90b3c498f402c92f09b4af980008111 --- /dev/null +++ b/004000/pretrained_model/config.json @@ -0,0 +1,113 @@ +{ + "type": "pi0", + "n_obs_steps": 1, + "input_features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "device": "cuda", + "use_amp": false, + "use_peft": true, + "push_to_hub": true, + "repo_id": "F-Fer/pi0-xlerobot-0", + "private": null, + "tags": null, + "license": null, + "pretrained_path": "lerobot/pi0_base", + "paligemma_variant": "gemma_2b", + "action_expert_variant": "gemma_300m", + "dtype": "bfloat16", + "chunk_size": 50, + "n_action_steps": 50, + "max_state_dim": 32, + "max_action_dim": 32, + "num_inference_steps": 10, + "time_sampling_beta_alpha": 1.5, + "time_sampling_beta_beta": 1.0, + "time_sampling_scale": 0.999, + "time_sampling_offset": 0.001, + "min_period": 0.004, + "max_period": 4.0, + "use_relative_actions": false, + "relative_exclude_joints": [ + "gripper" + ], + "action_feature_names": [ + "left_shoulder_pan.pos", + "left_shoulder_lift.pos", + "left_elbow_flex.pos", + "left_wrist_flex.pos", + "left_wrist_roll.pos", + "left_gripper.pos", + "right_shoulder_pan.pos", + "right_shoulder_lift.pos", + "right_elbow_flex.pos", + "right_wrist_flex.pos", + "right_wrist_roll.pos", + "right_gripper.pos" + ], + "rtc_config": null, + "image_resolution": [ + 224, + 224 + ], + "empty_cameras": 0, + "normalization_mapping": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + }, + "gradient_checkpointing": true, + "compile_model": true, + "compile_mode": "max-autotune", + "freeze_vision_encoder": false, + "train_expert_only": false, + "optimizer_lr": 0.001, + "optimizer_betas": [ + 0.9, + 0.95 + ], + "optimizer_eps": 1e-08, + "optimizer_weight_decay": 0.01, + "optimizer_grad_clip_norm": 1.0, + "scheduler_warmup_steps": 1000, + "scheduler_decay_steps": 30000, + "scheduler_decay_lr": 0.0001, + "tokenizer_max_length": 48 +} \ No newline at end of file diff --git a/004000/pretrained_model/policy_postprocessor.json b/004000/pretrained_model/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..25a2e623f8277a52e915c3701c7cd9cf77ddf8d6 --- /dev/null +++ b/004000/pretrained_model/policy_postprocessor.json @@ -0,0 +1,32 @@ +{ + "name": "policy_postprocessor", + "steps": [ + { + "registry_name": "unnormalizer_processor", + "config": { + "eps": 1e-08, + "features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "norm_map": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + } + }, + "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors" + }, + { + "registry_name": "device_processor", + "config": { + "device": "cpu", + "float_dtype": null + } + } + ] +} \ No newline at end of file diff --git a/004000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/004000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..08107a06a9226905f59e970a3d28ddc923a42695 --- /dev/null +++ b/004000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:794c084955a7613103705a47253d9fa3f66d03ead8978fa493cad135498ca62b +size 9192 diff --git a/004000/pretrained_model/policy_preprocessor.json b/004000/pretrained_model/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..8c39dc4710ff54b6f6d1631e34f0484383336eca --- /dev/null +++ b/004000/pretrained_model/policy_preprocessor.json @@ -0,0 +1,91 @@ +{ + "name": "policy_preprocessor", + "steps": [ + { + "registry_name": "rename_observations_processor", + "config": { + "rename_map": { + "observation.images.left_front": "observation.images.base_0_rgb", + "observation.images.left_left": "observation.images.left_wrist_0_rgb", + "observation.images.left_right": "observation.images.right_wrist_0_rgb" + } + } + }, + { + "registry_name": "to_batch_processor", + "config": {} + }, + { + "registry_name": "pi0_new_line_processor", + "config": {} + }, + { + "registry_name": "tokenizer_processor", + "config": { + "max_length": 48, + "task_key": "task", + "padding_side": "right", + "padding": "max_length", + "truncation": true, + "tokenizer_name": "google/paligemma-3b-pt-224" + } + }, + { + "registry_name": "device_processor", + "config": { + "device": "cuda", + "float_dtype": null + } + }, + { + "registry_name": "normalizer_processor", + "config": { + "eps": 1e-08, + "features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + }, + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "norm_map": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + } + }, + "state_file": "policy_preprocessor_step_5_normalizer_processor.safetensors" + } + ] +} \ No newline at end of file diff --git a/004000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors b/004000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..08107a06a9226905f59e970a3d28ddc923a42695 --- /dev/null +++ b/004000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:794c084955a7613103705a47253d9fa3f66d03ead8978fa493cad135498ca62b +size 9192 diff --git a/004000/pretrained_model/train_config.json b/004000/pretrained_model/train_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8b15c723555caefa3bdb9e5e9097937ebfef3fad --- /dev/null +++ b/004000/pretrained_model/train_config.json @@ -0,0 +1,261 @@ +{ + "dataset": { + "repo_id": "F-Fer/xlerobot-0", + "root": null, + "episodes": null, + "image_transforms": { + "enable": false, + "max_num_transforms": 3, + "random_order": false, + "tfs": { + "brightness": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "brightness": [ + 0.8, + 1.2 + ] + } + }, + "contrast": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "contrast": [ + 0.8, + 1.2 + ] + } + }, + "saturation": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "saturation": [ + 0.5, + 1.5 + ] + } + }, + "hue": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "hue": [ + -0.05, + 0.05 + ] + } + }, + "sharpness": { + "weight": 1.0, + "type": "SharpnessJitter", + "kwargs": { + "sharpness": [ + 0.5, + 1.5 + ] + } + }, + "affine": { + "weight": 1.0, + "type": "RandomAffine", + "kwargs": { + "degrees": [ + -5.0, + 5.0 + ], + "translate": [ + 0.05, + 0.05 + ] + } + } + } + }, + "revision": null, + "use_imagenet_stats": true, + "video_backend": "torchcodec", + "streaming": false + }, + "env": null, + "policy": { + "type": "pi0", + "n_obs_steps": 1, + "input_features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "device": "cuda", + "use_amp": false, + "use_peft": true, + "push_to_hub": true, + "repo_id": "F-Fer/pi0-xlerobot-0", + "private": null, + "tags": null, + "license": null, + "pretrained_path": "lerobot/pi0_base", + "paligemma_variant": "gemma_2b", + "action_expert_variant": "gemma_300m", + "dtype": "bfloat16", + "chunk_size": 50, + "n_action_steps": 50, + "max_state_dim": 32, + "max_action_dim": 32, + "num_inference_steps": 10, + "time_sampling_beta_alpha": 1.5, + "time_sampling_beta_beta": 1.0, + "time_sampling_scale": 0.999, + "time_sampling_offset": 0.001, + "min_period": 0.004, + "max_period": 4.0, + "use_relative_actions": false, + "relative_exclude_joints": [ + "gripper" + ], + "action_feature_names": [ + "left_shoulder_pan.pos", + "left_shoulder_lift.pos", + "left_elbow_flex.pos", + "left_wrist_flex.pos", + "left_wrist_roll.pos", + "left_gripper.pos", + "right_shoulder_pan.pos", + "right_shoulder_lift.pos", + "right_elbow_flex.pos", + "right_wrist_flex.pos", + "right_wrist_roll.pos", + "right_gripper.pos" + ], + "rtc_config": null, + "image_resolution": [ + 224, + 224 + ], + "empty_cameras": 0, + "normalization_mapping": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + }, + "gradient_checkpointing": true, + "compile_model": true, + "compile_mode": "max-autotune", + "freeze_vision_encoder": false, + "train_expert_only": false, + "optimizer_lr": 0.001, + "optimizer_betas": [ + 0.9, + 0.95 + ], + "optimizer_eps": 1e-08, + "optimizer_weight_decay": 0.01, + "optimizer_grad_clip_norm": 1.0, + "scheduler_warmup_steps": 1000, + "scheduler_decay_steps": 30000, + "scheduler_decay_lr": 0.0001, + "tokenizer_max_length": 48 + }, + "output_dir": "outputs/train/2026-04-10/14-19-24_pi0", + "job_name": "pi0", + "resume": false, + "seed": 1000, + "cudnn_deterministic": false, + "num_workers": 4, + "batch_size": 4, + "steps": 10000, + "eval_freq": -1, + "log_freq": 100, + "tolerance_s": 0.0001, + "save_checkpoint": true, + "save_freq": 1000, + "use_policy_training_preset": true, + "optimizer": { + "type": "adamw", + "lr": 0.001, + "weight_decay": 0.01, + "grad_clip_norm": 1.0, + "betas": [ + 0.9, + 0.95 + ], + "eps": 1e-08 + }, + "scheduler": { + "type": "cosine_decay_with_warmup", + "num_warmup_steps": 1000, + "num_decay_steps": 30000, + "peak_lr": 0.001, + "decay_lr": 0.0001 + }, + "eval": { + "n_episodes": 50, + "batch_size": 50, + "use_async_envs": false + }, + "wandb": { + "enable": true, + "disable_artifact": false, + "project": "xlerobot", + "entity": null, + "notes": null, + "run_id": "tmqkc0d7", + "mode": null, + "add_tags": true + }, + "peft": { + "target_modules": null, + "full_training_modules": null, + "method_type": "LORA", + "init_type": null, + "r": 32 + }, + "use_rabc": false, + "rabc_progress_path": null, + "rabc_kappa": 0.01, + "rabc_epsilon": 1e-06, + "rabc_head_mode": "sparse", + "rename_map": { + "observation.images.left_front": "observation.images.base_0_rgb", + "observation.images.left_left": "observation.images.left_wrist_0_rgb", + "observation.images.left_right": "observation.images.right_wrist_0_rgb" + }, + "checkpoint_path": null +} \ No newline at end of file diff --git a/004000/training_state/optimizer_param_groups.json b/004000/training_state/optimizer_param_groups.json new file mode 100644 index 0000000000000000000000000000000000000000..43101a53bb1d1a432b75386c6c1e7b1f94cd2003 --- /dev/null +++ b/004000/training_state/optimizer_param_groups.json @@ -0,0 +1,881 @@ +[ + { + "lr": 0.0006890576474687264, + "betas": [ + 0.9, + 0.95 + ], + "eps": 1e-08, + "weight_decay": 0.01, + "amsgrad": false, + "maximize": false, + "foreach": null, + "capturable": false, + "differentiable": false, + "fused": null, + "decoupled_weight_decay": true, + "initial_lr": 0.001, + "params": [ + 0, + 1, + 2, + 3, + 4, + 5, + 6, + 7, + 8, + 9, + 10, + 11, + 12, + 13, + 14, + 15, + 16, + 17, + 18, + 19, + 20, + 21, + 22, + 23, + 24, + 25, + 26, + 27, + 28, + 29, + 30, + 31, + 32, + 33, + 34, + 35, + 36, + 37, + 38, + 39, + 40, + 41, + 42, + 43, + 44, + 45, + 46, + 47, + 48, + 49, + 50, + 51, + 52, + 53, + 54, + 55, + 56, + 57, + 58, + 59, + 60, + 61, + 62, + 63, + 64, + 65, + 66, + 67, + 68, + 69, + 70, + 71, + 72, + 73, + 74, + 75, + 76, + 77, + 78, + 79, + 80, + 81, + 82, + 83, + 84, + 85, + 86, + 87, + 88, + 89, + 90, + 91, + 92, + 93, + 94, + 95, + 96, + 97, + 98, + 99, + 100, + 101, + 102, + 103, + 104, + 105, + 106, + 107, + 108, + 109, + 110, + 111, + 112, + 113, + 114, + 115, + 116, + 117, + 118, + 119, + 120, + 121, + 122, + 123, + 124, + 125, + 126, + 127, + 128, + 129, + 130, + 131, + 132, + 133, + 134, + 135, + 136, + 137, + 138, + 139, + 140, + 141, + 142, + 143, + 144, + 145, + 146, + 147, + 148, + 149, + 150, + 151, + 152, + 153, + 154, + 155, + 156, + 157, + 158, + 159, + 160, + 161, + 162, + 163, + 164, + 165, + 166, + 167, + 168, + 169, + 170, + 171, + 172, + 173, + 174, + 175, + 176, + 177, + 178, + 179, + 180, + 181, + 182, + 183, + 184, + 185, + 186, + 187, + 188, + 189, + 190, + 191, + 192, + 193, + 194, + 195, + 196, + 197, + 198, + 199, + 200, + 201, + 202, + 203, + 204, + 205, + 206, + 207, + 208, + 209, + 210, + 211, + 212, + 213, + 214, + 215, + 216, + 217, + 218, + 219, + 220, + 221, + 222, + 223, + 224, + 225, + 226, + 227, + 228, + 229, + 230, + 231, + 232, + 233, + 234, + 235, + 236, + 237, + 238, + 239, + 240, + 241, + 242, + 243, + 244, + 245, + 246, + 247, + 248, + 249, + 250, + 251, + 252, + 253, + 254, + 255, + 256, + 257, + 258, + 259, + 260, + 261, + 262, + 263, + 264, + 265, + 266, + 267, + 268, + 269, + 270, + 271, + 272, + 273, + 274, + 275, + 276, + 277, + 278, + 279, + 280, + 281, + 282, + 283, + 284, + 285, + 286, + 287, + 288, + 289, + 290, + 291, + 292, + 293, + 294, + 295, + 296, + 297, + 298, + 299, + 300, + 301, + 302, + 303, + 304, + 305, + 306, + 307, + 308, + 309, + 310, + 311, + 312, + 313, + 314, + 315, + 316, + 317, + 318, + 319, + 320, + 321, + 322, + 323, + 324, + 325, + 326, + 327, + 328, + 329, + 330, + 331, + 332, + 333, + 334, + 335, + 336, + 337, + 338, + 339, + 340, + 341, + 342, + 343, + 344, + 345, + 346, + 347, + 348, + 349, + 350, + 351, + 352, + 353, + 354, + 355, + 356, + 357, + 358, + 359, + 360, + 361, + 362, + 363, + 364, + 365, + 366, + 367, + 368, + 369, + 370, + 371, + 372, + 373, + 374, + 375, + 376, + 377, + 378, + 379, + 380, + 381, + 382, + 383, + 384, + 385, + 386, + 387, + 388, + 389, + 390, + 391, + 392, + 393, + 394, + 395, + 396, + 397, + 398, + 399, + 400, + 401, + 402, + 403, + 404, + 405, + 406, + 407, + 408, + 409, + 410, + 411, + 412, + 413, + 414, + 415, + 416, + 417, + 418, + 419, + 420, + 421, + 422, + 423, + 424, + 425, + 426, + 427, + 428, + 429, + 430, + 431, + 432, + 433, + 434, + 435, + 436, + 437, + 438, + 439, + 440, + 441, + 442, + 443, + 444, + 445, + 446, + 447, + 448, + 449, + 450, + 451, + 452, + 453, + 454, + 455, + 456, + 457, + 458, + 459, + 460, + 461, + 462, + 463, + 464, + 465, + 466, + 467, + 468, + 469, + 470, + 471, + 472, + 473, + 474, + 475, + 476, + 477, + 478, + 479, + 480, + 481, + 482, + 483, + 484, + 485, + 486, + 487, + 488, + 489, + 490, + 491, + 492, + 493, + 494, + 495, + 496, + 497, + 498, + 499, + 500, + 501, + 502, + 503, + 504, + 505, + 506, + 507, + 508, + 509, + 510, + 511, + 512, + 513, + 514, + 515, + 516, + 517, + 518, + 519, + 520, + 521, + 522, + 523, + 524, + 525, + 526, + 527, + 528, + 529, + 530, + 531, + 532, + 533, + 534, + 535, + 536, + 537, + 538, + 539, + 540, + 541, + 542, + 543, + 544, + 545, + 546, + 547, + 548, + 549, + 550, + 551, + 552, + 553, + 554, + 555, + 556, + 557, + 558, + 559, + 560, + 561, + 562, + 563, + 564, + 565, + 566, + 567, + 568, + 569, + 570, + 571, + 572, + 573, + 574, + 575, + 576, + 577, + 578, + 579, + 580, + 581, + 582, + 583, + 584, + 585, + 586, + 587, + 588, + 589, + 590, + 591, + 592, + 593, + 594, + 595, + 596, + 597, + 598, + 599, + 600, + 601, + 602, + 603, + 604, + 605, + 606, + 607, + 608, + 609, + 610, + 611, + 612, + 613, + 614, + 615, + 616, + 617, + 618, + 619, + 620, + 621, + 622, + 623, + 624, + 625, + 626, + 627, + 628, + 629, + 630, + 631, + 632, + 633, + 634, + 635, + 636, + 637, + 638, + 639, + 640, + 641, + 642, + 643, + 644, + 645, + 646, + 647, + 648, + 649, + 650, + 651, + 652, + 653, + 654, + 655, + 656, + 657, + 658, + 659, + 660, + 661, + 662, + 663, + 664, + 665, + 666, + 667, + 668, + 669, + 670, + 671, + 672, + 673, + 674, + 675, + 676, + 677, + 678, + 679, + 680, + 681, + 682, + 683, + 684, + 685, + 686, + 687, + 688, + 689, + 690, + 691, + 692, + 693, + 694, + 695, + 696, + 697, + 698, + 699, + 700, + 701, + 702, + 703, + 704, + 705, + 706, + 707, + 708, + 709, + 710, + 711, + 712, + 713, + 714, + 715, + 716, + 717, + 718, + 719, + 720, + 721, + 722, + 723, + 724, + 725, + 726, + 727, + 728, + 729, + 730, + 731, + 732, + 733, + 734, + 735, + 736, + 737, + 738, + 739, + 740, + 741, + 742, + 743, + 744, + 745, + 746, + 747, + 748, + 749, + 750, + 751, + 752, + 753, + 754, + 755, + 756, + 757, + 758, + 759, + 760, + 761, + 762, + 763, + 764, + 765, + 766, + 767, + 768, + 769, + 770, + 771, + 772, + 773, + 774, + 775, + 776, + 777, + 778, + 779, + 780, + 781, + 782, + 783, + 784, + 785, + 786, + 787, + 788, + 789, + 790, + 791, + 792, + 793, + 794, + 795, + 796, + 797, + 798, + 799, + 800, + 801, + 802, + 803, + 804, + 805, + 806, + 807, + 808, + 809, + 810, + 811, + 812, + 813, + 814, + 815, + 816, + 817, + 818, + 819, + 820, + 821, + 822, + 823, + 824, + 825, + 826, + 827, + 828, + 829, + 830, + 831, + 832, + 833, + 834, + 835, + 836, + 837, + 838, + 839, + 840, + 841, + 842, + 843, + 844, + 845, + 846, + 847, + 848, + 849, + 850, + 851, + 852, + 853, + 854, + 855, + 856, + 857, + 858, + 859 + ] + } +] \ No newline at end of file diff --git a/004000/training_state/optimizer_state.safetensors b/004000/training_state/optimizer_state.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..7ef393005b7e47c69567229dab82c8938ab44565 --- /dev/null +++ b/004000/training_state/optimizer_state.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:75c40863386b9c451a0c98a588d43e411f398be8c5bee0f8537c06727a3fb291 +size 22197120 diff --git a/004000/training_state/rng_state.safetensors b/004000/training_state/rng_state.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..a148bf4bcd4771b20b125b71fb650e1b44ba7f97 --- /dev/null +++ b/004000/training_state/rng_state.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0d22840af56c203a608e5d2a3ac7646d499aaad3d139e79b681308863299b1cc +size 15708 diff --git a/004000/training_state/scheduler_state.json b/004000/training_state/scheduler_state.json new file mode 100644 index 0000000000000000000000000000000000000000..37fd0187d16b6be1b290829a65af715b7fd19ea2 --- /dev/null +++ b/004000/training_state/scheduler_state.json @@ -0,0 +1,15 @@ +{ + "base_lrs": [ + 0.001 + ], + "last_epoch": 4000, + "_step_count": 4001, + "_is_initial": false, + "_get_lr_called_within_step": false, + "_last_lr": [ + 0.0006890576474687264 + ], + "lr_lambdas": [ + null + ] +} \ No newline at end of file diff --git a/004000/training_state/training_step.json b/004000/training_state/training_step.json new file mode 100644 index 0000000000000000000000000000000000000000..75d870521c191f77dd9eaa4d83486eab6e768f69 --- /dev/null +++ b/004000/training_state/training_step.json @@ -0,0 +1,3 @@ +{ + "step": 4000 +} \ No newline at end of file diff --git a/005000/pretrained_model/README.md b/005000/pretrained_model/README.md new file mode 100644 index 0000000000000000000000000000000000000000..6565338d7b6577ddbee192786d397d22079b8338 --- /dev/null +++ b/005000/pretrained_model/README.md @@ -0,0 +1,204 @@ +--- +library_name: peft +tags: +- base_model:adapter:lerobot/pi0_base +- lora +--- + +# Model Card for Model ID + + + + + +## Model Details + +### Model Description + + + + + +- **Developed by:** [More Information Needed] +- **Funded by [optional]:** [More Information Needed] +- **Shared by [optional]:** [More Information Needed] +- **Model type:** [More Information Needed] +- **Language(s) (NLP):** [More Information Needed] +- **License:** [More Information Needed] +- **Finetuned from model [optional]:** [More Information Needed] + +### Model Sources [optional] + + + +- **Repository:** [More Information Needed] +- **Paper [optional]:** [More Information Needed] +- **Demo [optional]:** [More Information Needed] + +## Uses + + + +### Direct Use + + + +[More Information Needed] + +### Downstream Use [optional] + + + +[More Information Needed] + +### Out-of-Scope Use + + + +[More Information Needed] + +## Bias, Risks, and Limitations + + + +[More Information Needed] + +### Recommendations + + + +Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations. + +## How to Get Started with the Model + +Use the code below to get started with the model. + +[More Information Needed] + +## Training Details + +### Training Data + + + +[More Information Needed] + +### Training Procedure + + + +#### Preprocessing [optional] + +[More Information Needed] + + +#### Training Hyperparameters + +- **Training regime:** [More Information Needed] + +#### Speeds, Sizes, Times [optional] + + + +[More Information Needed] + +## Evaluation + + + +### Testing Data, Factors & Metrics + +#### Testing Data + + + +[More Information Needed] + +#### Factors + + + +[More Information Needed] + +#### Metrics + + + +[More Information Needed] + +### Results + +[More Information Needed] + +#### Summary + + + +## Model Examination [optional] + + + +[More Information Needed] + +## Environmental Impact + + + +Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700). + +- **Hardware Type:** [More Information Needed] +- **Hours used:** [More Information Needed] +- **Cloud Provider:** [More Information Needed] +- **Compute Region:** [More Information Needed] +- **Carbon Emitted:** [More Information Needed] + +## Technical Specifications [optional] + +### Model Architecture and Objective + +[More Information Needed] + +### Compute Infrastructure + +[More Information Needed] + +#### Hardware + +[More Information Needed] + +#### Software + +[More Information Needed] + +## Citation [optional] + + + +**BibTeX:** + +[More Information Needed] + +**APA:** + +[More Information Needed] + +## Glossary [optional] + + + +[More Information Needed] + +## More Information [optional] + +[More Information Needed] + +## Model Card Authors [optional] + +[More Information Needed] + +## Model Card Contact + +[More Information Needed] +### Framework versions + +- PEFT 0.18.1 \ No newline at end of file diff --git a/005000/pretrained_model/adapter_config.json b/005000/pretrained_model/adapter_config.json new file mode 100644 index 0000000000000000000000000000000000000000..3c00acefa2527617e9a485e421d89b9166fed77b --- /dev/null +++ b/005000/pretrained_model/adapter_config.json @@ -0,0 +1,41 @@ +{ + "alora_invocation_tokens": null, + "alpha_pattern": {}, + "arrow_config": null, + "auto_mapping": { + "base_model_class": "PI0Policy", + "parent_library": "lerobot.policies.pi0.modeling_pi0" + }, + "base_model_name_or_path": "lerobot/pi0_base", + "bias": "none", + "corda_config": null, + "ensure_weight_tying": false, + "eva_config": null, + "exclude_modules": null, + "fan_in_fan_out": false, + "inference_mode": true, + "init_lora_weights": true, + "layer_replication": null, + "layers_pattern": null, + "layers_to_transform": null, + "loftq_config": {}, + "lora_alpha": 8, + "lora_bias": false, + "lora_dropout": 0.0, + "megatron_config": null, + "megatron_core": "megatron.core", + "modules_to_save": [], + "peft_type": "LORA", + "peft_version": "0.18.1", + "qalora_group_size": 16, + "r": 32, + "rank_pattern": {}, + "revision": null, + "target_modules": "(.*\\.gemma_expert\\..*\\.self_attn\\.(q|v)_proj|model\\.(state_proj|action_in_proj|action_out_proj|action_time_mlp_in|action_time_mlp_out))", + "target_parameters": null, + "task_type": null, + "trainable_token_indices": null, + "use_dora": false, + "use_qalora": false, + "use_rslora": false +} \ No newline at end of file diff --git a/005000/pretrained_model/adapter_model.safetensors b/005000/pretrained_model/adapter_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..787d4412cf9863a8d8b410d770bb8cefc96a2062 --- /dev/null +++ b/005000/pretrained_model/adapter_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8fb0529a81f5782255a15f154b2b875ca6acc9ec350b93c72d6a64506fdae040 +size 11101592 diff --git a/005000/pretrained_model/config.json b/005000/pretrained_model/config.json new file mode 100644 index 0000000000000000000000000000000000000000..5bded5dca90b3c498f402c92f09b4af980008111 --- /dev/null +++ b/005000/pretrained_model/config.json @@ -0,0 +1,113 @@ +{ + "type": "pi0", + "n_obs_steps": 1, + "input_features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "device": "cuda", + "use_amp": false, + "use_peft": true, + "push_to_hub": true, + "repo_id": "F-Fer/pi0-xlerobot-0", + "private": null, + "tags": null, + "license": null, + "pretrained_path": "lerobot/pi0_base", + "paligemma_variant": "gemma_2b", + "action_expert_variant": "gemma_300m", + "dtype": "bfloat16", + "chunk_size": 50, + "n_action_steps": 50, + "max_state_dim": 32, + "max_action_dim": 32, + "num_inference_steps": 10, + "time_sampling_beta_alpha": 1.5, + "time_sampling_beta_beta": 1.0, + "time_sampling_scale": 0.999, + "time_sampling_offset": 0.001, + "min_period": 0.004, + "max_period": 4.0, + "use_relative_actions": false, + "relative_exclude_joints": [ + "gripper" + ], + "action_feature_names": [ + "left_shoulder_pan.pos", + "left_shoulder_lift.pos", + "left_elbow_flex.pos", + "left_wrist_flex.pos", + "left_wrist_roll.pos", + "left_gripper.pos", + "right_shoulder_pan.pos", + "right_shoulder_lift.pos", + "right_elbow_flex.pos", + "right_wrist_flex.pos", + "right_wrist_roll.pos", + "right_gripper.pos" + ], + "rtc_config": null, + "image_resolution": [ + 224, + 224 + ], + "empty_cameras": 0, + "normalization_mapping": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + }, + "gradient_checkpointing": true, + "compile_model": true, + "compile_mode": "max-autotune", + "freeze_vision_encoder": false, + "train_expert_only": false, + "optimizer_lr": 0.001, + "optimizer_betas": [ + 0.9, + 0.95 + ], + "optimizer_eps": 1e-08, + "optimizer_weight_decay": 0.01, + "optimizer_grad_clip_norm": 1.0, + "scheduler_warmup_steps": 1000, + "scheduler_decay_steps": 30000, + "scheduler_decay_lr": 0.0001, + "tokenizer_max_length": 48 +} \ No newline at end of file diff --git a/005000/pretrained_model/policy_postprocessor.json b/005000/pretrained_model/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..25a2e623f8277a52e915c3701c7cd9cf77ddf8d6 --- /dev/null +++ b/005000/pretrained_model/policy_postprocessor.json @@ -0,0 +1,32 @@ +{ + "name": "policy_postprocessor", + "steps": [ + { + "registry_name": "unnormalizer_processor", + "config": { + "eps": 1e-08, + "features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "norm_map": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + } + }, + "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors" + }, + { + "registry_name": "device_processor", + "config": { + "device": "cpu", + "float_dtype": null + } + } + ] +} \ No newline at end of file diff --git a/005000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/005000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..08107a06a9226905f59e970a3d28ddc923a42695 --- /dev/null +++ b/005000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:794c084955a7613103705a47253d9fa3f66d03ead8978fa493cad135498ca62b +size 9192 diff --git a/005000/pretrained_model/policy_preprocessor.json b/005000/pretrained_model/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..8c39dc4710ff54b6f6d1631e34f0484383336eca --- /dev/null +++ b/005000/pretrained_model/policy_preprocessor.json @@ -0,0 +1,91 @@ +{ + "name": "policy_preprocessor", + "steps": [ + { + "registry_name": "rename_observations_processor", + "config": { + "rename_map": { + "observation.images.left_front": "observation.images.base_0_rgb", + "observation.images.left_left": "observation.images.left_wrist_0_rgb", + "observation.images.left_right": "observation.images.right_wrist_0_rgb" + } + } + }, + { + "registry_name": "to_batch_processor", + "config": {} + }, + { + "registry_name": "pi0_new_line_processor", + "config": {} + }, + { + "registry_name": "tokenizer_processor", + "config": { + "max_length": 48, + "task_key": "task", + "padding_side": "right", + "padding": "max_length", + "truncation": true, + "tokenizer_name": "google/paligemma-3b-pt-224" + } + }, + { + "registry_name": "device_processor", + "config": { + "device": "cuda", + "float_dtype": null + } + }, + { + "registry_name": "normalizer_processor", + "config": { + "eps": 1e-08, + "features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + }, + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "norm_map": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + } + }, + "state_file": "policy_preprocessor_step_5_normalizer_processor.safetensors" + } + ] +} \ No newline at end of file diff --git a/005000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors b/005000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..08107a06a9226905f59e970a3d28ddc923a42695 --- /dev/null +++ b/005000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:794c084955a7613103705a47253d9fa3f66d03ead8978fa493cad135498ca62b +size 9192 diff --git a/005000/pretrained_model/train_config.json b/005000/pretrained_model/train_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8b15c723555caefa3bdb9e5e9097937ebfef3fad --- /dev/null +++ b/005000/pretrained_model/train_config.json @@ -0,0 +1,261 @@ +{ + "dataset": { + "repo_id": "F-Fer/xlerobot-0", + "root": null, + "episodes": null, + "image_transforms": { + "enable": false, + "max_num_transforms": 3, + "random_order": false, + "tfs": { + "brightness": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "brightness": [ + 0.8, + 1.2 + ] + } + }, + "contrast": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "contrast": [ + 0.8, + 1.2 + ] + } + }, + "saturation": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "saturation": [ + 0.5, + 1.5 + ] + } + }, + "hue": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "hue": [ + -0.05, + 0.05 + ] + } + }, + "sharpness": { + "weight": 1.0, + "type": "SharpnessJitter", + "kwargs": { + "sharpness": [ + 0.5, + 1.5 + ] + } + }, + "affine": { + "weight": 1.0, + "type": "RandomAffine", + "kwargs": { + "degrees": [ + -5.0, + 5.0 + ], + "translate": [ + 0.05, + 0.05 + ] + } + } + } + }, + "revision": null, + "use_imagenet_stats": true, + "video_backend": "torchcodec", + "streaming": false + }, + "env": null, + "policy": { + "type": "pi0", + "n_obs_steps": 1, + "input_features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "device": "cuda", + "use_amp": false, + "use_peft": true, + "push_to_hub": true, + "repo_id": "F-Fer/pi0-xlerobot-0", + "private": null, + "tags": null, + "license": null, + "pretrained_path": "lerobot/pi0_base", + "paligemma_variant": "gemma_2b", + "action_expert_variant": "gemma_300m", + "dtype": "bfloat16", + "chunk_size": 50, + "n_action_steps": 50, + "max_state_dim": 32, + "max_action_dim": 32, + "num_inference_steps": 10, + "time_sampling_beta_alpha": 1.5, + "time_sampling_beta_beta": 1.0, + "time_sampling_scale": 0.999, + "time_sampling_offset": 0.001, + "min_period": 0.004, + "max_period": 4.0, + "use_relative_actions": false, + "relative_exclude_joints": [ + "gripper" + ], + "action_feature_names": [ + "left_shoulder_pan.pos", + "left_shoulder_lift.pos", + "left_elbow_flex.pos", + "left_wrist_flex.pos", + "left_wrist_roll.pos", + "left_gripper.pos", + "right_shoulder_pan.pos", + "right_shoulder_lift.pos", + "right_elbow_flex.pos", + "right_wrist_flex.pos", + "right_wrist_roll.pos", + "right_gripper.pos" + ], + "rtc_config": null, + "image_resolution": [ + 224, + 224 + ], + "empty_cameras": 0, + "normalization_mapping": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + }, + "gradient_checkpointing": true, + "compile_model": true, + "compile_mode": "max-autotune", + "freeze_vision_encoder": false, + "train_expert_only": false, + "optimizer_lr": 0.001, + "optimizer_betas": [ + 0.9, + 0.95 + ], + "optimizer_eps": 1e-08, + "optimizer_weight_decay": 0.01, + "optimizer_grad_clip_norm": 1.0, + "scheduler_warmup_steps": 1000, + "scheduler_decay_steps": 30000, + "scheduler_decay_lr": 0.0001, + "tokenizer_max_length": 48 + }, + "output_dir": "outputs/train/2026-04-10/14-19-24_pi0", + "job_name": "pi0", + "resume": false, + "seed": 1000, + "cudnn_deterministic": false, + "num_workers": 4, + "batch_size": 4, + "steps": 10000, + "eval_freq": -1, + "log_freq": 100, + "tolerance_s": 0.0001, + "save_checkpoint": true, + "save_freq": 1000, + "use_policy_training_preset": true, + "optimizer": { + "type": "adamw", + "lr": 0.001, + "weight_decay": 0.01, + "grad_clip_norm": 1.0, + "betas": [ + 0.9, + 0.95 + ], + "eps": 1e-08 + }, + "scheduler": { + "type": "cosine_decay_with_warmup", + "num_warmup_steps": 1000, + "num_decay_steps": 30000, + "peak_lr": 0.001, + "decay_lr": 0.0001 + }, + "eval": { + "n_episodes": 50, + "batch_size": 50, + "use_async_envs": false + }, + "wandb": { + "enable": true, + "disable_artifact": false, + "project": "xlerobot", + "entity": null, + "notes": null, + "run_id": "tmqkc0d7", + "mode": null, + "add_tags": true + }, + "peft": { + "target_modules": null, + "full_training_modules": null, + "method_type": "LORA", + "init_type": null, + "r": 32 + }, + "use_rabc": false, + "rabc_progress_path": null, + "rabc_kappa": 0.01, + "rabc_epsilon": 1e-06, + "rabc_head_mode": "sparse", + "rename_map": { + "observation.images.left_front": "observation.images.base_0_rgb", + "observation.images.left_left": "observation.images.left_wrist_0_rgb", + "observation.images.left_right": "observation.images.right_wrist_0_rgb" + }, + "checkpoint_path": null +} \ No newline at end of file diff --git a/005000/training_state/optimizer_param_groups.json b/005000/training_state/optimizer_param_groups.json new file mode 100644 index 0000000000000000000000000000000000000000..5e3da3aba789b67beeed87e9a83748b401d9f51d --- /dev/null +++ b/005000/training_state/optimizer_param_groups.json @@ -0,0 +1,881 @@ +[ + { + "lr": 0.00055, + "betas": [ + 0.9, + 0.95 + ], + "eps": 1e-08, + "weight_decay": 0.01, + "amsgrad": false, + "maximize": false, + "foreach": null, + "capturable": false, + "differentiable": false, + "fused": null, + "decoupled_weight_decay": true, + "initial_lr": 0.001, + "params": [ + 0, + 1, + 2, + 3, + 4, + 5, + 6, + 7, + 8, + 9, + 10, + 11, + 12, + 13, + 14, + 15, + 16, + 17, + 18, + 19, + 20, + 21, + 22, + 23, + 24, + 25, + 26, + 27, + 28, + 29, + 30, + 31, + 32, + 33, + 34, + 35, + 36, + 37, + 38, + 39, + 40, + 41, + 42, + 43, + 44, + 45, + 46, + 47, + 48, + 49, + 50, + 51, + 52, + 53, + 54, + 55, + 56, + 57, + 58, + 59, + 60, + 61, + 62, + 63, + 64, + 65, + 66, + 67, + 68, + 69, + 70, + 71, + 72, + 73, + 74, + 75, + 76, + 77, + 78, + 79, + 80, + 81, + 82, + 83, + 84, + 85, + 86, + 87, + 88, + 89, + 90, + 91, + 92, + 93, + 94, + 95, + 96, + 97, + 98, + 99, + 100, + 101, + 102, + 103, + 104, + 105, + 106, + 107, + 108, + 109, + 110, + 111, + 112, + 113, + 114, + 115, + 116, + 117, + 118, + 119, + 120, + 121, + 122, + 123, + 124, + 125, + 126, + 127, + 128, + 129, + 130, + 131, + 132, + 133, + 134, + 135, + 136, + 137, + 138, + 139, + 140, + 141, + 142, + 143, + 144, + 145, + 146, + 147, + 148, + 149, + 150, + 151, + 152, + 153, + 154, + 155, + 156, + 157, + 158, + 159, + 160, + 161, + 162, + 163, + 164, + 165, + 166, + 167, + 168, + 169, + 170, + 171, + 172, + 173, + 174, + 175, + 176, + 177, + 178, + 179, + 180, + 181, + 182, + 183, + 184, + 185, + 186, + 187, + 188, + 189, + 190, + 191, + 192, + 193, + 194, + 195, + 196, + 197, + 198, + 199, + 200, + 201, + 202, + 203, + 204, + 205, + 206, + 207, + 208, + 209, + 210, + 211, + 212, + 213, + 214, + 215, + 216, + 217, + 218, + 219, + 220, + 221, + 222, + 223, + 224, + 225, + 226, + 227, + 228, + 229, + 230, + 231, + 232, + 233, + 234, + 235, + 236, + 237, + 238, + 239, + 240, + 241, + 242, + 243, + 244, + 245, + 246, + 247, + 248, + 249, + 250, + 251, + 252, + 253, + 254, + 255, + 256, + 257, + 258, + 259, + 260, + 261, + 262, + 263, + 264, + 265, + 266, + 267, + 268, + 269, + 270, + 271, + 272, + 273, + 274, + 275, + 276, + 277, + 278, + 279, + 280, + 281, + 282, + 283, + 284, + 285, + 286, + 287, + 288, + 289, + 290, + 291, + 292, + 293, + 294, + 295, + 296, + 297, + 298, + 299, + 300, + 301, + 302, + 303, + 304, + 305, + 306, + 307, + 308, + 309, + 310, + 311, + 312, + 313, + 314, + 315, + 316, + 317, + 318, + 319, + 320, + 321, + 322, + 323, + 324, + 325, + 326, + 327, + 328, + 329, + 330, + 331, + 332, + 333, + 334, + 335, + 336, + 337, + 338, + 339, + 340, + 341, + 342, + 343, + 344, + 345, + 346, + 347, + 348, + 349, + 350, + 351, + 352, + 353, + 354, + 355, + 356, + 357, + 358, + 359, + 360, + 361, + 362, + 363, + 364, + 365, + 366, + 367, + 368, + 369, + 370, + 371, + 372, + 373, + 374, + 375, + 376, + 377, + 378, + 379, + 380, + 381, + 382, + 383, + 384, + 385, + 386, + 387, + 388, + 389, + 390, + 391, + 392, + 393, + 394, + 395, + 396, + 397, + 398, + 399, + 400, + 401, + 402, + 403, + 404, + 405, + 406, + 407, + 408, + 409, + 410, + 411, + 412, + 413, + 414, + 415, + 416, + 417, + 418, + 419, + 420, + 421, + 422, + 423, + 424, + 425, + 426, + 427, + 428, + 429, + 430, + 431, + 432, + 433, + 434, + 435, + 436, + 437, + 438, + 439, + 440, + 441, + 442, + 443, + 444, + 445, + 446, + 447, + 448, + 449, + 450, + 451, + 452, + 453, + 454, + 455, + 456, + 457, + 458, + 459, + 460, + 461, + 462, + 463, + 464, + 465, + 466, + 467, + 468, + 469, + 470, + 471, + 472, + 473, + 474, + 475, + 476, + 477, + 478, + 479, + 480, + 481, + 482, + 483, + 484, + 485, + 486, + 487, + 488, + 489, + 490, + 491, + 492, + 493, + 494, + 495, + 496, + 497, + 498, + 499, + 500, + 501, + 502, + 503, + 504, + 505, + 506, + 507, + 508, + 509, + 510, + 511, + 512, + 513, + 514, + 515, + 516, + 517, + 518, + 519, + 520, + 521, + 522, + 523, + 524, + 525, + 526, + 527, + 528, + 529, + 530, + 531, + 532, + 533, + 534, + 535, + 536, + 537, + 538, + 539, + 540, + 541, + 542, + 543, + 544, + 545, + 546, + 547, + 548, + 549, + 550, + 551, + 552, + 553, + 554, + 555, + 556, + 557, + 558, + 559, + 560, + 561, + 562, + 563, + 564, + 565, + 566, + 567, + 568, + 569, + 570, + 571, + 572, + 573, + 574, + 575, + 576, + 577, + 578, + 579, + 580, + 581, + 582, + 583, + 584, + 585, + 586, + 587, + 588, + 589, + 590, + 591, + 592, + 593, + 594, + 595, + 596, + 597, + 598, + 599, + 600, + 601, + 602, + 603, + 604, + 605, + 606, + 607, + 608, + 609, + 610, + 611, + 612, + 613, + 614, + 615, + 616, + 617, + 618, + 619, + 620, + 621, + 622, + 623, + 624, + 625, + 626, + 627, + 628, + 629, + 630, + 631, + 632, + 633, + 634, + 635, + 636, + 637, + 638, + 639, + 640, + 641, + 642, + 643, + 644, + 645, + 646, + 647, + 648, + 649, + 650, + 651, + 652, + 653, + 654, + 655, + 656, + 657, + 658, + 659, + 660, + 661, + 662, + 663, + 664, + 665, + 666, + 667, + 668, + 669, + 670, + 671, + 672, + 673, + 674, + 675, + 676, + 677, + 678, + 679, + 680, + 681, + 682, + 683, + 684, + 685, + 686, + 687, + 688, + 689, + 690, + 691, + 692, + 693, + 694, + 695, + 696, + 697, + 698, + 699, + 700, + 701, + 702, + 703, + 704, + 705, + 706, + 707, + 708, + 709, + 710, + 711, + 712, + 713, + 714, + 715, + 716, + 717, + 718, + 719, + 720, + 721, + 722, + 723, + 724, + 725, + 726, + 727, + 728, + 729, + 730, + 731, + 732, + 733, + 734, + 735, + 736, + 737, + 738, + 739, + 740, + 741, + 742, + 743, + 744, + 745, + 746, + 747, + 748, + 749, + 750, + 751, + 752, + 753, + 754, + 755, + 756, + 757, + 758, + 759, + 760, + 761, + 762, + 763, + 764, + 765, + 766, + 767, + 768, + 769, + 770, + 771, + 772, + 773, + 774, + 775, + 776, + 777, + 778, + 779, + 780, + 781, + 782, + 783, + 784, + 785, + 786, + 787, + 788, + 789, + 790, + 791, + 792, + 793, + 794, + 795, + 796, + 797, + 798, + 799, + 800, + 801, + 802, + 803, + 804, + 805, + 806, + 807, + 808, + 809, + 810, + 811, + 812, + 813, + 814, + 815, + 816, + 817, + 818, + 819, + 820, + 821, + 822, + 823, + 824, + 825, + 826, + 827, + 828, + 829, + 830, + 831, + 832, + 833, + 834, + 835, + 836, + 837, + 838, + 839, + 840, + 841, + 842, + 843, + 844, + 845, + 846, + 847, + 848, + 849, + 850, + 851, + 852, + 853, + 854, + 855, + 856, + 857, + 858, + 859 + ] + } +] \ No newline at end of file diff --git a/005000/training_state/optimizer_state.safetensors b/005000/training_state/optimizer_state.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..2ed2c85257edd4b3ff96a99d57961a1ac68324e5 --- /dev/null +++ b/005000/training_state/optimizer_state.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8db9870b1d5449064ade66cebc85b03e6d805f9688571bf380cc162cf282b6c0 +size 22197120 diff --git a/005000/training_state/rng_state.safetensors b/005000/training_state/rng_state.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..28f6494f03072068479a02c9cf2f6d832363bff6 --- /dev/null +++ b/005000/training_state/rng_state.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5c9e164f5f020507b5653fa627e97f49aaf46cdc153dde67a2de5cf965dd11e9 +size 15708 diff --git a/005000/training_state/scheduler_state.json b/005000/training_state/scheduler_state.json new file mode 100644 index 0000000000000000000000000000000000000000..81fa6778ba604ba799ad75a44716404e5081b460 --- /dev/null +++ b/005000/training_state/scheduler_state.json @@ -0,0 +1,15 @@ +{ + "base_lrs": [ + 0.001 + ], + "last_epoch": 5000, + "_step_count": 5001, + "_is_initial": false, + "_get_lr_called_within_step": false, + "_last_lr": [ + 0.00055 + ], + "lr_lambdas": [ + null + ] +} \ No newline at end of file diff --git a/005000/training_state/training_step.json b/005000/training_state/training_step.json new file mode 100644 index 0000000000000000000000000000000000000000..592449d3fc8b35c30c604eb1dabe60537e8224a0 --- /dev/null +++ b/005000/training_state/training_step.json @@ -0,0 +1,3 @@ +{ + "step": 5000 +} \ No newline at end of file diff --git a/006000/pretrained_model/README.md b/006000/pretrained_model/README.md new file mode 100644 index 0000000000000000000000000000000000000000..6565338d7b6577ddbee192786d397d22079b8338 --- /dev/null +++ b/006000/pretrained_model/README.md @@ -0,0 +1,204 @@ +--- +library_name: peft +tags: +- base_model:adapter:lerobot/pi0_base +- lora +--- + +# Model Card for Model ID + + + + + +## Model Details + +### Model Description + + + + + +- **Developed by:** [More Information Needed] +- **Funded by [optional]:** [More Information Needed] +- **Shared by [optional]:** [More Information Needed] +- **Model type:** [More Information Needed] +- **Language(s) (NLP):** [More Information Needed] +- **License:** [More Information Needed] +- **Finetuned from model [optional]:** [More Information Needed] + +### Model Sources [optional] + + + +- **Repository:** [More Information Needed] +- **Paper [optional]:** [More Information Needed] +- **Demo [optional]:** [More Information Needed] + +## Uses + + + +### Direct Use + + + +[More Information Needed] + +### Downstream Use [optional] + + + +[More Information Needed] + +### Out-of-Scope Use + + + +[More Information Needed] + +## Bias, Risks, and Limitations + + + +[More Information Needed] + +### Recommendations + + + +Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations. + +## How to Get Started with the Model + +Use the code below to get started with the model. + +[More Information Needed] + +## Training Details + +### Training Data + + + +[More Information Needed] + +### Training Procedure + + + +#### Preprocessing [optional] + +[More Information Needed] + + +#### Training Hyperparameters + +- **Training regime:** [More Information Needed] + +#### Speeds, Sizes, Times [optional] + + + +[More Information Needed] + +## Evaluation + + + +### Testing Data, Factors & Metrics + +#### Testing Data + + + +[More Information Needed] + +#### Factors + + + +[More Information Needed] + +#### Metrics + + + +[More Information Needed] + +### Results + +[More Information Needed] + +#### Summary + + + +## Model Examination [optional] + + + +[More Information Needed] + +## Environmental Impact + + + +Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700). + +- **Hardware Type:** [More Information Needed] +- **Hours used:** [More Information Needed] +- **Cloud Provider:** [More Information Needed] +- **Compute Region:** [More Information Needed] +- **Carbon Emitted:** [More Information Needed] + +## Technical Specifications [optional] + +### Model Architecture and Objective + +[More Information Needed] + +### Compute Infrastructure + +[More Information Needed] + +#### Hardware + +[More Information Needed] + +#### Software + +[More Information Needed] + +## Citation [optional] + + + +**BibTeX:** + +[More Information Needed] + +**APA:** + +[More Information Needed] + +## Glossary [optional] + + + +[More Information Needed] + +## More Information [optional] + +[More Information Needed] + +## Model Card Authors [optional] + +[More Information Needed] + +## Model Card Contact + +[More Information Needed] +### Framework versions + +- PEFT 0.18.1 \ No newline at end of file diff --git a/006000/pretrained_model/adapter_config.json b/006000/pretrained_model/adapter_config.json new file mode 100644 index 0000000000000000000000000000000000000000..3c00acefa2527617e9a485e421d89b9166fed77b --- /dev/null +++ b/006000/pretrained_model/adapter_config.json @@ -0,0 +1,41 @@ +{ + "alora_invocation_tokens": null, + "alpha_pattern": {}, + "arrow_config": null, + "auto_mapping": { + "base_model_class": "PI0Policy", + "parent_library": "lerobot.policies.pi0.modeling_pi0" + }, + "base_model_name_or_path": "lerobot/pi0_base", + "bias": "none", + "corda_config": null, + "ensure_weight_tying": false, + "eva_config": null, + "exclude_modules": null, + "fan_in_fan_out": false, + "inference_mode": true, + "init_lora_weights": true, + "layer_replication": null, + "layers_pattern": null, + "layers_to_transform": null, + "loftq_config": {}, + "lora_alpha": 8, + "lora_bias": false, + "lora_dropout": 0.0, + "megatron_config": null, + "megatron_core": "megatron.core", + "modules_to_save": [], + "peft_type": "LORA", + "peft_version": "0.18.1", + "qalora_group_size": 16, + "r": 32, + "rank_pattern": {}, + "revision": null, + "target_modules": "(.*\\.gemma_expert\\..*\\.self_attn\\.(q|v)_proj|model\\.(state_proj|action_in_proj|action_out_proj|action_time_mlp_in|action_time_mlp_out))", + "target_parameters": null, + "task_type": null, + "trainable_token_indices": null, + "use_dora": false, + "use_qalora": false, + "use_rslora": false +} \ No newline at end of file diff --git a/006000/pretrained_model/adapter_model.safetensors b/006000/pretrained_model/adapter_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..316d1fe103e5e9c3e33b8feb954ee52bb189bdea --- /dev/null +++ b/006000/pretrained_model/adapter_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c4b84622f03d23e3d9d5acbfa425a3dfcd374363185e73260d9d5c4a85d24be6 +size 11101592 diff --git a/006000/pretrained_model/config.json b/006000/pretrained_model/config.json new file mode 100644 index 0000000000000000000000000000000000000000..5bded5dca90b3c498f402c92f09b4af980008111 --- /dev/null +++ b/006000/pretrained_model/config.json @@ -0,0 +1,113 @@ +{ + "type": "pi0", + "n_obs_steps": 1, + "input_features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "device": "cuda", + "use_amp": false, + "use_peft": true, + "push_to_hub": true, + "repo_id": "F-Fer/pi0-xlerobot-0", + "private": null, + "tags": null, + "license": null, + "pretrained_path": "lerobot/pi0_base", + "paligemma_variant": "gemma_2b", + "action_expert_variant": "gemma_300m", + "dtype": "bfloat16", + "chunk_size": 50, + "n_action_steps": 50, + "max_state_dim": 32, + "max_action_dim": 32, + "num_inference_steps": 10, + "time_sampling_beta_alpha": 1.5, + "time_sampling_beta_beta": 1.0, + "time_sampling_scale": 0.999, + "time_sampling_offset": 0.001, + "min_period": 0.004, + "max_period": 4.0, + "use_relative_actions": false, + "relative_exclude_joints": [ + "gripper" + ], + "action_feature_names": [ + "left_shoulder_pan.pos", + "left_shoulder_lift.pos", + "left_elbow_flex.pos", + "left_wrist_flex.pos", + "left_wrist_roll.pos", + "left_gripper.pos", + "right_shoulder_pan.pos", + "right_shoulder_lift.pos", + "right_elbow_flex.pos", + "right_wrist_flex.pos", + "right_wrist_roll.pos", + "right_gripper.pos" + ], + "rtc_config": null, + "image_resolution": [ + 224, + 224 + ], + "empty_cameras": 0, + "normalization_mapping": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + }, + "gradient_checkpointing": true, + "compile_model": true, + "compile_mode": "max-autotune", + "freeze_vision_encoder": false, + "train_expert_only": false, + "optimizer_lr": 0.001, + "optimizer_betas": [ + 0.9, + 0.95 + ], + "optimizer_eps": 1e-08, + "optimizer_weight_decay": 0.01, + "optimizer_grad_clip_norm": 1.0, + "scheduler_warmup_steps": 1000, + "scheduler_decay_steps": 30000, + "scheduler_decay_lr": 0.0001, + "tokenizer_max_length": 48 +} \ No newline at end of file diff --git a/006000/pretrained_model/policy_postprocessor.json b/006000/pretrained_model/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..25a2e623f8277a52e915c3701c7cd9cf77ddf8d6 --- /dev/null +++ b/006000/pretrained_model/policy_postprocessor.json @@ -0,0 +1,32 @@ +{ + "name": "policy_postprocessor", + "steps": [ + { + "registry_name": "unnormalizer_processor", + "config": { + "eps": 1e-08, + "features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "norm_map": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + } + }, + "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors" + }, + { + "registry_name": "device_processor", + "config": { + "device": "cpu", + "float_dtype": null + } + } + ] +} \ No newline at end of file diff --git a/006000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/006000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..08107a06a9226905f59e970a3d28ddc923a42695 --- /dev/null +++ b/006000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:794c084955a7613103705a47253d9fa3f66d03ead8978fa493cad135498ca62b +size 9192 diff --git a/006000/pretrained_model/policy_preprocessor.json b/006000/pretrained_model/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..8c39dc4710ff54b6f6d1631e34f0484383336eca --- /dev/null +++ b/006000/pretrained_model/policy_preprocessor.json @@ -0,0 +1,91 @@ +{ + "name": "policy_preprocessor", + "steps": [ + { + "registry_name": "rename_observations_processor", + "config": { + "rename_map": { + "observation.images.left_front": "observation.images.base_0_rgb", + "observation.images.left_left": "observation.images.left_wrist_0_rgb", + "observation.images.left_right": "observation.images.right_wrist_0_rgb" + } + } + }, + { + "registry_name": "to_batch_processor", + "config": {} + }, + { + "registry_name": "pi0_new_line_processor", + "config": {} + }, + { + "registry_name": "tokenizer_processor", + "config": { + "max_length": 48, + "task_key": "task", + "padding_side": "right", + "padding": "max_length", + "truncation": true, + "tokenizer_name": "google/paligemma-3b-pt-224" + } + }, + { + "registry_name": "device_processor", + "config": { + "device": "cuda", + "float_dtype": null + } + }, + { + "registry_name": "normalizer_processor", + "config": { + "eps": 1e-08, + "features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + }, + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "norm_map": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + } + }, + "state_file": "policy_preprocessor_step_5_normalizer_processor.safetensors" + } + ] +} \ No newline at end of file diff --git a/006000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors b/006000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..08107a06a9226905f59e970a3d28ddc923a42695 --- /dev/null +++ b/006000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:794c084955a7613103705a47253d9fa3f66d03ead8978fa493cad135498ca62b +size 9192 diff --git a/006000/pretrained_model/train_config.json b/006000/pretrained_model/train_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8b15c723555caefa3bdb9e5e9097937ebfef3fad --- /dev/null +++ b/006000/pretrained_model/train_config.json @@ -0,0 +1,261 @@ +{ + "dataset": { + "repo_id": "F-Fer/xlerobot-0", + "root": null, + "episodes": null, + "image_transforms": { + "enable": false, + "max_num_transforms": 3, + "random_order": false, + "tfs": { + "brightness": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "brightness": [ + 0.8, + 1.2 + ] + } + }, + "contrast": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "contrast": [ + 0.8, + 1.2 + ] + } + }, + "saturation": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "saturation": [ + 0.5, + 1.5 + ] + } + }, + "hue": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "hue": [ + -0.05, + 0.05 + ] + } + }, + "sharpness": { + "weight": 1.0, + "type": "SharpnessJitter", + "kwargs": { + "sharpness": [ + 0.5, + 1.5 + ] + } + }, + "affine": { + "weight": 1.0, + "type": "RandomAffine", + "kwargs": { + "degrees": [ + -5.0, + 5.0 + ], + "translate": [ + 0.05, + 0.05 + ] + } + } + } + }, + "revision": null, + "use_imagenet_stats": true, + "video_backend": "torchcodec", + "streaming": false + }, + "env": null, + "policy": { + "type": "pi0", + "n_obs_steps": 1, + "input_features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "device": "cuda", + "use_amp": false, + "use_peft": true, + "push_to_hub": true, + "repo_id": "F-Fer/pi0-xlerobot-0", + "private": null, + "tags": null, + "license": null, + "pretrained_path": "lerobot/pi0_base", + "paligemma_variant": "gemma_2b", + "action_expert_variant": "gemma_300m", + "dtype": "bfloat16", + "chunk_size": 50, + "n_action_steps": 50, + "max_state_dim": 32, + "max_action_dim": 32, + "num_inference_steps": 10, + "time_sampling_beta_alpha": 1.5, + "time_sampling_beta_beta": 1.0, + "time_sampling_scale": 0.999, + "time_sampling_offset": 0.001, + "min_period": 0.004, + "max_period": 4.0, + "use_relative_actions": false, + "relative_exclude_joints": [ + "gripper" + ], + "action_feature_names": [ + "left_shoulder_pan.pos", + "left_shoulder_lift.pos", + "left_elbow_flex.pos", + "left_wrist_flex.pos", + "left_wrist_roll.pos", + "left_gripper.pos", + "right_shoulder_pan.pos", + "right_shoulder_lift.pos", + "right_elbow_flex.pos", + "right_wrist_flex.pos", + "right_wrist_roll.pos", + "right_gripper.pos" + ], + "rtc_config": null, + "image_resolution": [ + 224, + 224 + ], + "empty_cameras": 0, + "normalization_mapping": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + }, + "gradient_checkpointing": true, + "compile_model": true, + "compile_mode": "max-autotune", + "freeze_vision_encoder": false, + "train_expert_only": false, + "optimizer_lr": 0.001, + "optimizer_betas": [ + 0.9, + 0.95 + ], + "optimizer_eps": 1e-08, + "optimizer_weight_decay": 0.01, + "optimizer_grad_clip_norm": 1.0, + "scheduler_warmup_steps": 1000, + "scheduler_decay_steps": 30000, + "scheduler_decay_lr": 0.0001, + "tokenizer_max_length": 48 + }, + "output_dir": "outputs/train/2026-04-10/14-19-24_pi0", + "job_name": "pi0", + "resume": false, + "seed": 1000, + "cudnn_deterministic": false, + "num_workers": 4, + "batch_size": 4, + "steps": 10000, + "eval_freq": -1, + "log_freq": 100, + "tolerance_s": 0.0001, + "save_checkpoint": true, + "save_freq": 1000, + "use_policy_training_preset": true, + "optimizer": { + "type": "adamw", + "lr": 0.001, + "weight_decay": 0.01, + "grad_clip_norm": 1.0, + "betas": [ + 0.9, + 0.95 + ], + "eps": 1e-08 + }, + "scheduler": { + "type": "cosine_decay_with_warmup", + "num_warmup_steps": 1000, + "num_decay_steps": 30000, + "peak_lr": 0.001, + "decay_lr": 0.0001 + }, + "eval": { + "n_episodes": 50, + "batch_size": 50, + "use_async_envs": false + }, + "wandb": { + "enable": true, + "disable_artifact": false, + "project": "xlerobot", + "entity": null, + "notes": null, + "run_id": "tmqkc0d7", + "mode": null, + "add_tags": true + }, + "peft": { + "target_modules": null, + "full_training_modules": null, + "method_type": "LORA", + "init_type": null, + "r": 32 + }, + "use_rabc": false, + "rabc_progress_path": null, + "rabc_kappa": 0.01, + "rabc_epsilon": 1e-06, + "rabc_head_mode": "sparse", + "rename_map": { + "observation.images.left_front": "observation.images.base_0_rgb", + "observation.images.left_left": "observation.images.left_wrist_0_rgb", + "observation.images.left_right": "observation.images.right_wrist_0_rgb" + }, + "checkpoint_path": null +} \ No newline at end of file diff --git a/006000/training_state/optimizer_param_groups.json b/006000/training_state/optimizer_param_groups.json new file mode 100644 index 0000000000000000000000000000000000000000..793d8306fd62b0b3538efefda76c14ce62397038 --- /dev/null +++ b/006000/training_state/optimizer_param_groups.json @@ -0,0 +1,881 @@ +[ + { + "lr": 0.0004109423525312738, + "betas": [ + 0.9, + 0.95 + ], + "eps": 1e-08, + "weight_decay": 0.01, + "amsgrad": false, + "maximize": false, + "foreach": null, + "capturable": false, + "differentiable": false, + "fused": null, + "decoupled_weight_decay": true, + "initial_lr": 0.001, + "params": [ + 0, + 1, + 2, + 3, + 4, + 5, + 6, + 7, + 8, + 9, + 10, + 11, + 12, + 13, + 14, + 15, + 16, + 17, + 18, + 19, + 20, + 21, + 22, + 23, + 24, + 25, + 26, + 27, + 28, + 29, + 30, + 31, + 32, + 33, + 34, + 35, + 36, + 37, + 38, + 39, + 40, + 41, + 42, + 43, + 44, + 45, + 46, + 47, + 48, + 49, + 50, + 51, + 52, + 53, + 54, + 55, + 56, + 57, + 58, + 59, + 60, + 61, + 62, + 63, + 64, + 65, + 66, + 67, + 68, + 69, + 70, + 71, + 72, + 73, + 74, + 75, + 76, + 77, + 78, + 79, + 80, + 81, + 82, + 83, + 84, + 85, + 86, + 87, + 88, + 89, + 90, + 91, + 92, + 93, + 94, + 95, + 96, + 97, + 98, + 99, + 100, + 101, + 102, + 103, + 104, + 105, + 106, + 107, + 108, + 109, + 110, + 111, + 112, + 113, + 114, + 115, + 116, + 117, + 118, + 119, + 120, + 121, + 122, + 123, + 124, + 125, + 126, + 127, + 128, + 129, + 130, + 131, + 132, + 133, + 134, + 135, + 136, + 137, + 138, + 139, + 140, + 141, + 142, + 143, + 144, + 145, + 146, + 147, + 148, + 149, + 150, + 151, + 152, + 153, + 154, + 155, + 156, + 157, + 158, + 159, + 160, + 161, + 162, + 163, + 164, + 165, + 166, + 167, + 168, + 169, + 170, + 171, + 172, + 173, + 174, + 175, + 176, + 177, + 178, + 179, + 180, + 181, + 182, + 183, + 184, + 185, + 186, + 187, + 188, + 189, + 190, + 191, + 192, + 193, + 194, + 195, + 196, + 197, + 198, + 199, + 200, + 201, + 202, + 203, + 204, + 205, + 206, + 207, + 208, + 209, + 210, + 211, + 212, + 213, + 214, + 215, + 216, + 217, + 218, + 219, + 220, + 221, + 222, + 223, + 224, + 225, + 226, + 227, + 228, + 229, + 230, + 231, + 232, + 233, + 234, + 235, + 236, + 237, + 238, + 239, + 240, + 241, + 242, + 243, + 244, + 245, + 246, + 247, + 248, + 249, + 250, + 251, + 252, + 253, + 254, + 255, + 256, + 257, + 258, + 259, + 260, + 261, + 262, + 263, + 264, + 265, + 266, + 267, + 268, + 269, + 270, + 271, + 272, + 273, + 274, + 275, + 276, + 277, + 278, + 279, + 280, + 281, + 282, + 283, + 284, + 285, + 286, + 287, + 288, + 289, + 290, + 291, + 292, + 293, + 294, + 295, + 296, + 297, + 298, + 299, + 300, + 301, + 302, + 303, + 304, + 305, + 306, + 307, + 308, + 309, + 310, + 311, + 312, + 313, + 314, + 315, + 316, + 317, + 318, + 319, + 320, + 321, + 322, + 323, + 324, + 325, + 326, + 327, + 328, + 329, + 330, + 331, + 332, + 333, + 334, + 335, + 336, + 337, + 338, + 339, + 340, + 341, + 342, + 343, + 344, + 345, + 346, + 347, + 348, + 349, + 350, + 351, + 352, + 353, + 354, + 355, + 356, + 357, + 358, + 359, + 360, + 361, + 362, + 363, + 364, + 365, + 366, + 367, + 368, + 369, + 370, + 371, + 372, + 373, + 374, + 375, + 376, + 377, + 378, + 379, + 380, + 381, + 382, + 383, + 384, + 385, + 386, + 387, + 388, + 389, + 390, + 391, + 392, + 393, + 394, + 395, + 396, + 397, + 398, + 399, + 400, + 401, + 402, + 403, + 404, + 405, + 406, + 407, + 408, + 409, + 410, + 411, + 412, + 413, + 414, + 415, + 416, + 417, + 418, + 419, + 420, + 421, + 422, + 423, + 424, + 425, + 426, + 427, + 428, + 429, + 430, + 431, + 432, + 433, + 434, + 435, + 436, + 437, + 438, + 439, + 440, + 441, + 442, + 443, + 444, + 445, + 446, + 447, + 448, + 449, + 450, + 451, + 452, + 453, + 454, + 455, + 456, + 457, + 458, + 459, + 460, + 461, + 462, + 463, + 464, + 465, + 466, + 467, + 468, + 469, + 470, + 471, + 472, + 473, + 474, + 475, + 476, + 477, + 478, + 479, + 480, + 481, + 482, + 483, + 484, + 485, + 486, + 487, + 488, + 489, + 490, + 491, + 492, + 493, + 494, + 495, + 496, + 497, + 498, + 499, + 500, + 501, + 502, + 503, + 504, + 505, + 506, + 507, + 508, + 509, + 510, + 511, + 512, + 513, + 514, + 515, + 516, + 517, + 518, + 519, + 520, + 521, + 522, + 523, + 524, + 525, + 526, + 527, + 528, + 529, + 530, + 531, + 532, + 533, + 534, + 535, + 536, + 537, + 538, + 539, + 540, + 541, + 542, + 543, + 544, + 545, + 546, + 547, + 548, + 549, + 550, + 551, + 552, + 553, + 554, + 555, + 556, + 557, + 558, + 559, + 560, + 561, + 562, + 563, + 564, + 565, + 566, + 567, + 568, + 569, + 570, + 571, + 572, + 573, + 574, + 575, + 576, + 577, + 578, + 579, + 580, + 581, + 582, + 583, + 584, + 585, + 586, + 587, + 588, + 589, + 590, + 591, + 592, + 593, + 594, + 595, + 596, + 597, + 598, + 599, + 600, + 601, + 602, + 603, + 604, + 605, + 606, + 607, + 608, + 609, + 610, + 611, + 612, + 613, + 614, + 615, + 616, + 617, + 618, + 619, + 620, + 621, + 622, + 623, + 624, + 625, + 626, + 627, + 628, + 629, + 630, + 631, + 632, + 633, + 634, + 635, + 636, + 637, + 638, + 639, + 640, + 641, + 642, + 643, + 644, + 645, + 646, + 647, + 648, + 649, + 650, + 651, + 652, + 653, + 654, + 655, + 656, + 657, + 658, + 659, + 660, + 661, + 662, + 663, + 664, + 665, + 666, + 667, + 668, + 669, + 670, + 671, + 672, + 673, + 674, + 675, + 676, + 677, + 678, + 679, + 680, + 681, + 682, + 683, + 684, + 685, + 686, + 687, + 688, + 689, + 690, + 691, + 692, + 693, + 694, + 695, + 696, + 697, + 698, + 699, + 700, + 701, + 702, + 703, + 704, + 705, + 706, + 707, + 708, + 709, + 710, + 711, + 712, + 713, + 714, + 715, + 716, + 717, + 718, + 719, + 720, + 721, + 722, + 723, + 724, + 725, + 726, + 727, + 728, + 729, + 730, + 731, + 732, + 733, + 734, + 735, + 736, + 737, + 738, + 739, + 740, + 741, + 742, + 743, + 744, + 745, + 746, + 747, + 748, + 749, + 750, + 751, + 752, + 753, + 754, + 755, + 756, + 757, + 758, + 759, + 760, + 761, + 762, + 763, + 764, + 765, + 766, + 767, + 768, + 769, + 770, + 771, + 772, + 773, + 774, + 775, + 776, + 777, + 778, + 779, + 780, + 781, + 782, + 783, + 784, + 785, + 786, + 787, + 788, + 789, + 790, + 791, + 792, + 793, + 794, + 795, + 796, + 797, + 798, + 799, + 800, + 801, + 802, + 803, + 804, + 805, + 806, + 807, + 808, + 809, + 810, + 811, + 812, + 813, + 814, + 815, + 816, + 817, + 818, + 819, + 820, + 821, + 822, + 823, + 824, + 825, + 826, + 827, + 828, + 829, + 830, + 831, + 832, + 833, + 834, + 835, + 836, + 837, + 838, + 839, + 840, + 841, + 842, + 843, + 844, + 845, + 846, + 847, + 848, + 849, + 850, + 851, + 852, + 853, + 854, + 855, + 856, + 857, + 858, + 859 + ] + } +] \ No newline at end of file diff --git a/006000/training_state/optimizer_state.safetensors b/006000/training_state/optimizer_state.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..ea4980a279cfb1847578910af36e022da7fe2532 --- /dev/null +++ b/006000/training_state/optimizer_state.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:77044f7c39079bdbc7dfb9b2f7ebe36d6d1343dcdf9b36bea266102666606490 +size 22197120 diff --git a/006000/training_state/rng_state.safetensors b/006000/training_state/rng_state.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..7a5785012b63caea0001ff368cc3a60fcf8178a9 --- /dev/null +++ b/006000/training_state/rng_state.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:368f9ec2a3e8c0247f79b4a0f175ad5fb3249f9863b4b5510fb7c7e2b99491b8 +size 15708 diff --git a/006000/training_state/scheduler_state.json b/006000/training_state/scheduler_state.json new file mode 100644 index 0000000000000000000000000000000000000000..d488d9982f4c9b7366b1068b8b4d6cfb81f933a8 --- /dev/null +++ b/006000/training_state/scheduler_state.json @@ -0,0 +1,15 @@ +{ + "base_lrs": [ + 0.001 + ], + "last_epoch": 6000, + "_step_count": 6001, + "_is_initial": false, + "_get_lr_called_within_step": false, + "_last_lr": [ + 0.0004109423525312738 + ], + "lr_lambdas": [ + null + ] +} \ No newline at end of file diff --git a/006000/training_state/training_step.json b/006000/training_state/training_step.json new file mode 100644 index 0000000000000000000000000000000000000000..e267ac589be64705f8674638b9f5099c886778da --- /dev/null +++ b/006000/training_state/training_step.json @@ -0,0 +1,3 @@ +{ + "step": 6000 +} \ No newline at end of file diff --git a/007000/pretrained_model/README.md b/007000/pretrained_model/README.md new file mode 100644 index 0000000000000000000000000000000000000000..6565338d7b6577ddbee192786d397d22079b8338 --- /dev/null +++ b/007000/pretrained_model/README.md @@ -0,0 +1,204 @@ +--- +library_name: peft +tags: +- base_model:adapter:lerobot/pi0_base +- lora +--- + +# Model Card for Model ID + + + + + +## Model Details + +### Model Description + + + + + +- **Developed by:** [More Information Needed] +- **Funded by [optional]:** [More Information Needed] +- **Shared by [optional]:** [More Information Needed] +- **Model type:** [More Information Needed] +- **Language(s) (NLP):** [More Information Needed] +- **License:** [More Information Needed] +- **Finetuned from model [optional]:** [More Information Needed] + +### Model Sources [optional] + + + +- **Repository:** [More Information Needed] +- **Paper [optional]:** [More Information Needed] +- **Demo [optional]:** [More Information Needed] + +## Uses + + + +### Direct Use + + + +[More Information Needed] + +### Downstream Use [optional] + + + +[More Information Needed] + +### Out-of-Scope Use + + + +[More Information Needed] + +## Bias, Risks, and Limitations + + + +[More Information Needed] + +### Recommendations + + + +Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations. + +## How to Get Started with the Model + +Use the code below to get started with the model. + +[More Information Needed] + +## Training Details + +### Training Data + + + +[More Information Needed] + +### Training Procedure + + + +#### Preprocessing [optional] + +[More Information Needed] + + +#### Training Hyperparameters + +- **Training regime:** [More Information Needed] + +#### Speeds, Sizes, Times [optional] + + + +[More Information Needed] + +## Evaluation + + + +### Testing Data, Factors & Metrics + +#### Testing Data + + + +[More Information Needed] + +#### Factors + + + +[More Information Needed] + +#### Metrics + + + +[More Information Needed] + +### Results + +[More Information Needed] + +#### Summary + + + +## Model Examination [optional] + + + +[More Information Needed] + +## Environmental Impact + + + +Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700). + +- **Hardware Type:** [More Information Needed] +- **Hours used:** [More Information Needed] +- **Cloud Provider:** [More Information Needed] +- **Compute Region:** [More Information Needed] +- **Carbon Emitted:** [More Information Needed] + +## Technical Specifications [optional] + +### Model Architecture and Objective + +[More Information Needed] + +### Compute Infrastructure + +[More Information Needed] + +#### Hardware + +[More Information Needed] + +#### Software + +[More Information Needed] + +## Citation [optional] + + + +**BibTeX:** + +[More Information Needed] + +**APA:** + +[More Information Needed] + +## Glossary [optional] + + + +[More Information Needed] + +## More Information [optional] + +[More Information Needed] + +## Model Card Authors [optional] + +[More Information Needed] + +## Model Card Contact + +[More Information Needed] +### Framework versions + +- PEFT 0.18.1 \ No newline at end of file diff --git a/007000/pretrained_model/adapter_config.json b/007000/pretrained_model/adapter_config.json new file mode 100644 index 0000000000000000000000000000000000000000..3c00acefa2527617e9a485e421d89b9166fed77b --- /dev/null +++ b/007000/pretrained_model/adapter_config.json @@ -0,0 +1,41 @@ +{ + "alora_invocation_tokens": null, + "alpha_pattern": {}, + "arrow_config": null, + "auto_mapping": { + "base_model_class": "PI0Policy", + "parent_library": "lerobot.policies.pi0.modeling_pi0" + }, + "base_model_name_or_path": "lerobot/pi0_base", + "bias": "none", + "corda_config": null, + "ensure_weight_tying": false, + "eva_config": null, + "exclude_modules": null, + "fan_in_fan_out": false, + "inference_mode": true, + "init_lora_weights": true, + "layer_replication": null, + "layers_pattern": null, + "layers_to_transform": null, + "loftq_config": {}, + "lora_alpha": 8, + "lora_bias": false, + "lora_dropout": 0.0, + "megatron_config": null, + "megatron_core": "megatron.core", + "modules_to_save": [], + "peft_type": "LORA", + "peft_version": "0.18.1", + "qalora_group_size": 16, + "r": 32, + "rank_pattern": {}, + "revision": null, + "target_modules": "(.*\\.gemma_expert\\..*\\.self_attn\\.(q|v)_proj|model\\.(state_proj|action_in_proj|action_out_proj|action_time_mlp_in|action_time_mlp_out))", + "target_parameters": null, + "task_type": null, + "trainable_token_indices": null, + "use_dora": false, + "use_qalora": false, + "use_rslora": false +} \ No newline at end of file diff --git a/007000/pretrained_model/adapter_model.safetensors b/007000/pretrained_model/adapter_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..e2251319283c03ecf52683b90a69e847ae5520db --- /dev/null +++ b/007000/pretrained_model/adapter_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1d8db50f8b58857a62075d4f954ca85daa1c539ba43bdacd7517c5453a6ba38c +size 11101592 diff --git a/007000/pretrained_model/config.json b/007000/pretrained_model/config.json new file mode 100644 index 0000000000000000000000000000000000000000..5bded5dca90b3c498f402c92f09b4af980008111 --- /dev/null +++ b/007000/pretrained_model/config.json @@ -0,0 +1,113 @@ +{ + "type": "pi0", + "n_obs_steps": 1, + "input_features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "device": "cuda", + "use_amp": false, + "use_peft": true, + "push_to_hub": true, + "repo_id": "F-Fer/pi0-xlerobot-0", + "private": null, + "tags": null, + "license": null, + "pretrained_path": "lerobot/pi0_base", + "paligemma_variant": "gemma_2b", + "action_expert_variant": "gemma_300m", + "dtype": "bfloat16", + "chunk_size": 50, + "n_action_steps": 50, + "max_state_dim": 32, + "max_action_dim": 32, + "num_inference_steps": 10, + "time_sampling_beta_alpha": 1.5, + "time_sampling_beta_beta": 1.0, + "time_sampling_scale": 0.999, + "time_sampling_offset": 0.001, + "min_period": 0.004, + "max_period": 4.0, + "use_relative_actions": false, + "relative_exclude_joints": [ + "gripper" + ], + "action_feature_names": [ + "left_shoulder_pan.pos", + "left_shoulder_lift.pos", + "left_elbow_flex.pos", + "left_wrist_flex.pos", + "left_wrist_roll.pos", + "left_gripper.pos", + "right_shoulder_pan.pos", + "right_shoulder_lift.pos", + "right_elbow_flex.pos", + "right_wrist_flex.pos", + "right_wrist_roll.pos", + "right_gripper.pos" + ], + "rtc_config": null, + "image_resolution": [ + 224, + 224 + ], + "empty_cameras": 0, + "normalization_mapping": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + }, + "gradient_checkpointing": true, + "compile_model": true, + "compile_mode": "max-autotune", + "freeze_vision_encoder": false, + "train_expert_only": false, + "optimizer_lr": 0.001, + "optimizer_betas": [ + 0.9, + 0.95 + ], + "optimizer_eps": 1e-08, + "optimizer_weight_decay": 0.01, + "optimizer_grad_clip_norm": 1.0, + "scheduler_warmup_steps": 1000, + "scheduler_decay_steps": 30000, + "scheduler_decay_lr": 0.0001, + "tokenizer_max_length": 48 +} \ No newline at end of file diff --git a/007000/pretrained_model/policy_postprocessor.json b/007000/pretrained_model/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..25a2e623f8277a52e915c3701c7cd9cf77ddf8d6 --- /dev/null +++ b/007000/pretrained_model/policy_postprocessor.json @@ -0,0 +1,32 @@ +{ + "name": "policy_postprocessor", + "steps": [ + { + "registry_name": "unnormalizer_processor", + "config": { + "eps": 1e-08, + "features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "norm_map": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + } + }, + "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors" + }, + { + "registry_name": "device_processor", + "config": { + "device": "cpu", + "float_dtype": null + } + } + ] +} \ No newline at end of file diff --git a/007000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/007000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..08107a06a9226905f59e970a3d28ddc923a42695 --- /dev/null +++ b/007000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:794c084955a7613103705a47253d9fa3f66d03ead8978fa493cad135498ca62b +size 9192 diff --git a/007000/pretrained_model/policy_preprocessor.json b/007000/pretrained_model/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..8c39dc4710ff54b6f6d1631e34f0484383336eca --- /dev/null +++ b/007000/pretrained_model/policy_preprocessor.json @@ -0,0 +1,91 @@ +{ + "name": "policy_preprocessor", + "steps": [ + { + "registry_name": "rename_observations_processor", + "config": { + "rename_map": { + "observation.images.left_front": "observation.images.base_0_rgb", + "observation.images.left_left": "observation.images.left_wrist_0_rgb", + "observation.images.left_right": "observation.images.right_wrist_0_rgb" + } + } + }, + { + "registry_name": "to_batch_processor", + "config": {} + }, + { + "registry_name": "pi0_new_line_processor", + "config": {} + }, + { + "registry_name": "tokenizer_processor", + "config": { + "max_length": 48, + "task_key": "task", + "padding_side": "right", + "padding": "max_length", + "truncation": true, + "tokenizer_name": "google/paligemma-3b-pt-224" + } + }, + { + "registry_name": "device_processor", + "config": { + "device": "cuda", + "float_dtype": null + } + }, + { + "registry_name": "normalizer_processor", + "config": { + "eps": 1e-08, + "features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + }, + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "norm_map": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + } + }, + "state_file": "policy_preprocessor_step_5_normalizer_processor.safetensors" + } + ] +} \ No newline at end of file diff --git a/007000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors b/007000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..08107a06a9226905f59e970a3d28ddc923a42695 --- /dev/null +++ b/007000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:794c084955a7613103705a47253d9fa3f66d03ead8978fa493cad135498ca62b +size 9192 diff --git a/007000/pretrained_model/train_config.json b/007000/pretrained_model/train_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8b15c723555caefa3bdb9e5e9097937ebfef3fad --- /dev/null +++ b/007000/pretrained_model/train_config.json @@ -0,0 +1,261 @@ +{ + "dataset": { + "repo_id": "F-Fer/xlerobot-0", + "root": null, + "episodes": null, + "image_transforms": { + "enable": false, + "max_num_transforms": 3, + "random_order": false, + "tfs": { + "brightness": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "brightness": [ + 0.8, + 1.2 + ] + } + }, + "contrast": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "contrast": [ + 0.8, + 1.2 + ] + } + }, + "saturation": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "saturation": [ + 0.5, + 1.5 + ] + } + }, + "hue": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "hue": [ + -0.05, + 0.05 + ] + } + }, + "sharpness": { + "weight": 1.0, + "type": "SharpnessJitter", + "kwargs": { + "sharpness": [ + 0.5, + 1.5 + ] + } + }, + "affine": { + "weight": 1.0, + "type": "RandomAffine", + "kwargs": { + "degrees": [ + -5.0, + 5.0 + ], + "translate": [ + 0.05, + 0.05 + ] + } + } + } + }, + "revision": null, + "use_imagenet_stats": true, + "video_backend": "torchcodec", + "streaming": false + }, + "env": null, + "policy": { + "type": "pi0", + "n_obs_steps": 1, + "input_features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "device": "cuda", + "use_amp": false, + "use_peft": true, + "push_to_hub": true, + "repo_id": "F-Fer/pi0-xlerobot-0", + "private": null, + "tags": null, + "license": null, + "pretrained_path": "lerobot/pi0_base", + "paligemma_variant": "gemma_2b", + "action_expert_variant": "gemma_300m", + "dtype": "bfloat16", + "chunk_size": 50, + "n_action_steps": 50, + "max_state_dim": 32, + "max_action_dim": 32, + "num_inference_steps": 10, + "time_sampling_beta_alpha": 1.5, + "time_sampling_beta_beta": 1.0, + "time_sampling_scale": 0.999, + "time_sampling_offset": 0.001, + "min_period": 0.004, + "max_period": 4.0, + "use_relative_actions": false, + "relative_exclude_joints": [ + "gripper" + ], + "action_feature_names": [ + "left_shoulder_pan.pos", + "left_shoulder_lift.pos", + "left_elbow_flex.pos", + "left_wrist_flex.pos", + "left_wrist_roll.pos", + "left_gripper.pos", + "right_shoulder_pan.pos", + "right_shoulder_lift.pos", + "right_elbow_flex.pos", + "right_wrist_flex.pos", + "right_wrist_roll.pos", + "right_gripper.pos" + ], + "rtc_config": null, + "image_resolution": [ + 224, + 224 + ], + "empty_cameras": 0, + "normalization_mapping": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + }, + "gradient_checkpointing": true, + "compile_model": true, + "compile_mode": "max-autotune", + "freeze_vision_encoder": false, + "train_expert_only": false, + "optimizer_lr": 0.001, + "optimizer_betas": [ + 0.9, + 0.95 + ], + "optimizer_eps": 1e-08, + "optimizer_weight_decay": 0.01, + "optimizer_grad_clip_norm": 1.0, + "scheduler_warmup_steps": 1000, + "scheduler_decay_steps": 30000, + "scheduler_decay_lr": 0.0001, + "tokenizer_max_length": 48 + }, + "output_dir": "outputs/train/2026-04-10/14-19-24_pi0", + "job_name": "pi0", + "resume": false, + "seed": 1000, + "cudnn_deterministic": false, + "num_workers": 4, + "batch_size": 4, + "steps": 10000, + "eval_freq": -1, + "log_freq": 100, + "tolerance_s": 0.0001, + "save_checkpoint": true, + "save_freq": 1000, + "use_policy_training_preset": true, + "optimizer": { + "type": "adamw", + "lr": 0.001, + "weight_decay": 0.01, + "grad_clip_norm": 1.0, + "betas": [ + 0.9, + 0.95 + ], + "eps": 1e-08 + }, + "scheduler": { + "type": "cosine_decay_with_warmup", + "num_warmup_steps": 1000, + "num_decay_steps": 30000, + "peak_lr": 0.001, + "decay_lr": 0.0001 + }, + "eval": { + "n_episodes": 50, + "batch_size": 50, + "use_async_envs": false + }, + "wandb": { + "enable": true, + "disable_artifact": false, + "project": "xlerobot", + "entity": null, + "notes": null, + "run_id": "tmqkc0d7", + "mode": null, + "add_tags": true + }, + "peft": { + "target_modules": null, + "full_training_modules": null, + "method_type": "LORA", + "init_type": null, + "r": 32 + }, + "use_rabc": false, + "rabc_progress_path": null, + "rabc_kappa": 0.01, + "rabc_epsilon": 1e-06, + "rabc_head_mode": "sparse", + "rename_map": { + "observation.images.left_front": "observation.images.base_0_rgb", + "observation.images.left_left": "observation.images.left_wrist_0_rgb", + "observation.images.left_right": "observation.images.right_wrist_0_rgb" + }, + "checkpoint_path": null +} \ No newline at end of file diff --git a/007000/training_state/optimizer_param_groups.json b/007000/training_state/optimizer_param_groups.json new file mode 100644 index 0000000000000000000000000000000000000000..9607f3225554a3d60c0e5d649fb216db045129bc --- /dev/null +++ b/007000/training_state/optimizer_param_groups.json @@ -0,0 +1,881 @@ +[ + { + "lr": 0.0002854966364683872, + "betas": [ + 0.9, + 0.95 + ], + "eps": 1e-08, + "weight_decay": 0.01, + "amsgrad": false, + "maximize": false, + "foreach": null, + "capturable": false, + "differentiable": false, + "fused": null, + "decoupled_weight_decay": true, + "initial_lr": 0.001, + "params": [ + 0, + 1, + 2, + 3, + 4, + 5, + 6, + 7, + 8, + 9, + 10, + 11, + 12, + 13, + 14, + 15, + 16, + 17, + 18, + 19, + 20, + 21, + 22, + 23, + 24, + 25, + 26, + 27, + 28, + 29, + 30, + 31, + 32, + 33, + 34, + 35, + 36, + 37, + 38, + 39, + 40, + 41, + 42, + 43, + 44, + 45, + 46, + 47, + 48, + 49, + 50, + 51, + 52, + 53, + 54, + 55, + 56, + 57, + 58, + 59, + 60, + 61, + 62, + 63, + 64, + 65, + 66, + 67, + 68, + 69, + 70, + 71, + 72, + 73, + 74, + 75, + 76, + 77, + 78, + 79, + 80, + 81, + 82, + 83, + 84, + 85, + 86, + 87, + 88, + 89, + 90, + 91, + 92, + 93, + 94, + 95, + 96, + 97, + 98, + 99, + 100, + 101, + 102, + 103, + 104, + 105, + 106, + 107, + 108, + 109, + 110, + 111, + 112, + 113, + 114, + 115, + 116, + 117, + 118, + 119, + 120, + 121, + 122, + 123, + 124, + 125, + 126, + 127, + 128, + 129, + 130, + 131, + 132, + 133, + 134, + 135, + 136, + 137, + 138, + 139, + 140, + 141, + 142, + 143, + 144, + 145, + 146, + 147, + 148, + 149, + 150, + 151, + 152, + 153, + 154, + 155, + 156, + 157, + 158, + 159, + 160, + 161, + 162, + 163, + 164, + 165, + 166, + 167, + 168, + 169, + 170, + 171, + 172, + 173, + 174, + 175, + 176, + 177, + 178, + 179, + 180, + 181, + 182, + 183, + 184, + 185, + 186, + 187, + 188, + 189, + 190, + 191, + 192, + 193, + 194, + 195, + 196, + 197, + 198, + 199, + 200, + 201, + 202, + 203, + 204, + 205, + 206, + 207, + 208, + 209, + 210, + 211, + 212, + 213, + 214, + 215, + 216, + 217, + 218, + 219, + 220, + 221, + 222, + 223, + 224, + 225, + 226, + 227, + 228, + 229, + 230, + 231, + 232, + 233, + 234, + 235, + 236, + 237, + 238, + 239, + 240, + 241, + 242, + 243, + 244, + 245, + 246, + 247, + 248, + 249, + 250, + 251, + 252, + 253, + 254, + 255, + 256, + 257, + 258, + 259, + 260, + 261, + 262, + 263, + 264, + 265, + 266, + 267, + 268, + 269, + 270, + 271, + 272, + 273, + 274, + 275, + 276, + 277, + 278, + 279, + 280, + 281, + 282, + 283, + 284, + 285, + 286, + 287, + 288, + 289, + 290, + 291, + 292, + 293, + 294, + 295, + 296, + 297, + 298, + 299, + 300, + 301, + 302, + 303, + 304, + 305, + 306, + 307, + 308, + 309, + 310, + 311, + 312, + 313, + 314, + 315, + 316, + 317, + 318, + 319, + 320, + 321, + 322, + 323, + 324, + 325, + 326, + 327, + 328, + 329, + 330, + 331, + 332, + 333, + 334, + 335, + 336, + 337, + 338, + 339, + 340, + 341, + 342, + 343, + 344, + 345, + 346, + 347, + 348, + 349, + 350, + 351, + 352, + 353, + 354, + 355, + 356, + 357, + 358, + 359, + 360, + 361, + 362, + 363, + 364, + 365, + 366, + 367, + 368, + 369, + 370, + 371, + 372, + 373, + 374, + 375, + 376, + 377, + 378, + 379, + 380, + 381, + 382, + 383, + 384, + 385, + 386, + 387, + 388, + 389, + 390, + 391, + 392, + 393, + 394, + 395, + 396, + 397, + 398, + 399, + 400, + 401, + 402, + 403, + 404, + 405, + 406, + 407, + 408, + 409, + 410, + 411, + 412, + 413, + 414, + 415, + 416, + 417, + 418, + 419, + 420, + 421, + 422, + 423, + 424, + 425, + 426, + 427, + 428, + 429, + 430, + 431, + 432, + 433, + 434, + 435, + 436, + 437, + 438, + 439, + 440, + 441, + 442, + 443, + 444, + 445, + 446, + 447, + 448, + 449, + 450, + 451, + 452, + 453, + 454, + 455, + 456, + 457, + 458, + 459, + 460, + 461, + 462, + 463, + 464, + 465, + 466, + 467, + 468, + 469, + 470, + 471, + 472, + 473, + 474, + 475, + 476, + 477, + 478, + 479, + 480, + 481, + 482, + 483, + 484, + 485, + 486, + 487, + 488, + 489, + 490, + 491, + 492, + 493, + 494, + 495, + 496, + 497, + 498, + 499, + 500, + 501, + 502, + 503, + 504, + 505, + 506, + 507, + 508, + 509, + 510, + 511, + 512, + 513, + 514, + 515, + 516, + 517, + 518, + 519, + 520, + 521, + 522, + 523, + 524, + 525, + 526, + 527, + 528, + 529, + 530, + 531, + 532, + 533, + 534, + 535, + 536, + 537, + 538, + 539, + 540, + 541, + 542, + 543, + 544, + 545, + 546, + 547, + 548, + 549, + 550, + 551, + 552, + 553, + 554, + 555, + 556, + 557, + 558, + 559, + 560, + 561, + 562, + 563, + 564, + 565, + 566, + 567, + 568, + 569, + 570, + 571, + 572, + 573, + 574, + 575, + 576, + 577, + 578, + 579, + 580, + 581, + 582, + 583, + 584, + 585, + 586, + 587, + 588, + 589, + 590, + 591, + 592, + 593, + 594, + 595, + 596, + 597, + 598, + 599, + 600, + 601, + 602, + 603, + 604, + 605, + 606, + 607, + 608, + 609, + 610, + 611, + 612, + 613, + 614, + 615, + 616, + 617, + 618, + 619, + 620, + 621, + 622, + 623, + 624, + 625, + 626, + 627, + 628, + 629, + 630, + 631, + 632, + 633, + 634, + 635, + 636, + 637, + 638, + 639, + 640, + 641, + 642, + 643, + 644, + 645, + 646, + 647, + 648, + 649, + 650, + 651, + 652, + 653, + 654, + 655, + 656, + 657, + 658, + 659, + 660, + 661, + 662, + 663, + 664, + 665, + 666, + 667, + 668, + 669, + 670, + 671, + 672, + 673, + 674, + 675, + 676, + 677, + 678, + 679, + 680, + 681, + 682, + 683, + 684, + 685, + 686, + 687, + 688, + 689, + 690, + 691, + 692, + 693, + 694, + 695, + 696, + 697, + 698, + 699, + 700, + 701, + 702, + 703, + 704, + 705, + 706, + 707, + 708, + 709, + 710, + 711, + 712, + 713, + 714, + 715, + 716, + 717, + 718, + 719, + 720, + 721, + 722, + 723, + 724, + 725, + 726, + 727, + 728, + 729, + 730, + 731, + 732, + 733, + 734, + 735, + 736, + 737, + 738, + 739, + 740, + 741, + 742, + 743, + 744, + 745, + 746, + 747, + 748, + 749, + 750, + 751, + 752, + 753, + 754, + 755, + 756, + 757, + 758, + 759, + 760, + 761, + 762, + 763, + 764, + 765, + 766, + 767, + 768, + 769, + 770, + 771, + 772, + 773, + 774, + 775, + 776, + 777, + 778, + 779, + 780, + 781, + 782, + 783, + 784, + 785, + 786, + 787, + 788, + 789, + 790, + 791, + 792, + 793, + 794, + 795, + 796, + 797, + 798, + 799, + 800, + 801, + 802, + 803, + 804, + 805, + 806, + 807, + 808, + 809, + 810, + 811, + 812, + 813, + 814, + 815, + 816, + 817, + 818, + 819, + 820, + 821, + 822, + 823, + 824, + 825, + 826, + 827, + 828, + 829, + 830, + 831, + 832, + 833, + 834, + 835, + 836, + 837, + 838, + 839, + 840, + 841, + 842, + 843, + 844, + 845, + 846, + 847, + 848, + 849, + 850, + 851, + 852, + 853, + 854, + 855, + 856, + 857, + 858, + 859 + ] + } +] \ No newline at end of file diff --git a/007000/training_state/optimizer_state.safetensors b/007000/training_state/optimizer_state.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..2befcb4660843a306023533c008a02cd91590390 --- /dev/null +++ b/007000/training_state/optimizer_state.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:93c53bc9f7890818cfaad99fe25a0501d8c4c4cebdc87131939d969298453778 +size 22197120 diff --git a/007000/training_state/rng_state.safetensors b/007000/training_state/rng_state.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..9cce70714a1c33fcf42e251e5965d5357fb7b43f --- /dev/null +++ b/007000/training_state/rng_state.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f1e47963fa12e14b1bbdeee1ccb8007ae6368e02b2192692529ffe7098d7ac94 +size 15708 diff --git a/007000/training_state/scheduler_state.json b/007000/training_state/scheduler_state.json new file mode 100644 index 0000000000000000000000000000000000000000..1a03e327cd4a2b1485c6e407efaad4ceeec389e8 --- /dev/null +++ b/007000/training_state/scheduler_state.json @@ -0,0 +1,15 @@ +{ + "base_lrs": [ + 0.001 + ], + "last_epoch": 7000, + "_step_count": 7001, + "_is_initial": false, + "_get_lr_called_within_step": false, + "_last_lr": [ + 0.0002854966364683872 + ], + "lr_lambdas": [ + null + ] +} \ No newline at end of file diff --git a/007000/training_state/training_step.json b/007000/training_state/training_step.json new file mode 100644 index 0000000000000000000000000000000000000000..1f4c730c4d88ece076ae9d4e01ed7fbe60b0415d --- /dev/null +++ b/007000/training_state/training_step.json @@ -0,0 +1,3 @@ +{ + "step": 7000 +} \ No newline at end of file diff --git a/008000/pretrained_model/README.md b/008000/pretrained_model/README.md new file mode 100644 index 0000000000000000000000000000000000000000..6565338d7b6577ddbee192786d397d22079b8338 --- /dev/null +++ b/008000/pretrained_model/README.md @@ -0,0 +1,204 @@ +--- +library_name: peft +tags: +- base_model:adapter:lerobot/pi0_base +- lora +--- + +# Model Card for Model ID + + + + + +## Model Details + +### Model Description + + + + + +- **Developed by:** [More Information Needed] +- **Funded by [optional]:** [More Information Needed] +- **Shared by [optional]:** [More Information Needed] +- **Model type:** [More Information Needed] +- **Language(s) (NLP):** [More Information Needed] +- **License:** [More Information Needed] +- **Finetuned from model [optional]:** [More Information Needed] + +### Model Sources [optional] + + + +- **Repository:** [More Information Needed] +- **Paper [optional]:** [More Information Needed] +- **Demo [optional]:** [More Information Needed] + +## Uses + + + +### Direct Use + + + +[More Information Needed] + +### Downstream Use [optional] + + + +[More Information Needed] + +### Out-of-Scope Use + + + +[More Information Needed] + +## Bias, Risks, and Limitations + + + +[More Information Needed] + +### Recommendations + + + +Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations. + +## How to Get Started with the Model + +Use the code below to get started with the model. + +[More Information Needed] + +## Training Details + +### Training Data + + + +[More Information Needed] + +### Training Procedure + + + +#### Preprocessing [optional] + +[More Information Needed] + + +#### Training Hyperparameters + +- **Training regime:** [More Information Needed] + +#### Speeds, Sizes, Times [optional] + + + +[More Information Needed] + +## Evaluation + + + +### Testing Data, Factors & Metrics + +#### Testing Data + + + +[More Information Needed] + +#### Factors + + + +[More Information Needed] + +#### Metrics + + + +[More Information Needed] + +### Results + +[More Information Needed] + +#### Summary + + + +## Model Examination [optional] + + + +[More Information Needed] + +## Environmental Impact + + + +Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700). + +- **Hardware Type:** [More Information Needed] +- **Hours used:** [More Information Needed] +- **Cloud Provider:** [More Information Needed] +- **Compute Region:** [More Information Needed] +- **Carbon Emitted:** [More Information Needed] + +## Technical Specifications [optional] + +### Model Architecture and Objective + +[More Information Needed] + +### Compute Infrastructure + +[More Information Needed] + +#### Hardware + +[More Information Needed] + +#### Software + +[More Information Needed] + +## Citation [optional] + + + +**BibTeX:** + +[More Information Needed] + +**APA:** + +[More Information Needed] + +## Glossary [optional] + + + +[More Information Needed] + +## More Information [optional] + +[More Information Needed] + +## Model Card Authors [optional] + +[More Information Needed] + +## Model Card Contact + +[More Information Needed] +### Framework versions + +- PEFT 0.18.1 \ No newline at end of file diff --git a/008000/pretrained_model/adapter_config.json b/008000/pretrained_model/adapter_config.json new file mode 100644 index 0000000000000000000000000000000000000000..3c00acefa2527617e9a485e421d89b9166fed77b --- /dev/null +++ b/008000/pretrained_model/adapter_config.json @@ -0,0 +1,41 @@ +{ + "alora_invocation_tokens": null, + "alpha_pattern": {}, + "arrow_config": null, + "auto_mapping": { + "base_model_class": "PI0Policy", + "parent_library": "lerobot.policies.pi0.modeling_pi0" + }, + "base_model_name_or_path": "lerobot/pi0_base", + "bias": "none", + "corda_config": null, + "ensure_weight_tying": false, + "eva_config": null, + "exclude_modules": null, + "fan_in_fan_out": false, + "inference_mode": true, + "init_lora_weights": true, + "layer_replication": null, + "layers_pattern": null, + "layers_to_transform": null, + "loftq_config": {}, + "lora_alpha": 8, + "lora_bias": false, + "lora_dropout": 0.0, + "megatron_config": null, + "megatron_core": "megatron.core", + "modules_to_save": [], + "peft_type": "LORA", + "peft_version": "0.18.1", + "qalora_group_size": 16, + "r": 32, + "rank_pattern": {}, + "revision": null, + "target_modules": "(.*\\.gemma_expert\\..*\\.self_attn\\.(q|v)_proj|model\\.(state_proj|action_in_proj|action_out_proj|action_time_mlp_in|action_time_mlp_out))", + "target_parameters": null, + "task_type": null, + "trainable_token_indices": null, + "use_dora": false, + "use_qalora": false, + "use_rslora": false +} \ No newline at end of file diff --git a/008000/pretrained_model/adapter_model.safetensors b/008000/pretrained_model/adapter_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..5924bba01206ad1e1b827a8c97aa1c2c8aff1f5d --- /dev/null +++ b/008000/pretrained_model/adapter_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fa711c15f3e2bed5ddc2eb35a1bdfe4516344d13134e1e7e5ee199665452bfa3 +size 11101592 diff --git a/008000/pretrained_model/config.json b/008000/pretrained_model/config.json new file mode 100644 index 0000000000000000000000000000000000000000..5bded5dca90b3c498f402c92f09b4af980008111 --- /dev/null +++ b/008000/pretrained_model/config.json @@ -0,0 +1,113 @@ +{ + "type": "pi0", + "n_obs_steps": 1, + "input_features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "device": "cuda", + "use_amp": false, + "use_peft": true, + "push_to_hub": true, + "repo_id": "F-Fer/pi0-xlerobot-0", + "private": null, + "tags": null, + "license": null, + "pretrained_path": "lerobot/pi0_base", + "paligemma_variant": "gemma_2b", + "action_expert_variant": "gemma_300m", + "dtype": "bfloat16", + "chunk_size": 50, + "n_action_steps": 50, + "max_state_dim": 32, + "max_action_dim": 32, + "num_inference_steps": 10, + "time_sampling_beta_alpha": 1.5, + "time_sampling_beta_beta": 1.0, + "time_sampling_scale": 0.999, + "time_sampling_offset": 0.001, + "min_period": 0.004, + "max_period": 4.0, + "use_relative_actions": false, + "relative_exclude_joints": [ + "gripper" + ], + "action_feature_names": [ + "left_shoulder_pan.pos", + "left_shoulder_lift.pos", + "left_elbow_flex.pos", + "left_wrist_flex.pos", + "left_wrist_roll.pos", + "left_gripper.pos", + "right_shoulder_pan.pos", + "right_shoulder_lift.pos", + "right_elbow_flex.pos", + "right_wrist_flex.pos", + "right_wrist_roll.pos", + "right_gripper.pos" + ], + "rtc_config": null, + "image_resolution": [ + 224, + 224 + ], + "empty_cameras": 0, + "normalization_mapping": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + }, + "gradient_checkpointing": true, + "compile_model": true, + "compile_mode": "max-autotune", + "freeze_vision_encoder": false, + "train_expert_only": false, + "optimizer_lr": 0.001, + "optimizer_betas": [ + 0.9, + 0.95 + ], + "optimizer_eps": 1e-08, + "optimizer_weight_decay": 0.01, + "optimizer_grad_clip_norm": 1.0, + "scheduler_warmup_steps": 1000, + "scheduler_decay_steps": 30000, + "scheduler_decay_lr": 0.0001, + "tokenizer_max_length": 48 +} \ No newline at end of file diff --git a/008000/pretrained_model/policy_postprocessor.json b/008000/pretrained_model/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..25a2e623f8277a52e915c3701c7cd9cf77ddf8d6 --- /dev/null +++ b/008000/pretrained_model/policy_postprocessor.json @@ -0,0 +1,32 @@ +{ + "name": "policy_postprocessor", + "steps": [ + { + "registry_name": "unnormalizer_processor", + "config": { + "eps": 1e-08, + "features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "norm_map": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + } + }, + "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors" + }, + { + "registry_name": "device_processor", + "config": { + "device": "cpu", + "float_dtype": null + } + } + ] +} \ No newline at end of file diff --git a/008000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/008000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..08107a06a9226905f59e970a3d28ddc923a42695 --- /dev/null +++ b/008000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:794c084955a7613103705a47253d9fa3f66d03ead8978fa493cad135498ca62b +size 9192 diff --git a/008000/pretrained_model/policy_preprocessor.json b/008000/pretrained_model/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..8c39dc4710ff54b6f6d1631e34f0484383336eca --- /dev/null +++ b/008000/pretrained_model/policy_preprocessor.json @@ -0,0 +1,91 @@ +{ + "name": "policy_preprocessor", + "steps": [ + { + "registry_name": "rename_observations_processor", + "config": { + "rename_map": { + "observation.images.left_front": "observation.images.base_0_rgb", + "observation.images.left_left": "observation.images.left_wrist_0_rgb", + "observation.images.left_right": "observation.images.right_wrist_0_rgb" + } + } + }, + { + "registry_name": "to_batch_processor", + "config": {} + }, + { + "registry_name": "pi0_new_line_processor", + "config": {} + }, + { + "registry_name": "tokenizer_processor", + "config": { + "max_length": 48, + "task_key": "task", + "padding_side": "right", + "padding": "max_length", + "truncation": true, + "tokenizer_name": "google/paligemma-3b-pt-224" + } + }, + { + "registry_name": "device_processor", + "config": { + "device": "cuda", + "float_dtype": null + } + }, + { + "registry_name": "normalizer_processor", + "config": { + "eps": 1e-08, + "features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + }, + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "norm_map": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + } + }, + "state_file": "policy_preprocessor_step_5_normalizer_processor.safetensors" + } + ] +} \ No newline at end of file diff --git a/008000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors b/008000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..08107a06a9226905f59e970a3d28ddc923a42695 --- /dev/null +++ b/008000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:794c084955a7613103705a47253d9fa3f66d03ead8978fa493cad135498ca62b +size 9192 diff --git a/008000/pretrained_model/train_config.json b/008000/pretrained_model/train_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8b15c723555caefa3bdb9e5e9097937ebfef3fad --- /dev/null +++ b/008000/pretrained_model/train_config.json @@ -0,0 +1,261 @@ +{ + "dataset": { + "repo_id": "F-Fer/xlerobot-0", + "root": null, + "episodes": null, + "image_transforms": { + "enable": false, + "max_num_transforms": 3, + "random_order": false, + "tfs": { + "brightness": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "brightness": [ + 0.8, + 1.2 + ] + } + }, + "contrast": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "contrast": [ + 0.8, + 1.2 + ] + } + }, + "saturation": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "saturation": [ + 0.5, + 1.5 + ] + } + }, + "hue": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "hue": [ + -0.05, + 0.05 + ] + } + }, + "sharpness": { + "weight": 1.0, + "type": "SharpnessJitter", + "kwargs": { + "sharpness": [ + 0.5, + 1.5 + ] + } + }, + "affine": { + "weight": 1.0, + "type": "RandomAffine", + "kwargs": { + "degrees": [ + -5.0, + 5.0 + ], + "translate": [ + 0.05, + 0.05 + ] + } + } + } + }, + "revision": null, + "use_imagenet_stats": true, + "video_backend": "torchcodec", + "streaming": false + }, + "env": null, + "policy": { + "type": "pi0", + "n_obs_steps": 1, + "input_features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "device": "cuda", + "use_amp": false, + "use_peft": true, + "push_to_hub": true, + "repo_id": "F-Fer/pi0-xlerobot-0", + "private": null, + "tags": null, + "license": null, + "pretrained_path": "lerobot/pi0_base", + "paligemma_variant": "gemma_2b", + "action_expert_variant": "gemma_300m", + "dtype": "bfloat16", + "chunk_size": 50, + "n_action_steps": 50, + "max_state_dim": 32, + "max_action_dim": 32, + "num_inference_steps": 10, + "time_sampling_beta_alpha": 1.5, + "time_sampling_beta_beta": 1.0, + "time_sampling_scale": 0.999, + "time_sampling_offset": 0.001, + "min_period": 0.004, + "max_period": 4.0, + "use_relative_actions": false, + "relative_exclude_joints": [ + "gripper" + ], + "action_feature_names": [ + "left_shoulder_pan.pos", + "left_shoulder_lift.pos", + "left_elbow_flex.pos", + "left_wrist_flex.pos", + "left_wrist_roll.pos", + "left_gripper.pos", + "right_shoulder_pan.pos", + "right_shoulder_lift.pos", + "right_elbow_flex.pos", + "right_wrist_flex.pos", + "right_wrist_roll.pos", + "right_gripper.pos" + ], + "rtc_config": null, + "image_resolution": [ + 224, + 224 + ], + "empty_cameras": 0, + "normalization_mapping": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + }, + "gradient_checkpointing": true, + "compile_model": true, + "compile_mode": "max-autotune", + "freeze_vision_encoder": false, + "train_expert_only": false, + "optimizer_lr": 0.001, + "optimizer_betas": [ + 0.9, + 0.95 + ], + "optimizer_eps": 1e-08, + "optimizer_weight_decay": 0.01, + "optimizer_grad_clip_norm": 1.0, + "scheduler_warmup_steps": 1000, + "scheduler_decay_steps": 30000, + "scheduler_decay_lr": 0.0001, + "tokenizer_max_length": 48 + }, + "output_dir": "outputs/train/2026-04-10/14-19-24_pi0", + "job_name": "pi0", + "resume": false, + "seed": 1000, + "cudnn_deterministic": false, + "num_workers": 4, + "batch_size": 4, + "steps": 10000, + "eval_freq": -1, + "log_freq": 100, + "tolerance_s": 0.0001, + "save_checkpoint": true, + "save_freq": 1000, + "use_policy_training_preset": true, + "optimizer": { + "type": "adamw", + "lr": 0.001, + "weight_decay": 0.01, + "grad_clip_norm": 1.0, + "betas": [ + 0.9, + 0.95 + ], + "eps": 1e-08 + }, + "scheduler": { + "type": "cosine_decay_with_warmup", + "num_warmup_steps": 1000, + "num_decay_steps": 30000, + "peak_lr": 0.001, + "decay_lr": 0.0001 + }, + "eval": { + "n_episodes": 50, + "batch_size": 50, + "use_async_envs": false + }, + "wandb": { + "enable": true, + "disable_artifact": false, + "project": "xlerobot", + "entity": null, + "notes": null, + "run_id": "tmqkc0d7", + "mode": null, + "add_tags": true + }, + "peft": { + "target_modules": null, + "full_training_modules": null, + "method_type": "LORA", + "init_type": null, + "r": 32 + }, + "use_rabc": false, + "rabc_progress_path": null, + "rabc_kappa": 0.01, + "rabc_epsilon": 1e-06, + "rabc_head_mode": "sparse", + "rename_map": { + "observation.images.left_front": "observation.images.base_0_rgb", + "observation.images.left_left": "observation.images.left_wrist_0_rgb", + "observation.images.left_right": "observation.images.right_wrist_0_rgb" + }, + "checkpoint_path": null +} \ No newline at end of file diff --git a/008000/training_state/optimizer_param_groups.json b/008000/training_state/optimizer_param_groups.json new file mode 100644 index 0000000000000000000000000000000000000000..0e68f6772a15fcdb7035f278fa0342fe91816e74 --- /dev/null +++ b/008000/training_state/optimizer_param_groups.json @@ -0,0 +1,881 @@ +[ + { + "lr": 0.00018594235253127373, + "betas": [ + 0.9, + 0.95 + ], + "eps": 1e-08, + "weight_decay": 0.01, + "amsgrad": false, + "maximize": false, + "foreach": null, + "capturable": false, + "differentiable": false, + "fused": null, + "decoupled_weight_decay": true, + "initial_lr": 0.001, + "params": [ + 0, + 1, + 2, + 3, + 4, + 5, + 6, + 7, + 8, + 9, + 10, + 11, + 12, + 13, + 14, + 15, + 16, + 17, + 18, + 19, + 20, + 21, + 22, + 23, + 24, + 25, + 26, + 27, + 28, + 29, + 30, + 31, + 32, + 33, + 34, + 35, + 36, + 37, + 38, + 39, + 40, + 41, + 42, + 43, + 44, + 45, + 46, + 47, + 48, + 49, + 50, + 51, + 52, + 53, + 54, + 55, + 56, + 57, + 58, + 59, + 60, + 61, + 62, + 63, + 64, + 65, + 66, + 67, + 68, + 69, + 70, + 71, + 72, + 73, + 74, + 75, + 76, + 77, + 78, + 79, + 80, + 81, + 82, + 83, + 84, + 85, + 86, + 87, + 88, + 89, + 90, + 91, + 92, + 93, + 94, + 95, + 96, + 97, + 98, + 99, + 100, + 101, + 102, + 103, + 104, + 105, + 106, + 107, + 108, + 109, + 110, + 111, + 112, + 113, + 114, + 115, + 116, + 117, + 118, + 119, + 120, + 121, + 122, + 123, + 124, + 125, + 126, + 127, + 128, + 129, + 130, + 131, + 132, + 133, + 134, + 135, + 136, + 137, + 138, + 139, + 140, + 141, + 142, + 143, + 144, + 145, + 146, + 147, + 148, + 149, + 150, + 151, + 152, + 153, + 154, + 155, + 156, + 157, + 158, + 159, + 160, + 161, + 162, + 163, + 164, + 165, + 166, + 167, + 168, + 169, + 170, + 171, + 172, + 173, + 174, + 175, + 176, + 177, + 178, + 179, + 180, + 181, + 182, + 183, + 184, + 185, + 186, + 187, + 188, + 189, + 190, + 191, + 192, + 193, + 194, + 195, + 196, + 197, + 198, + 199, + 200, + 201, + 202, + 203, + 204, + 205, + 206, + 207, + 208, + 209, + 210, + 211, + 212, + 213, + 214, + 215, + 216, + 217, + 218, + 219, + 220, + 221, + 222, + 223, + 224, + 225, + 226, + 227, + 228, + 229, + 230, + 231, + 232, + 233, + 234, + 235, + 236, + 237, + 238, + 239, + 240, + 241, + 242, + 243, + 244, + 245, + 246, + 247, + 248, + 249, + 250, + 251, + 252, + 253, + 254, + 255, + 256, + 257, + 258, + 259, + 260, + 261, + 262, + 263, + 264, + 265, + 266, + 267, + 268, + 269, + 270, + 271, + 272, + 273, + 274, + 275, + 276, + 277, + 278, + 279, + 280, + 281, + 282, + 283, + 284, + 285, + 286, + 287, + 288, + 289, + 290, + 291, + 292, + 293, + 294, + 295, + 296, + 297, + 298, + 299, + 300, + 301, + 302, + 303, + 304, + 305, + 306, + 307, + 308, + 309, + 310, + 311, + 312, + 313, + 314, + 315, + 316, + 317, + 318, + 319, + 320, + 321, + 322, + 323, + 324, + 325, + 326, + 327, + 328, + 329, + 330, + 331, + 332, + 333, + 334, + 335, + 336, + 337, + 338, + 339, + 340, + 341, + 342, + 343, + 344, + 345, + 346, + 347, + 348, + 349, + 350, + 351, + 352, + 353, + 354, + 355, + 356, + 357, + 358, + 359, + 360, + 361, + 362, + 363, + 364, + 365, + 366, + 367, + 368, + 369, + 370, + 371, + 372, + 373, + 374, + 375, + 376, + 377, + 378, + 379, + 380, + 381, + 382, + 383, + 384, + 385, + 386, + 387, + 388, + 389, + 390, + 391, + 392, + 393, + 394, + 395, + 396, + 397, + 398, + 399, + 400, + 401, + 402, + 403, + 404, + 405, + 406, + 407, + 408, + 409, + 410, + 411, + 412, + 413, + 414, + 415, + 416, + 417, + 418, + 419, + 420, + 421, + 422, + 423, + 424, + 425, + 426, + 427, + 428, + 429, + 430, + 431, + 432, + 433, + 434, + 435, + 436, + 437, + 438, + 439, + 440, + 441, + 442, + 443, + 444, + 445, + 446, + 447, + 448, + 449, + 450, + 451, + 452, + 453, + 454, + 455, + 456, + 457, + 458, + 459, + 460, + 461, + 462, + 463, + 464, + 465, + 466, + 467, + 468, + 469, + 470, + 471, + 472, + 473, + 474, + 475, + 476, + 477, + 478, + 479, + 480, + 481, + 482, + 483, + 484, + 485, + 486, + 487, + 488, + 489, + 490, + 491, + 492, + 493, + 494, + 495, + 496, + 497, + 498, + 499, + 500, + 501, + 502, + 503, + 504, + 505, + 506, + 507, + 508, + 509, + 510, + 511, + 512, + 513, + 514, + 515, + 516, + 517, + 518, + 519, + 520, + 521, + 522, + 523, + 524, + 525, + 526, + 527, + 528, + 529, + 530, + 531, + 532, + 533, + 534, + 535, + 536, + 537, + 538, + 539, + 540, + 541, + 542, + 543, + 544, + 545, + 546, + 547, + 548, + 549, + 550, + 551, + 552, + 553, + 554, + 555, + 556, + 557, + 558, + 559, + 560, + 561, + 562, + 563, + 564, + 565, + 566, + 567, + 568, + 569, + 570, + 571, + 572, + 573, + 574, + 575, + 576, + 577, + 578, + 579, + 580, + 581, + 582, + 583, + 584, + 585, + 586, + 587, + 588, + 589, + 590, + 591, + 592, + 593, + 594, + 595, + 596, + 597, + 598, + 599, + 600, + 601, + 602, + 603, + 604, + 605, + 606, + 607, + 608, + 609, + 610, + 611, + 612, + 613, + 614, + 615, + 616, + 617, + 618, + 619, + 620, + 621, + 622, + 623, + 624, + 625, + 626, + 627, + 628, + 629, + 630, + 631, + 632, + 633, + 634, + 635, + 636, + 637, + 638, + 639, + 640, + 641, + 642, + 643, + 644, + 645, + 646, + 647, + 648, + 649, + 650, + 651, + 652, + 653, + 654, + 655, + 656, + 657, + 658, + 659, + 660, + 661, + 662, + 663, + 664, + 665, + 666, + 667, + 668, + 669, + 670, + 671, + 672, + 673, + 674, + 675, + 676, + 677, + 678, + 679, + 680, + 681, + 682, + 683, + 684, + 685, + 686, + 687, + 688, + 689, + 690, + 691, + 692, + 693, + 694, + 695, + 696, + 697, + 698, + 699, + 700, + 701, + 702, + 703, + 704, + 705, + 706, + 707, + 708, + 709, + 710, + 711, + 712, + 713, + 714, + 715, + 716, + 717, + 718, + 719, + 720, + 721, + 722, + 723, + 724, + 725, + 726, + 727, + 728, + 729, + 730, + 731, + 732, + 733, + 734, + 735, + 736, + 737, + 738, + 739, + 740, + 741, + 742, + 743, + 744, + 745, + 746, + 747, + 748, + 749, + 750, + 751, + 752, + 753, + 754, + 755, + 756, + 757, + 758, + 759, + 760, + 761, + 762, + 763, + 764, + 765, + 766, + 767, + 768, + 769, + 770, + 771, + 772, + 773, + 774, + 775, + 776, + 777, + 778, + 779, + 780, + 781, + 782, + 783, + 784, + 785, + 786, + 787, + 788, + 789, + 790, + 791, + 792, + 793, + 794, + 795, + 796, + 797, + 798, + 799, + 800, + 801, + 802, + 803, + 804, + 805, + 806, + 807, + 808, + 809, + 810, + 811, + 812, + 813, + 814, + 815, + 816, + 817, + 818, + 819, + 820, + 821, + 822, + 823, + 824, + 825, + 826, + 827, + 828, + 829, + 830, + 831, + 832, + 833, + 834, + 835, + 836, + 837, + 838, + 839, + 840, + 841, + 842, + 843, + 844, + 845, + 846, + 847, + 848, + 849, + 850, + 851, + 852, + 853, + 854, + 855, + 856, + 857, + 858, + 859 + ] + } +] \ No newline at end of file diff --git a/008000/training_state/optimizer_state.safetensors b/008000/training_state/optimizer_state.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..d508273fc8d1040161f581c747722ed41a2a5ef2 --- /dev/null +++ b/008000/training_state/optimizer_state.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:90d259e01d61199071e3916423c18dccc29dbfd4809ca5b729bf7e46a1615f3e +size 22197120 diff --git a/008000/training_state/rng_state.safetensors b/008000/training_state/rng_state.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..227ccc1227a00f057d915ed6594a370a198d3ce5 --- /dev/null +++ b/008000/training_state/rng_state.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:80b8295b5c931911da35294ad18512c32b07b849861612452a808d3a839d8d6d +size 15708 diff --git a/008000/training_state/scheduler_state.json b/008000/training_state/scheduler_state.json new file mode 100644 index 0000000000000000000000000000000000000000..86bec3ca145825fde5ad3000d10aed4649c64545 --- /dev/null +++ b/008000/training_state/scheduler_state.json @@ -0,0 +1,15 @@ +{ + "base_lrs": [ + 0.001 + ], + "last_epoch": 8000, + "_step_count": 8001, + "_is_initial": false, + "_get_lr_called_within_step": false, + "_last_lr": [ + 0.00018594235253127373 + ], + "lr_lambdas": [ + null + ] +} \ No newline at end of file diff --git a/008000/training_state/training_step.json b/008000/training_state/training_step.json new file mode 100644 index 0000000000000000000000000000000000000000..74d8cba01cab8506617b2cbae6f268fe80fbfa79 --- /dev/null +++ b/008000/training_state/training_step.json @@ -0,0 +1,3 @@ +{ + "step": 8000 +} \ No newline at end of file diff --git a/009000/pretrained_model/README.md b/009000/pretrained_model/README.md new file mode 100644 index 0000000000000000000000000000000000000000..6565338d7b6577ddbee192786d397d22079b8338 --- /dev/null +++ b/009000/pretrained_model/README.md @@ -0,0 +1,204 @@ +--- +library_name: peft +tags: +- base_model:adapter:lerobot/pi0_base +- lora +--- + +# Model Card for Model ID + + + + + +## Model Details + +### Model Description + + + + + +- **Developed by:** [More Information Needed] +- **Funded by [optional]:** [More Information Needed] +- **Shared by [optional]:** [More Information Needed] +- **Model type:** [More Information Needed] +- **Language(s) (NLP):** [More Information Needed] +- **License:** [More Information Needed] +- **Finetuned from model [optional]:** [More Information Needed] + +### Model Sources [optional] + + + +- **Repository:** [More Information Needed] +- **Paper [optional]:** [More Information Needed] +- **Demo [optional]:** [More Information Needed] + +## Uses + + + +### Direct Use + + + +[More Information Needed] + +### Downstream Use [optional] + + + +[More Information Needed] + +### Out-of-Scope Use + + + +[More Information Needed] + +## Bias, Risks, and Limitations + + + +[More Information Needed] + +### Recommendations + + + +Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations. + +## How to Get Started with the Model + +Use the code below to get started with the model. + +[More Information Needed] + +## Training Details + +### Training Data + + + +[More Information Needed] + +### Training Procedure + + + +#### Preprocessing [optional] + +[More Information Needed] + + +#### Training Hyperparameters + +- **Training regime:** [More Information Needed] + +#### Speeds, Sizes, Times [optional] + + + +[More Information Needed] + +## Evaluation + + + +### Testing Data, Factors & Metrics + +#### Testing Data + + + +[More Information Needed] + +#### Factors + + + +[More Information Needed] + +#### Metrics + + + +[More Information Needed] + +### Results + +[More Information Needed] + +#### Summary + + + +## Model Examination [optional] + + + +[More Information Needed] + +## Environmental Impact + + + +Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700). + +- **Hardware Type:** [More Information Needed] +- **Hours used:** [More Information Needed] +- **Cloud Provider:** [More Information Needed] +- **Compute Region:** [More Information Needed] +- **Carbon Emitted:** [More Information Needed] + +## Technical Specifications [optional] + +### Model Architecture and Objective + +[More Information Needed] + +### Compute Infrastructure + +[More Information Needed] + +#### Hardware + +[More Information Needed] + +#### Software + +[More Information Needed] + +## Citation [optional] + + + +**BibTeX:** + +[More Information Needed] + +**APA:** + +[More Information Needed] + +## Glossary [optional] + + + +[More Information Needed] + +## More Information [optional] + +[More Information Needed] + +## Model Card Authors [optional] + +[More Information Needed] + +## Model Card Contact + +[More Information Needed] +### Framework versions + +- PEFT 0.18.1 \ No newline at end of file diff --git a/009000/pretrained_model/adapter_config.json b/009000/pretrained_model/adapter_config.json new file mode 100644 index 0000000000000000000000000000000000000000..3c00acefa2527617e9a485e421d89b9166fed77b --- /dev/null +++ b/009000/pretrained_model/adapter_config.json @@ -0,0 +1,41 @@ +{ + "alora_invocation_tokens": null, + "alpha_pattern": {}, + "arrow_config": null, + "auto_mapping": { + "base_model_class": "PI0Policy", + "parent_library": "lerobot.policies.pi0.modeling_pi0" + }, + "base_model_name_or_path": "lerobot/pi0_base", + "bias": "none", + "corda_config": null, + "ensure_weight_tying": false, + "eva_config": null, + "exclude_modules": null, + "fan_in_fan_out": false, + "inference_mode": true, + "init_lora_weights": true, + "layer_replication": null, + "layers_pattern": null, + "layers_to_transform": null, + "loftq_config": {}, + "lora_alpha": 8, + "lora_bias": false, + "lora_dropout": 0.0, + "megatron_config": null, + "megatron_core": "megatron.core", + "modules_to_save": [], + "peft_type": "LORA", + "peft_version": "0.18.1", + "qalora_group_size": 16, + "r": 32, + "rank_pattern": {}, + "revision": null, + "target_modules": "(.*\\.gemma_expert\\..*\\.self_attn\\.(q|v)_proj|model\\.(state_proj|action_in_proj|action_out_proj|action_time_mlp_in|action_time_mlp_out))", + "target_parameters": null, + "task_type": null, + "trainable_token_indices": null, + "use_dora": false, + "use_qalora": false, + "use_rslora": false +} \ No newline at end of file diff --git a/009000/pretrained_model/adapter_model.safetensors b/009000/pretrained_model/adapter_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..a50b29314d510d83939dc61d073dde38fcca09a5 --- /dev/null +++ b/009000/pretrained_model/adapter_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:352a156fc0f7294fef7a79b8a874b7151a81b514e26f536de37867e87854bc46 +size 11101592 diff --git a/009000/pretrained_model/config.json b/009000/pretrained_model/config.json new file mode 100644 index 0000000000000000000000000000000000000000..5bded5dca90b3c498f402c92f09b4af980008111 --- /dev/null +++ b/009000/pretrained_model/config.json @@ -0,0 +1,113 @@ +{ + "type": "pi0", + "n_obs_steps": 1, + "input_features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "device": "cuda", + "use_amp": false, + "use_peft": true, + "push_to_hub": true, + "repo_id": "F-Fer/pi0-xlerobot-0", + "private": null, + "tags": null, + "license": null, + "pretrained_path": "lerobot/pi0_base", + "paligemma_variant": "gemma_2b", + "action_expert_variant": "gemma_300m", + "dtype": "bfloat16", + "chunk_size": 50, + "n_action_steps": 50, + "max_state_dim": 32, + "max_action_dim": 32, + "num_inference_steps": 10, + "time_sampling_beta_alpha": 1.5, + "time_sampling_beta_beta": 1.0, + "time_sampling_scale": 0.999, + "time_sampling_offset": 0.001, + "min_period": 0.004, + "max_period": 4.0, + "use_relative_actions": false, + "relative_exclude_joints": [ + "gripper" + ], + "action_feature_names": [ + "left_shoulder_pan.pos", + "left_shoulder_lift.pos", + "left_elbow_flex.pos", + "left_wrist_flex.pos", + "left_wrist_roll.pos", + "left_gripper.pos", + "right_shoulder_pan.pos", + "right_shoulder_lift.pos", + "right_elbow_flex.pos", + "right_wrist_flex.pos", + "right_wrist_roll.pos", + "right_gripper.pos" + ], + "rtc_config": null, + "image_resolution": [ + 224, + 224 + ], + "empty_cameras": 0, + "normalization_mapping": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + }, + "gradient_checkpointing": true, + "compile_model": true, + "compile_mode": "max-autotune", + "freeze_vision_encoder": false, + "train_expert_only": false, + "optimizer_lr": 0.001, + "optimizer_betas": [ + 0.9, + 0.95 + ], + "optimizer_eps": 1e-08, + "optimizer_weight_decay": 0.01, + "optimizer_grad_clip_norm": 1.0, + "scheduler_warmup_steps": 1000, + "scheduler_decay_steps": 30000, + "scheduler_decay_lr": 0.0001, + "tokenizer_max_length": 48 +} \ No newline at end of file diff --git a/009000/pretrained_model/policy_postprocessor.json b/009000/pretrained_model/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..25a2e623f8277a52e915c3701c7cd9cf77ddf8d6 --- /dev/null +++ b/009000/pretrained_model/policy_postprocessor.json @@ -0,0 +1,32 @@ +{ + "name": "policy_postprocessor", + "steps": [ + { + "registry_name": "unnormalizer_processor", + "config": { + "eps": 1e-08, + "features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "norm_map": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + } + }, + "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors" + }, + { + "registry_name": "device_processor", + "config": { + "device": "cpu", + "float_dtype": null + } + } + ] +} \ No newline at end of file diff --git a/009000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/009000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..08107a06a9226905f59e970a3d28ddc923a42695 --- /dev/null +++ b/009000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:794c084955a7613103705a47253d9fa3f66d03ead8978fa493cad135498ca62b +size 9192 diff --git a/009000/pretrained_model/policy_preprocessor.json b/009000/pretrained_model/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..8c39dc4710ff54b6f6d1631e34f0484383336eca --- /dev/null +++ b/009000/pretrained_model/policy_preprocessor.json @@ -0,0 +1,91 @@ +{ + "name": "policy_preprocessor", + "steps": [ + { + "registry_name": "rename_observations_processor", + "config": { + "rename_map": { + "observation.images.left_front": "observation.images.base_0_rgb", + "observation.images.left_left": "observation.images.left_wrist_0_rgb", + "observation.images.left_right": "observation.images.right_wrist_0_rgb" + } + } + }, + { + "registry_name": "to_batch_processor", + "config": {} + }, + { + "registry_name": "pi0_new_line_processor", + "config": {} + }, + { + "registry_name": "tokenizer_processor", + "config": { + "max_length": 48, + "task_key": "task", + "padding_side": "right", + "padding": "max_length", + "truncation": true, + "tokenizer_name": "google/paligemma-3b-pt-224" + } + }, + { + "registry_name": "device_processor", + "config": { + "device": "cuda", + "float_dtype": null + } + }, + { + "registry_name": "normalizer_processor", + "config": { + "eps": 1e-08, + "features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + }, + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "norm_map": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + } + }, + "state_file": "policy_preprocessor_step_5_normalizer_processor.safetensors" + } + ] +} \ No newline at end of file diff --git a/009000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors b/009000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..08107a06a9226905f59e970a3d28ddc923a42695 --- /dev/null +++ b/009000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:794c084955a7613103705a47253d9fa3f66d03ead8978fa493cad135498ca62b +size 9192 diff --git a/009000/pretrained_model/train_config.json b/009000/pretrained_model/train_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8b15c723555caefa3bdb9e5e9097937ebfef3fad --- /dev/null +++ b/009000/pretrained_model/train_config.json @@ -0,0 +1,261 @@ +{ + "dataset": { + "repo_id": "F-Fer/xlerobot-0", + "root": null, + "episodes": null, + "image_transforms": { + "enable": false, + "max_num_transforms": 3, + "random_order": false, + "tfs": { + "brightness": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "brightness": [ + 0.8, + 1.2 + ] + } + }, + "contrast": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "contrast": [ + 0.8, + 1.2 + ] + } + }, + "saturation": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "saturation": [ + 0.5, + 1.5 + ] + } + }, + "hue": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "hue": [ + -0.05, + 0.05 + ] + } + }, + "sharpness": { + "weight": 1.0, + "type": "SharpnessJitter", + "kwargs": { + "sharpness": [ + 0.5, + 1.5 + ] + } + }, + "affine": { + "weight": 1.0, + "type": "RandomAffine", + "kwargs": { + "degrees": [ + -5.0, + 5.0 + ], + "translate": [ + 0.05, + 0.05 + ] + } + } + } + }, + "revision": null, + "use_imagenet_stats": true, + "video_backend": "torchcodec", + "streaming": false + }, + "env": null, + "policy": { + "type": "pi0", + "n_obs_steps": 1, + "input_features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "device": "cuda", + "use_amp": false, + "use_peft": true, + "push_to_hub": true, + "repo_id": "F-Fer/pi0-xlerobot-0", + "private": null, + "tags": null, + "license": null, + "pretrained_path": "lerobot/pi0_base", + "paligemma_variant": "gemma_2b", + "action_expert_variant": "gemma_300m", + "dtype": "bfloat16", + "chunk_size": 50, + "n_action_steps": 50, + "max_state_dim": 32, + "max_action_dim": 32, + "num_inference_steps": 10, + "time_sampling_beta_alpha": 1.5, + "time_sampling_beta_beta": 1.0, + "time_sampling_scale": 0.999, + "time_sampling_offset": 0.001, + "min_period": 0.004, + "max_period": 4.0, + "use_relative_actions": false, + "relative_exclude_joints": [ + "gripper" + ], + "action_feature_names": [ + "left_shoulder_pan.pos", + "left_shoulder_lift.pos", + "left_elbow_flex.pos", + "left_wrist_flex.pos", + "left_wrist_roll.pos", + "left_gripper.pos", + "right_shoulder_pan.pos", + "right_shoulder_lift.pos", + "right_elbow_flex.pos", + "right_wrist_flex.pos", + "right_wrist_roll.pos", + "right_gripper.pos" + ], + "rtc_config": null, + "image_resolution": [ + 224, + 224 + ], + "empty_cameras": 0, + "normalization_mapping": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + }, + "gradient_checkpointing": true, + "compile_model": true, + "compile_mode": "max-autotune", + "freeze_vision_encoder": false, + "train_expert_only": false, + "optimizer_lr": 0.001, + "optimizer_betas": [ + 0.9, + 0.95 + ], + "optimizer_eps": 1e-08, + "optimizer_weight_decay": 0.01, + "optimizer_grad_clip_norm": 1.0, + "scheduler_warmup_steps": 1000, + "scheduler_decay_steps": 30000, + "scheduler_decay_lr": 0.0001, + "tokenizer_max_length": 48 + }, + "output_dir": "outputs/train/2026-04-10/14-19-24_pi0", + "job_name": "pi0", + "resume": false, + "seed": 1000, + "cudnn_deterministic": false, + "num_workers": 4, + "batch_size": 4, + "steps": 10000, + "eval_freq": -1, + "log_freq": 100, + "tolerance_s": 0.0001, + "save_checkpoint": true, + "save_freq": 1000, + "use_policy_training_preset": true, + "optimizer": { + "type": "adamw", + "lr": 0.001, + "weight_decay": 0.01, + "grad_clip_norm": 1.0, + "betas": [ + 0.9, + 0.95 + ], + "eps": 1e-08 + }, + "scheduler": { + "type": "cosine_decay_with_warmup", + "num_warmup_steps": 1000, + "num_decay_steps": 30000, + "peak_lr": 0.001, + "decay_lr": 0.0001 + }, + "eval": { + "n_episodes": 50, + "batch_size": 50, + "use_async_envs": false + }, + "wandb": { + "enable": true, + "disable_artifact": false, + "project": "xlerobot", + "entity": null, + "notes": null, + "run_id": "tmqkc0d7", + "mode": null, + "add_tags": true + }, + "peft": { + "target_modules": null, + "full_training_modules": null, + "method_type": "LORA", + "init_type": null, + "r": 32 + }, + "use_rabc": false, + "rabc_progress_path": null, + "rabc_kappa": 0.01, + "rabc_epsilon": 1e-06, + "rabc_head_mode": "sparse", + "rename_map": { + "observation.images.left_front": "observation.images.base_0_rgb", + "observation.images.left_left": "observation.images.left_wrist_0_rgb", + "observation.images.left_right": "observation.images.right_wrist_0_rgb" + }, + "checkpoint_path": null +} \ No newline at end of file diff --git a/009000/training_state/optimizer_param_groups.json b/009000/training_state/optimizer_param_groups.json new file mode 100644 index 0000000000000000000000000000000000000000..6a8992cc6cf0420a1da52da1b89c66524affe053 --- /dev/null +++ b/009000/training_state/optimizer_param_groups.json @@ -0,0 +1,881 @@ +[ + { + "lr": 0.00012202456766718093, + "betas": [ + 0.9, + 0.95 + ], + "eps": 1e-08, + "weight_decay": 0.01, + "amsgrad": false, + "maximize": false, + "foreach": null, + "capturable": false, + "differentiable": false, + "fused": null, + "decoupled_weight_decay": true, + "initial_lr": 0.001, + "params": [ + 0, + 1, + 2, + 3, + 4, + 5, + 6, + 7, + 8, + 9, + 10, + 11, + 12, + 13, + 14, + 15, + 16, + 17, + 18, + 19, + 20, + 21, + 22, + 23, + 24, + 25, + 26, + 27, + 28, + 29, + 30, + 31, + 32, + 33, + 34, + 35, + 36, + 37, + 38, + 39, + 40, + 41, + 42, + 43, + 44, + 45, + 46, + 47, + 48, + 49, + 50, + 51, + 52, + 53, + 54, + 55, + 56, + 57, + 58, + 59, + 60, + 61, + 62, + 63, + 64, + 65, + 66, + 67, + 68, + 69, + 70, + 71, + 72, + 73, + 74, + 75, + 76, + 77, + 78, + 79, + 80, + 81, + 82, + 83, + 84, + 85, + 86, + 87, + 88, + 89, + 90, + 91, + 92, + 93, + 94, + 95, + 96, + 97, + 98, + 99, + 100, + 101, + 102, + 103, + 104, + 105, + 106, + 107, + 108, + 109, + 110, + 111, + 112, + 113, + 114, + 115, + 116, + 117, + 118, + 119, + 120, + 121, + 122, + 123, + 124, + 125, + 126, + 127, + 128, + 129, + 130, + 131, + 132, + 133, + 134, + 135, + 136, + 137, + 138, + 139, + 140, + 141, + 142, + 143, + 144, + 145, + 146, + 147, + 148, + 149, + 150, + 151, + 152, + 153, + 154, + 155, + 156, + 157, + 158, + 159, + 160, + 161, + 162, + 163, + 164, + 165, + 166, + 167, + 168, + 169, + 170, + 171, + 172, + 173, + 174, + 175, + 176, + 177, + 178, + 179, + 180, + 181, + 182, + 183, + 184, + 185, + 186, + 187, + 188, + 189, + 190, + 191, + 192, + 193, + 194, + 195, + 196, + 197, + 198, + 199, + 200, + 201, + 202, + 203, + 204, + 205, + 206, + 207, + 208, + 209, + 210, + 211, + 212, + 213, + 214, + 215, + 216, + 217, + 218, + 219, + 220, + 221, + 222, + 223, + 224, + 225, + 226, + 227, + 228, + 229, + 230, + 231, + 232, + 233, + 234, + 235, + 236, + 237, + 238, + 239, + 240, + 241, + 242, + 243, + 244, + 245, + 246, + 247, + 248, + 249, + 250, + 251, + 252, + 253, + 254, + 255, + 256, + 257, + 258, + 259, + 260, + 261, + 262, + 263, + 264, + 265, + 266, + 267, + 268, + 269, + 270, + 271, + 272, + 273, + 274, + 275, + 276, + 277, + 278, + 279, + 280, + 281, + 282, + 283, + 284, + 285, + 286, + 287, + 288, + 289, + 290, + 291, + 292, + 293, + 294, + 295, + 296, + 297, + 298, + 299, + 300, + 301, + 302, + 303, + 304, + 305, + 306, + 307, + 308, + 309, + 310, + 311, + 312, + 313, + 314, + 315, + 316, + 317, + 318, + 319, + 320, + 321, + 322, + 323, + 324, + 325, + 326, + 327, + 328, + 329, + 330, + 331, + 332, + 333, + 334, + 335, + 336, + 337, + 338, + 339, + 340, + 341, + 342, + 343, + 344, + 345, + 346, + 347, + 348, + 349, + 350, + 351, + 352, + 353, + 354, + 355, + 356, + 357, + 358, + 359, + 360, + 361, + 362, + 363, + 364, + 365, + 366, + 367, + 368, + 369, + 370, + 371, + 372, + 373, + 374, + 375, + 376, + 377, + 378, + 379, + 380, + 381, + 382, + 383, + 384, + 385, + 386, + 387, + 388, + 389, + 390, + 391, + 392, + 393, + 394, + 395, + 396, + 397, + 398, + 399, + 400, + 401, + 402, + 403, + 404, + 405, + 406, + 407, + 408, + 409, + 410, + 411, + 412, + 413, + 414, + 415, + 416, + 417, + 418, + 419, + 420, + 421, + 422, + 423, + 424, + 425, + 426, + 427, + 428, + 429, + 430, + 431, + 432, + 433, + 434, + 435, + 436, + 437, + 438, + 439, + 440, + 441, + 442, + 443, + 444, + 445, + 446, + 447, + 448, + 449, + 450, + 451, + 452, + 453, + 454, + 455, + 456, + 457, + 458, + 459, + 460, + 461, + 462, + 463, + 464, + 465, + 466, + 467, + 468, + 469, + 470, + 471, + 472, + 473, + 474, + 475, + 476, + 477, + 478, + 479, + 480, + 481, + 482, + 483, + 484, + 485, + 486, + 487, + 488, + 489, + 490, + 491, + 492, + 493, + 494, + 495, + 496, + 497, + 498, + 499, + 500, + 501, + 502, + 503, + 504, + 505, + 506, + 507, + 508, + 509, + 510, + 511, + 512, + 513, + 514, + 515, + 516, + 517, + 518, + 519, + 520, + 521, + 522, + 523, + 524, + 525, + 526, + 527, + 528, + 529, + 530, + 531, + 532, + 533, + 534, + 535, + 536, + 537, + 538, + 539, + 540, + 541, + 542, + 543, + 544, + 545, + 546, + 547, + 548, + 549, + 550, + 551, + 552, + 553, + 554, + 555, + 556, + 557, + 558, + 559, + 560, + 561, + 562, + 563, + 564, + 565, + 566, + 567, + 568, + 569, + 570, + 571, + 572, + 573, + 574, + 575, + 576, + 577, + 578, + 579, + 580, + 581, + 582, + 583, + 584, + 585, + 586, + 587, + 588, + 589, + 590, + 591, + 592, + 593, + 594, + 595, + 596, + 597, + 598, + 599, + 600, + 601, + 602, + 603, + 604, + 605, + 606, + 607, + 608, + 609, + 610, + 611, + 612, + 613, + 614, + 615, + 616, + 617, + 618, + 619, + 620, + 621, + 622, + 623, + 624, + 625, + 626, + 627, + 628, + 629, + 630, + 631, + 632, + 633, + 634, + 635, + 636, + 637, + 638, + 639, + 640, + 641, + 642, + 643, + 644, + 645, + 646, + 647, + 648, + 649, + 650, + 651, + 652, + 653, + 654, + 655, + 656, + 657, + 658, + 659, + 660, + 661, + 662, + 663, + 664, + 665, + 666, + 667, + 668, + 669, + 670, + 671, + 672, + 673, + 674, + 675, + 676, + 677, + 678, + 679, + 680, + 681, + 682, + 683, + 684, + 685, + 686, + 687, + 688, + 689, + 690, + 691, + 692, + 693, + 694, + 695, + 696, + 697, + 698, + 699, + 700, + 701, + 702, + 703, + 704, + 705, + 706, + 707, + 708, + 709, + 710, + 711, + 712, + 713, + 714, + 715, + 716, + 717, + 718, + 719, + 720, + 721, + 722, + 723, + 724, + 725, + 726, + 727, + 728, + 729, + 730, + 731, + 732, + 733, + 734, + 735, + 736, + 737, + 738, + 739, + 740, + 741, + 742, + 743, + 744, + 745, + 746, + 747, + 748, + 749, + 750, + 751, + 752, + 753, + 754, + 755, + 756, + 757, + 758, + 759, + 760, + 761, + 762, + 763, + 764, + 765, + 766, + 767, + 768, + 769, + 770, + 771, + 772, + 773, + 774, + 775, + 776, + 777, + 778, + 779, + 780, + 781, + 782, + 783, + 784, + 785, + 786, + 787, + 788, + 789, + 790, + 791, + 792, + 793, + 794, + 795, + 796, + 797, + 798, + 799, + 800, + 801, + 802, + 803, + 804, + 805, + 806, + 807, + 808, + 809, + 810, + 811, + 812, + 813, + 814, + 815, + 816, + 817, + 818, + 819, + 820, + 821, + 822, + 823, + 824, + 825, + 826, + 827, + 828, + 829, + 830, + 831, + 832, + 833, + 834, + 835, + 836, + 837, + 838, + 839, + 840, + 841, + 842, + 843, + 844, + 845, + 846, + 847, + 848, + 849, + 850, + 851, + 852, + 853, + 854, + 855, + 856, + 857, + 858, + 859 + ] + } +] \ No newline at end of file diff --git a/009000/training_state/optimizer_state.safetensors b/009000/training_state/optimizer_state.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..483ea64c3a784aa2031d13b3b6f645169abad46b --- /dev/null +++ b/009000/training_state/optimizer_state.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:68f737365682afdf1646fa35f3b52e239e42896d441735dcd25b7110d7476964 +size 22197120 diff --git a/009000/training_state/rng_state.safetensors b/009000/training_state/rng_state.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..9ff1f68d523fbfc64a2d0224e2e7618aa75b4adb --- /dev/null +++ b/009000/training_state/rng_state.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a2de68a54a179eac357ae6abeb10fcfccda441a9431f6bc0256a1bcaf27b76d9 +size 15708 diff --git a/009000/training_state/scheduler_state.json b/009000/training_state/scheduler_state.json new file mode 100644 index 0000000000000000000000000000000000000000..99a54280745a4841b2b6d64975155cd9b95fa825 --- /dev/null +++ b/009000/training_state/scheduler_state.json @@ -0,0 +1,15 @@ +{ + "base_lrs": [ + 0.001 + ], + "last_epoch": 9000, + "_step_count": 9001, + "_is_initial": false, + "_get_lr_called_within_step": false, + "_last_lr": [ + 0.00012202456766718093 + ], + "lr_lambdas": [ + null + ] +} \ No newline at end of file diff --git a/009000/training_state/training_step.json b/009000/training_state/training_step.json new file mode 100644 index 0000000000000000000000000000000000000000..7247596175281e9d3f295d2b6936944a71b64d34 --- /dev/null +++ b/009000/training_state/training_step.json @@ -0,0 +1,3 @@ +{ + "step": 9000 +} \ No newline at end of file diff --git a/010000/pretrained_model/README.md b/010000/pretrained_model/README.md new file mode 100644 index 0000000000000000000000000000000000000000..6565338d7b6577ddbee192786d397d22079b8338 --- /dev/null +++ b/010000/pretrained_model/README.md @@ -0,0 +1,204 @@ +--- +library_name: peft +tags: +- base_model:adapter:lerobot/pi0_base +- lora +--- + +# Model Card for Model ID + + + + + +## Model Details + +### Model Description + + + + + +- **Developed by:** [More Information Needed] +- **Funded by [optional]:** [More Information Needed] +- **Shared by [optional]:** [More Information Needed] +- **Model type:** [More Information Needed] +- **Language(s) (NLP):** [More Information Needed] +- **License:** [More Information Needed] +- **Finetuned from model [optional]:** [More Information Needed] + +### Model Sources [optional] + + + +- **Repository:** [More Information Needed] +- **Paper [optional]:** [More Information Needed] +- **Demo [optional]:** [More Information Needed] + +## Uses + + + +### Direct Use + + + +[More Information Needed] + +### Downstream Use [optional] + + + +[More Information Needed] + +### Out-of-Scope Use + + + +[More Information Needed] + +## Bias, Risks, and Limitations + + + +[More Information Needed] + +### Recommendations + + + +Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations. + +## How to Get Started with the Model + +Use the code below to get started with the model. + +[More Information Needed] + +## Training Details + +### Training Data + + + +[More Information Needed] + +### Training Procedure + + + +#### Preprocessing [optional] + +[More Information Needed] + + +#### Training Hyperparameters + +- **Training regime:** [More Information Needed] + +#### Speeds, Sizes, Times [optional] + + + +[More Information Needed] + +## Evaluation + + + +### Testing Data, Factors & Metrics + +#### Testing Data + + + +[More Information Needed] + +#### Factors + + + +[More Information Needed] + +#### Metrics + + + +[More Information Needed] + +### Results + +[More Information Needed] + +#### Summary + + + +## Model Examination [optional] + + + +[More Information Needed] + +## Environmental Impact + + + +Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700). + +- **Hardware Type:** [More Information Needed] +- **Hours used:** [More Information Needed] +- **Cloud Provider:** [More Information Needed] +- **Compute Region:** [More Information Needed] +- **Carbon Emitted:** [More Information Needed] + +## Technical Specifications [optional] + +### Model Architecture and Objective + +[More Information Needed] + +### Compute Infrastructure + +[More Information Needed] + +#### Hardware + +[More Information Needed] + +#### Software + +[More Information Needed] + +## Citation [optional] + + + +**BibTeX:** + +[More Information Needed] + +**APA:** + +[More Information Needed] + +## Glossary [optional] + + + +[More Information Needed] + +## More Information [optional] + +[More Information Needed] + +## Model Card Authors [optional] + +[More Information Needed] + +## Model Card Contact + +[More Information Needed] +### Framework versions + +- PEFT 0.18.1 \ No newline at end of file diff --git a/010000/pretrained_model/adapter_config.json b/010000/pretrained_model/adapter_config.json new file mode 100644 index 0000000000000000000000000000000000000000..3c00acefa2527617e9a485e421d89b9166fed77b --- /dev/null +++ b/010000/pretrained_model/adapter_config.json @@ -0,0 +1,41 @@ +{ + "alora_invocation_tokens": null, + "alpha_pattern": {}, + "arrow_config": null, + "auto_mapping": { + "base_model_class": "PI0Policy", + "parent_library": "lerobot.policies.pi0.modeling_pi0" + }, + "base_model_name_or_path": "lerobot/pi0_base", + "bias": "none", + "corda_config": null, + "ensure_weight_tying": false, + "eva_config": null, + "exclude_modules": null, + "fan_in_fan_out": false, + "inference_mode": true, + "init_lora_weights": true, + "layer_replication": null, + "layers_pattern": null, + "layers_to_transform": null, + "loftq_config": {}, + "lora_alpha": 8, + "lora_bias": false, + "lora_dropout": 0.0, + "megatron_config": null, + "megatron_core": "megatron.core", + "modules_to_save": [], + "peft_type": "LORA", + "peft_version": "0.18.1", + "qalora_group_size": 16, + "r": 32, + "rank_pattern": {}, + "revision": null, + "target_modules": "(.*\\.gemma_expert\\..*\\.self_attn\\.(q|v)_proj|model\\.(state_proj|action_in_proj|action_out_proj|action_time_mlp_in|action_time_mlp_out))", + "target_parameters": null, + "task_type": null, + "trainable_token_indices": null, + "use_dora": false, + "use_qalora": false, + "use_rslora": false +} \ No newline at end of file diff --git a/010000/pretrained_model/adapter_model.safetensors b/010000/pretrained_model/adapter_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..46a657f7b00ab79fcc971743389c304b5fe7911e --- /dev/null +++ b/010000/pretrained_model/adapter_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:92df8bd89d2f1d46e3db6c1e82f755af708d118c8b547b852d6a343f8f5870f8 +size 11101592 diff --git a/010000/pretrained_model/config.json b/010000/pretrained_model/config.json new file mode 100644 index 0000000000000000000000000000000000000000..5bded5dca90b3c498f402c92f09b4af980008111 --- /dev/null +++ b/010000/pretrained_model/config.json @@ -0,0 +1,113 @@ +{ + "type": "pi0", + "n_obs_steps": 1, + "input_features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "device": "cuda", + "use_amp": false, + "use_peft": true, + "push_to_hub": true, + "repo_id": "F-Fer/pi0-xlerobot-0", + "private": null, + "tags": null, + "license": null, + "pretrained_path": "lerobot/pi0_base", + "paligemma_variant": "gemma_2b", + "action_expert_variant": "gemma_300m", + "dtype": "bfloat16", + "chunk_size": 50, + "n_action_steps": 50, + "max_state_dim": 32, + "max_action_dim": 32, + "num_inference_steps": 10, + "time_sampling_beta_alpha": 1.5, + "time_sampling_beta_beta": 1.0, + "time_sampling_scale": 0.999, + "time_sampling_offset": 0.001, + "min_period": 0.004, + "max_period": 4.0, + "use_relative_actions": false, + "relative_exclude_joints": [ + "gripper" + ], + "action_feature_names": [ + "left_shoulder_pan.pos", + "left_shoulder_lift.pos", + "left_elbow_flex.pos", + "left_wrist_flex.pos", + "left_wrist_roll.pos", + "left_gripper.pos", + "right_shoulder_pan.pos", + "right_shoulder_lift.pos", + "right_elbow_flex.pos", + "right_wrist_flex.pos", + "right_wrist_roll.pos", + "right_gripper.pos" + ], + "rtc_config": null, + "image_resolution": [ + 224, + 224 + ], + "empty_cameras": 0, + "normalization_mapping": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + }, + "gradient_checkpointing": true, + "compile_model": true, + "compile_mode": "max-autotune", + "freeze_vision_encoder": false, + "train_expert_only": false, + "optimizer_lr": 0.001, + "optimizer_betas": [ + 0.9, + 0.95 + ], + "optimizer_eps": 1e-08, + "optimizer_weight_decay": 0.01, + "optimizer_grad_clip_norm": 1.0, + "scheduler_warmup_steps": 1000, + "scheduler_decay_steps": 30000, + "scheduler_decay_lr": 0.0001, + "tokenizer_max_length": 48 +} \ No newline at end of file diff --git a/010000/pretrained_model/policy_postprocessor.json b/010000/pretrained_model/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..25a2e623f8277a52e915c3701c7cd9cf77ddf8d6 --- /dev/null +++ b/010000/pretrained_model/policy_postprocessor.json @@ -0,0 +1,32 @@ +{ + "name": "policy_postprocessor", + "steps": [ + { + "registry_name": "unnormalizer_processor", + "config": { + "eps": 1e-08, + "features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "norm_map": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + } + }, + "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors" + }, + { + "registry_name": "device_processor", + "config": { + "device": "cpu", + "float_dtype": null + } + } + ] +} \ No newline at end of file diff --git a/010000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/010000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..08107a06a9226905f59e970a3d28ddc923a42695 --- /dev/null +++ b/010000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:794c084955a7613103705a47253d9fa3f66d03ead8978fa493cad135498ca62b +size 9192 diff --git a/010000/pretrained_model/policy_preprocessor.json b/010000/pretrained_model/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..8c39dc4710ff54b6f6d1631e34f0484383336eca --- /dev/null +++ b/010000/pretrained_model/policy_preprocessor.json @@ -0,0 +1,91 @@ +{ + "name": "policy_preprocessor", + "steps": [ + { + "registry_name": "rename_observations_processor", + "config": { + "rename_map": { + "observation.images.left_front": "observation.images.base_0_rgb", + "observation.images.left_left": "observation.images.left_wrist_0_rgb", + "observation.images.left_right": "observation.images.right_wrist_0_rgb" + } + } + }, + { + "registry_name": "to_batch_processor", + "config": {} + }, + { + "registry_name": "pi0_new_line_processor", + "config": {} + }, + { + "registry_name": "tokenizer_processor", + "config": { + "max_length": 48, + "task_key": "task", + "padding_side": "right", + "padding": "max_length", + "truncation": true, + "tokenizer_name": "google/paligemma-3b-pt-224" + } + }, + { + "registry_name": "device_processor", + "config": { + "device": "cuda", + "float_dtype": null + } + }, + { + "registry_name": "normalizer_processor", + "config": { + "eps": 1e-08, + "features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + }, + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "norm_map": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + } + }, + "state_file": "policy_preprocessor_step_5_normalizer_processor.safetensors" + } + ] +} \ No newline at end of file diff --git a/010000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors b/010000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..08107a06a9226905f59e970a3d28ddc923a42695 --- /dev/null +++ b/010000/pretrained_model/policy_preprocessor_step_5_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:794c084955a7613103705a47253d9fa3f66d03ead8978fa493cad135498ca62b +size 9192 diff --git a/010000/pretrained_model/train_config.json b/010000/pretrained_model/train_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8b15c723555caefa3bdb9e5e9097937ebfef3fad --- /dev/null +++ b/010000/pretrained_model/train_config.json @@ -0,0 +1,261 @@ +{ + "dataset": { + "repo_id": "F-Fer/xlerobot-0", + "root": null, + "episodes": null, + "image_transforms": { + "enable": false, + "max_num_transforms": 3, + "random_order": false, + "tfs": { + "brightness": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "brightness": [ + 0.8, + 1.2 + ] + } + }, + "contrast": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "contrast": [ + 0.8, + 1.2 + ] + } + }, + "saturation": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "saturation": [ + 0.5, + 1.5 + ] + } + }, + "hue": { + "weight": 1.0, + "type": "ColorJitter", + "kwargs": { + "hue": [ + -0.05, + 0.05 + ] + } + }, + "sharpness": { + "weight": 1.0, + "type": "SharpnessJitter", + "kwargs": { + "sharpness": [ + 0.5, + 1.5 + ] + } + }, + "affine": { + "weight": 1.0, + "type": "RandomAffine", + "kwargs": { + "degrees": [ + -5.0, + 5.0 + ], + "translate": [ + 0.05, + 0.05 + ] + } + } + } + }, + "revision": null, + "use_imagenet_stats": true, + "video_backend": "torchcodec", + "streaming": false + }, + "env": null, + "policy": { + "type": "pi0", + "n_obs_steps": 1, + "input_features": { + "observation.images.base_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.left_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.images.right_wrist_0_rgb": { + "type": "VISUAL", + "shape": [ + 3, + 224, + 224 + ] + }, + "observation.state": { + "type": "STATE", + "shape": [ + 32 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 12 + ] + } + }, + "device": "cuda", + "use_amp": false, + "use_peft": true, + "push_to_hub": true, + "repo_id": "F-Fer/pi0-xlerobot-0", + "private": null, + "tags": null, + "license": null, + "pretrained_path": "lerobot/pi0_base", + "paligemma_variant": "gemma_2b", + "action_expert_variant": "gemma_300m", + "dtype": "bfloat16", + "chunk_size": 50, + "n_action_steps": 50, + "max_state_dim": 32, + "max_action_dim": 32, + "num_inference_steps": 10, + "time_sampling_beta_alpha": 1.5, + "time_sampling_beta_beta": 1.0, + "time_sampling_scale": 0.999, + "time_sampling_offset": 0.001, + "min_period": 0.004, + "max_period": 4.0, + "use_relative_actions": false, + "relative_exclude_joints": [ + "gripper" + ], + "action_feature_names": [ + "left_shoulder_pan.pos", + "left_shoulder_lift.pos", + "left_elbow_flex.pos", + "left_wrist_flex.pos", + "left_wrist_roll.pos", + "left_gripper.pos", + "right_shoulder_pan.pos", + "right_shoulder_lift.pos", + "right_elbow_flex.pos", + "right_wrist_flex.pos", + "right_wrist_roll.pos", + "right_gripper.pos" + ], + "rtc_config": null, + "image_resolution": [ + 224, + 224 + ], + "empty_cameras": 0, + "normalization_mapping": { + "VISUAL": "IDENTITY", + "STATE": "MEAN_STD", + "ACTION": "MEAN_STD" + }, + "gradient_checkpointing": true, + "compile_model": true, + "compile_mode": "max-autotune", + "freeze_vision_encoder": false, + "train_expert_only": false, + "optimizer_lr": 0.001, + "optimizer_betas": [ + 0.9, + 0.95 + ], + "optimizer_eps": 1e-08, + "optimizer_weight_decay": 0.01, + "optimizer_grad_clip_norm": 1.0, + "scheduler_warmup_steps": 1000, + "scheduler_decay_steps": 30000, + "scheduler_decay_lr": 0.0001, + "tokenizer_max_length": 48 + }, + "output_dir": "outputs/train/2026-04-10/14-19-24_pi0", + "job_name": "pi0", + "resume": false, + "seed": 1000, + "cudnn_deterministic": false, + "num_workers": 4, + "batch_size": 4, + "steps": 10000, + "eval_freq": -1, + "log_freq": 100, + "tolerance_s": 0.0001, + "save_checkpoint": true, + "save_freq": 1000, + "use_policy_training_preset": true, + "optimizer": { + "type": "adamw", + "lr": 0.001, + "weight_decay": 0.01, + "grad_clip_norm": 1.0, + "betas": [ + 0.9, + 0.95 + ], + "eps": 1e-08 + }, + "scheduler": { + "type": "cosine_decay_with_warmup", + "num_warmup_steps": 1000, + "num_decay_steps": 30000, + "peak_lr": 0.001, + "decay_lr": 0.0001 + }, + "eval": { + "n_episodes": 50, + "batch_size": 50, + "use_async_envs": false + }, + "wandb": { + "enable": true, + "disable_artifact": false, + "project": "xlerobot", + "entity": null, + "notes": null, + "run_id": "tmqkc0d7", + "mode": null, + "add_tags": true + }, + "peft": { + "target_modules": null, + "full_training_modules": null, + "method_type": "LORA", + "init_type": null, + "r": 32 + }, + "use_rabc": false, + "rabc_progress_path": null, + "rabc_kappa": 0.01, + "rabc_epsilon": 1e-06, + "rabc_head_mode": "sparse", + "rename_map": { + "observation.images.left_front": "observation.images.base_0_rgb", + "observation.images.left_left": "observation.images.left_wrist_0_rgb", + "observation.images.left_right": "observation.images.right_wrist_0_rgb" + }, + "checkpoint_path": null +} \ No newline at end of file diff --git a/010000/training_state/optimizer_param_groups.json b/010000/training_state/optimizer_param_groups.json new file mode 100644 index 0000000000000000000000000000000000000000..9634e1ba917ba49ba4f7ed154c00fd2fc762e4f2 --- /dev/null +++ b/010000/training_state/optimizer_param_groups.json @@ -0,0 +1,881 @@ +[ + { + "lr": 0.0001, + "betas": [ + 0.9, + 0.95 + ], + "eps": 1e-08, + "weight_decay": 0.01, + "amsgrad": false, + "maximize": false, + "foreach": null, + "capturable": false, + "differentiable": false, + "fused": null, + "decoupled_weight_decay": true, + "initial_lr": 0.001, + "params": [ + 0, + 1, + 2, + 3, + 4, + 5, + 6, + 7, + 8, + 9, + 10, + 11, + 12, + 13, + 14, + 15, + 16, + 17, + 18, + 19, + 20, + 21, + 22, + 23, + 24, + 25, + 26, + 27, + 28, + 29, + 30, + 31, + 32, + 33, + 34, + 35, + 36, + 37, + 38, + 39, + 40, + 41, + 42, + 43, + 44, + 45, + 46, + 47, + 48, + 49, + 50, + 51, + 52, + 53, + 54, + 55, + 56, + 57, + 58, + 59, + 60, + 61, + 62, + 63, + 64, + 65, + 66, + 67, + 68, + 69, + 70, + 71, + 72, + 73, + 74, + 75, + 76, + 77, + 78, + 79, + 80, + 81, + 82, + 83, + 84, + 85, + 86, + 87, + 88, + 89, + 90, + 91, + 92, + 93, + 94, + 95, + 96, + 97, + 98, + 99, + 100, + 101, + 102, + 103, + 104, + 105, + 106, + 107, + 108, + 109, + 110, + 111, + 112, + 113, + 114, + 115, + 116, + 117, + 118, + 119, + 120, + 121, + 122, + 123, + 124, + 125, + 126, + 127, + 128, + 129, + 130, + 131, + 132, + 133, + 134, + 135, + 136, + 137, + 138, + 139, + 140, + 141, + 142, + 143, + 144, + 145, + 146, + 147, + 148, + 149, + 150, + 151, + 152, + 153, + 154, + 155, + 156, + 157, + 158, + 159, + 160, + 161, + 162, + 163, + 164, + 165, + 166, + 167, + 168, + 169, + 170, + 171, + 172, + 173, + 174, + 175, + 176, + 177, + 178, + 179, + 180, + 181, + 182, + 183, + 184, + 185, + 186, + 187, + 188, + 189, + 190, + 191, + 192, + 193, + 194, + 195, + 196, + 197, + 198, + 199, + 200, + 201, + 202, + 203, + 204, + 205, + 206, + 207, + 208, + 209, + 210, + 211, + 212, + 213, + 214, + 215, + 216, + 217, + 218, + 219, + 220, + 221, + 222, + 223, + 224, + 225, + 226, + 227, + 228, + 229, + 230, + 231, + 232, + 233, + 234, + 235, + 236, + 237, + 238, + 239, + 240, + 241, + 242, + 243, + 244, + 245, + 246, + 247, + 248, + 249, + 250, + 251, + 252, + 253, + 254, + 255, + 256, + 257, + 258, + 259, + 260, + 261, + 262, + 263, + 264, + 265, + 266, + 267, + 268, + 269, + 270, + 271, + 272, + 273, + 274, + 275, + 276, + 277, + 278, + 279, + 280, + 281, + 282, + 283, + 284, + 285, + 286, + 287, + 288, + 289, + 290, + 291, + 292, + 293, + 294, + 295, + 296, + 297, + 298, + 299, + 300, + 301, + 302, + 303, + 304, + 305, + 306, + 307, + 308, + 309, + 310, + 311, + 312, + 313, + 314, + 315, + 316, + 317, + 318, + 319, + 320, + 321, + 322, + 323, + 324, + 325, + 326, + 327, + 328, + 329, + 330, + 331, + 332, + 333, + 334, + 335, + 336, + 337, + 338, + 339, + 340, + 341, + 342, + 343, + 344, + 345, + 346, + 347, + 348, + 349, + 350, + 351, + 352, + 353, + 354, + 355, + 356, + 357, + 358, + 359, + 360, + 361, + 362, + 363, + 364, + 365, + 366, + 367, + 368, + 369, + 370, + 371, + 372, + 373, + 374, + 375, + 376, + 377, + 378, + 379, + 380, + 381, + 382, + 383, + 384, + 385, + 386, + 387, + 388, + 389, + 390, + 391, + 392, + 393, + 394, + 395, + 396, + 397, + 398, + 399, + 400, + 401, + 402, + 403, + 404, + 405, + 406, + 407, + 408, + 409, + 410, + 411, + 412, + 413, + 414, + 415, + 416, + 417, + 418, + 419, + 420, + 421, + 422, + 423, + 424, + 425, + 426, + 427, + 428, + 429, + 430, + 431, + 432, + 433, + 434, + 435, + 436, + 437, + 438, + 439, + 440, + 441, + 442, + 443, + 444, + 445, + 446, + 447, + 448, + 449, + 450, + 451, + 452, + 453, + 454, + 455, + 456, + 457, + 458, + 459, + 460, + 461, + 462, + 463, + 464, + 465, + 466, + 467, + 468, + 469, + 470, + 471, + 472, + 473, + 474, + 475, + 476, + 477, + 478, + 479, + 480, + 481, + 482, + 483, + 484, + 485, + 486, + 487, + 488, + 489, + 490, + 491, + 492, + 493, + 494, + 495, + 496, + 497, + 498, + 499, + 500, + 501, + 502, + 503, + 504, + 505, + 506, + 507, + 508, + 509, + 510, + 511, + 512, + 513, + 514, + 515, + 516, + 517, + 518, + 519, + 520, + 521, + 522, + 523, + 524, + 525, + 526, + 527, + 528, + 529, + 530, + 531, + 532, + 533, + 534, + 535, + 536, + 537, + 538, + 539, + 540, + 541, + 542, + 543, + 544, + 545, + 546, + 547, + 548, + 549, + 550, + 551, + 552, + 553, + 554, + 555, + 556, + 557, + 558, + 559, + 560, + 561, + 562, + 563, + 564, + 565, + 566, + 567, + 568, + 569, + 570, + 571, + 572, + 573, + 574, + 575, + 576, + 577, + 578, + 579, + 580, + 581, + 582, + 583, + 584, + 585, + 586, + 587, + 588, + 589, + 590, + 591, + 592, + 593, + 594, + 595, + 596, + 597, + 598, + 599, + 600, + 601, + 602, + 603, + 604, + 605, + 606, + 607, + 608, + 609, + 610, + 611, + 612, + 613, + 614, + 615, + 616, + 617, + 618, + 619, + 620, + 621, + 622, + 623, + 624, + 625, + 626, + 627, + 628, + 629, + 630, + 631, + 632, + 633, + 634, + 635, + 636, + 637, + 638, + 639, + 640, + 641, + 642, + 643, + 644, + 645, + 646, + 647, + 648, + 649, + 650, + 651, + 652, + 653, + 654, + 655, + 656, + 657, + 658, + 659, + 660, + 661, + 662, + 663, + 664, + 665, + 666, + 667, + 668, + 669, + 670, + 671, + 672, + 673, + 674, + 675, + 676, + 677, + 678, + 679, + 680, + 681, + 682, + 683, + 684, + 685, + 686, + 687, + 688, + 689, + 690, + 691, + 692, + 693, + 694, + 695, + 696, + 697, + 698, + 699, + 700, + 701, + 702, + 703, + 704, + 705, + 706, + 707, + 708, + 709, + 710, + 711, + 712, + 713, + 714, + 715, + 716, + 717, + 718, + 719, + 720, + 721, + 722, + 723, + 724, + 725, + 726, + 727, + 728, + 729, + 730, + 731, + 732, + 733, + 734, + 735, + 736, + 737, + 738, + 739, + 740, + 741, + 742, + 743, + 744, + 745, + 746, + 747, + 748, + 749, + 750, + 751, + 752, + 753, + 754, + 755, + 756, + 757, + 758, + 759, + 760, + 761, + 762, + 763, + 764, + 765, + 766, + 767, + 768, + 769, + 770, + 771, + 772, + 773, + 774, + 775, + 776, + 777, + 778, + 779, + 780, + 781, + 782, + 783, + 784, + 785, + 786, + 787, + 788, + 789, + 790, + 791, + 792, + 793, + 794, + 795, + 796, + 797, + 798, + 799, + 800, + 801, + 802, + 803, + 804, + 805, + 806, + 807, + 808, + 809, + 810, + 811, + 812, + 813, + 814, + 815, + 816, + 817, + 818, + 819, + 820, + 821, + 822, + 823, + 824, + 825, + 826, + 827, + 828, + 829, + 830, + 831, + 832, + 833, + 834, + 835, + 836, + 837, + 838, + 839, + 840, + 841, + 842, + 843, + 844, + 845, + 846, + 847, + 848, + 849, + 850, + 851, + 852, + 853, + 854, + 855, + 856, + 857, + 858, + 859 + ] + } +] \ No newline at end of file diff --git a/010000/training_state/optimizer_state.safetensors b/010000/training_state/optimizer_state.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..f1e822cd10bec7b3ec7c9f0df1396c45b46cbd1b --- /dev/null +++ b/010000/training_state/optimizer_state.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3f2e3944bb65fe5ff1ad8b9141d46bddc718a980c23ab5895d7964696a25e0cc +size 22197120 diff --git a/010000/training_state/rng_state.safetensors b/010000/training_state/rng_state.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..b836eae2877cbd2c0d2f7d47eb3f28569134ec8c --- /dev/null +++ b/010000/training_state/rng_state.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:91c5f418c4990cf2c70b61330a2c66dc9a7c57d214099e1f4a94728fc2e9b87e +size 15708 diff --git a/010000/training_state/scheduler_state.json b/010000/training_state/scheduler_state.json new file mode 100644 index 0000000000000000000000000000000000000000..8eee341f67650bc134c1b342c9fb73117cb4b58e --- /dev/null +++ b/010000/training_state/scheduler_state.json @@ -0,0 +1,15 @@ +{ + "base_lrs": [ + 0.001 + ], + "last_epoch": 10000, + "_step_count": 10001, + "_is_initial": false, + "_get_lr_called_within_step": false, + "_last_lr": [ + 0.0001 + ], + "lr_lambdas": [ + null + ] +} \ No newline at end of file diff --git a/010000/training_state/training_step.json b/010000/training_state/training_step.json new file mode 100644 index 0000000000000000000000000000000000000000..7cb7c0986e9e7461ca851ce71e95d235ae3d2732 --- /dev/null +++ b/010000/training_state/training_step.json @@ -0,0 +1,3 @@ +{ + "step": 10000 +} \ No newline at end of file