{ "dataset": { "repo_id": "wuc1/bi_so101_ffp_20260603_200349_subtask", "root": null, "episodes": null, "image_transforms": { "enable": true, "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", "return_uint8": false, "streaming": false }, "env": null, "policy": { "type": "smolvla", "n_obs_steps": 1, "input_features": { "observation.state": { "type": "STATE", "shape": [ 12 ] }, "observation.images.left_camera1": { "type": "VISUAL", "shape": [ 3, 480, 640 ] }, "observation.images.left_camera3": { "type": "VISUAL", "shape": [ 3, 480, 640 ] }, "observation.images.left_camera4": { "type": "VISUAL", "shape": [ 3, 240, 320 ] }, "observation.images.right_camera2": { "type": "VISUAL", "shape": [ 3, 240, 320 ] } }, "output_features": { "action": { "type": "ACTION", "shape": [ 12 ] } }, "device": "cuda", "use_amp": false, "use_peft": false, "push_to_hub": true, "repo_id": "wuc1/bi_so101_ffp_RA-BC", "private": null, "tags": null, "license": null, "pretrained_path": "wuc1/bi_so101_cloth-base_256", "chunk_size": 50, "n_action_steps": 50, "normalization_mapping": { "VISUAL": "IDENTITY", "STATE": "MEAN_STD", "ACTION": "MEAN_STD" }, "max_state_dim": 32, "max_action_dim": 32, "use_state": true, "input_dropout_prob": 0.0, "input_dropout_features": [], "training_time_rtc_enabled": false, "training_time_rtc_max_delay_steps": 12, "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" ], "resize_imgs_with_padding": [ 512, 512 ], "image_seq_len": 256, "empty_cameras": 0, "adapt_to_pi_aloha": false, "use_delta_joint_actions_aloha": false, "tokenizer_max_length": 48, "num_steps": 10, "use_cache": true, "freeze_vision_encoder": true, "train_expert_only": true, "train_state_proj": true, "optimizer_lr": 0.0001, "optimizer_betas": [ 0.9, 0.95 ], "optimizer_eps": 1e-08, "optimizer_weight_decay": 1e-10, "optimizer_grad_clip_norm": 10.0, "scheduler_warmup_steps": 1000, "scheduler_decay_steps": 100000, "scheduler_decay_lr": 2.5e-06, "vlm_model_name": "HuggingFaceTB/SmolVLM2-500M-Video-Instruct", "load_vlm_weights": true, "add_image_special_tokens": false, "attention_mode": "cross_attn", "prefix_length": 0, "pad_language_to": "max_length", "num_expert_layers": 0, "num_vlm_layers": 16, "self_attn_every_n_layers": 2, "expert_width_multiplier": 0.75, "min_period": 0.004, "max_period": 4.0, "rtc_config": null, "compile_model": true, "compile_mode": "default" }, "reward_model": null, "output_dir": "outputs/train/2026-06-11/02-42-58_smolvla", "job_name": "smolvla", "resume": false, "seed": 1000, "cudnn_deterministic": false, "num_workers": 4, "batch_size": 8, "prefetch_factor": 4, "persistent_workers": true, "steps": 100000, "eval_freq": 20000, "log_freq": 1000, "tolerance_s": 0.01, "save_checkpoint": true, "save_freq": 120000, "use_policy_training_preset": true, "optimizer": { "type": "adamw", "lr": 0.0001, "weight_decay": 1e-10, "grad_clip_norm": 10.0, "betas": [ 0.9, 0.95 ], "eps": 1e-08 }, "scheduler": { "type": "cosine_decay_with_warmup", "num_warmup_steps": 1000, "num_decay_steps": 100000, "peak_lr": 0.0001, "decay_lr": 2.5e-06 }, "eval": { "n_episodes": 50, "batch_size": 22, "use_async_envs": true }, "wandb": { "enable": true, "disable_artifact": false, "project": "lerobot", "entity": null, "notes": null, "run_id": "n75aulok", "mode": null, "add_tags": true }, "peft": null, "use_subtasks": false, "sample_weighting": { "type": "rabc", "progress_path": null, "head_mode": "sparse", "kappa": 0.01, "epsilon": 1e-06, "extra_params": {} }, "rename_map": {}, "checkpoint_path": null }