{ "dataset": { "repo_id": [ "chomeed/anubis_pick_carrot_out_of_pot_randomized_train", "chomeed/anubis_stack_tuna_train", "lerobot/aloha_static_cups_open_with_reward_train", "chomeed/aloha_static_candy_train", "jellyho/aloha_box_into_pot_train", "jellyho/aloha_dish_drainer_train", "jellyho/aloha_lift_box_train" ], "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": "smolvlm_rm", "n_obs_steps": 1, "input_features": { "observation.state": { "type": "STATE", "shape": [ 14 ] }, "observation.images.front": { "type": "VISUAL", "shape": [ 3, 240, 320 ] }, "observation.images.wrist_left": { "type": "VISUAL", "shape": [ 3, 240, 320 ] }, "observation.images.wrist_right": { "type": "VISUAL", "shape": [ 3, 240, 320 ] } }, "output_features": { "action": { "type": "ACTION", "shape": [ 23 ] }, "action_quat": { "type": "ACTION", "shape": [ 19 ] } }, "device": "cuda", "use_amp": false, "use_peft": false, "push_to_hub": true, "repo_id": "chomeed/test_smolvlm_rm", "private": null, "tags": null, "license": null, "pretrained_path": null, "normalization_mapping": { "VISUAL": "IDENTITY", "STATE": "MEAN_STD", "ACTION": "MEAN_STD" }, "max_state_dim": 32, "resize_imgs_with_padding": [ 512, 512 ], "empty_cameras": 0, "tokenizer_max_length": 48, "num_steps": 10, "use_cache": true, "freeze_vision_encoder": false, "train_expert_only": false, "optimizer_lr": 0.0001, "optimizer_betas": [ 0.9, 0.95 ], "optimizer_eps": 1e-08, "optimizer_weight_decay": 1e-10, "optimizer_grad_clip_norm": 10, "scheduler_warmup_steps": 1000, "scheduler_decay_steps": 20000, "scheduler_decay_lr": 2.5e-06, "scheduler_peak_lr": 5e-05, "vlm_model_name": "HuggingFaceTB/SmolVLM-500M-Instruct", "load_vlm_weights": true, "add_image_special_tokens": false, "attention_mode": "cross_attn", "prefix_length": -1, "pad_language_to": "longest", "num_expert_layers": -1, "num_vlm_layers": 16, "self_attn_every_n_layers": 2, "expert_width_multiplier": 0.75, "min_period": 0.004, "max_period": 4.0 }, "output_dir": "outputs/train/2026-02-02/02-14-18_smolvlm_500M_rm", "job_name": "smolvlm_500M_rm", "resume": false, "seed": 1000, "num_workers": 4, "batch_size": 16, "steps": 20000, "eval_freq": 1000, "log_freq": 200, "tolerance_s": 0.0001, "save_checkpoint": true, "save_freq": 5000, "use_policy_training_preset": true, "optimizer": { "type": "adamw", "lr": 0.0001, "weight_decay": 1e-10, "grad_clip_norm": 10, "betas": [ 0.9, 0.95 ], "eps": 1e-08 }, "scheduler": { "type": "cosine_decay_with_warmup", "num_warmup_steps": 1000, "num_decay_steps": 20000, "peak_lr": 5e-05, "decay_lr": 2.5e-06 }, "val": { "repo_id": [ "chomeed/anubis_pick_carrot_out_of_pot_randomized_val_single", "chomeed/anubis_stack_tuna_single", "chomeed/aloha_static_cups_open_single", "chomeed/aloha_static_candy_single", "jellyho/aloha_box_into_pot_single", "jellyho/aloha_dish_drainer_single", "jellyho/aloha_lift_box_single" ], "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 }, "test": { "repo_id": [ "chomeed/aloha_static_coffee_single", "jellyho/aloha_handover_box_single" ], "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 }, "wandb": { "enable": true, "disable_artifact": true, "project": "hrm", "entity": null, "notes": null, "run_id": "obruhl6j", "mode": null }, "peft": null, "use_rabc": false, "rabc_progress_path": null, "rabc_kappa": 0.01, "rabc_epsilon": 1e-06, "rabc_head_mode": "sparse", "rename_map": {}, "checkpoint_path": null }