arclabmit/xarm7_diffusion_beavrsim_shellgame_model
08
1{2 "dataset": {3 "repo_id": "arclabmit/xarm7_beavrsim_shellgame_dataset",4 "root": null,5 "episodes": null,6 "image_transforms": {7 "enable": false,8 "max_num_transforms": 3,9 "random_order": false,10 "tfs": {11 "brightness": {12 "weight": 1.0,13 "type": "ColorJitter",14 "kwargs": {15 "brightness": [16 0.8,17 1.218 ]19 }20 },21 "contrast": {22 "weight": 1.0,23 "type": "ColorJitter",24 "kwargs": {25 "contrast": [26 0.8,27 1.228 ]29 }30 },31 "saturation": {32 "weight": 1.0,33 "type": "ColorJitter",34 "kwargs": {35 "saturation": [36 0.5,37 1.538 ]39 }40 },41 "hue": {42 "weight": 1.0,43 "type": "ColorJitter",44 "kwargs": {45 "hue": [46 -0.05,47 0.0548 ]49 }50 },51 "sharpness": {52 "weight": 1.0,53 "type": "SharpnessJitter",54 "kwargs": {55 "sharpness": [56 0.5,57 1.558 ]59 }60 },61 "affine": {62 "weight": 1.0,63 "type": "RandomAffine",64 "kwargs": {65 "degrees": [66 -5.0,67 5.068 ],69 "translate": [70 0.05,71 0.0572 ]73 }74 }75 }76 },77 "revision": null,78 "use_imagenet_stats": true,79 "video_backend": "pyav",80 "streaming": false81 },82 "env": {83 "type": "beavr",84 "task": null,85 "fps": 30,86 "features": {87 "action": {88 "type": "ACTION",89 "shape": [90 891 ]92 },93 "agent_pos": {94 "type": "STATE",95 "shape": [96 897 ]98 },99 "overhead": {100 "type": "VISUAL",101 "shape": [102 3,103 480,104 640105 ]106 }107 },108 "features_map": {109 "action": "action",110 "agent_pos": "observation.state",111 "overhead": "observation.images.overhead"112 },113 "max_parallel_tasks": 1,114 "disable_env_checker": true,115 "scene": "scene_shell_game",116 "episode_length": 1000117 },118 "policy": {119 "type": "diffusion",120 "n_obs_steps": 2,121 "input_features": {122 "observation.state": {123 "type": "STATE",124 "shape": [125 8126 ]127 },128 "observation.images.overhead": {129 "type": "VISUAL",130 "shape": [131 3,132 480,133 640134 ]135 }136 },137 "output_features": {138 "action": {139 "type": "ACTION",140 "shape": [141 8142 ]143 }144 },145 "device": "cuda",146 "use_amp": false,147 "use_peft": false,148 "push_to_hub": true,149 "repo_id": "arclabmit/xarm7_diffusion_beavrsim_shellgame_model",150 "private": null,151 "tags": null,152 "license": null,153 "pretrained_path": null,154 "horizon": 16,155 "n_action_steps": 8,156 "normalization_mapping": {157 "VISUAL": "MEAN_STD",158 "STATE": "MIN_MAX",159 "ACTION": "MIN_MAX"160 },161 "drop_n_last_frames": 7,162 "vision_backbone": "resnet18",163 "crop_shape": [164 84,165 84166 ],167 "crop_is_random": true,168 "pretrained_backbone_weights": null,169 "use_group_norm": true,170 "spatial_softmax_num_keypoints": 32,171 "use_separate_rgb_encoder_per_camera": false,172 "down_dims": [173 512,174 1024,175 2048176 ],177 "kernel_size": 5,178 "n_groups": 8,179 "diffusion_step_embed_dim": 128,180 "use_film_scale_modulation": true,181 "noise_scheduler_type": "DDPM",182 "num_train_timesteps": 100,183 "beta_schedule": "squaredcos_cap_v2",184 "beta_start": 0.0001,185 "beta_end": 0.02,186 "prediction_type": "epsilon",187 "clip_sample": true,188 "clip_sample_range": 1.0,189 "num_inference_steps": null,190 "do_mask_loss_for_padding": false,191 "optimizer_lr": 0.0001,192 "optimizer_betas": [193 0.95,194 0.999195 ],196 "optimizer_eps": 1e-08,197 "optimizer_weight_decay": 1e-06,198 "scheduler_name": "cosine",199 "scheduler_warmup_steps": 500200 },201 "output_dir": "outputs/train/2026-01-22/19-28-59_xarm7_diffusion_beavrsim_shellgame_model",202 "job_name": "xarm7_diffusion_beavrsim_shellgame_model",203 "resume": false,204 "seed": 1000,205 "num_workers": 4,206 "batch_size": 8,207 "steps": 100000,208 "eval_freq": 20000,209 "log_freq": 200,210 "tolerance_s": 0.0001,211 "save_checkpoint": true,212 "save_freq": 20000,213 "use_policy_training_preset": true,214 "optimizer": {215 "type": "adam",216 "lr": 0.0001,217 "weight_decay": 1e-06,218 "grad_clip_norm": 10.0,219 "betas": [220 0.95,221 0.999222 ],223 "eps": 1e-08224 },225 "scheduler": {226 "type": "diffuser",227 "num_warmup_steps": 500,228 "name": "cosine"229 },230 "eval": {231 "n_episodes": 50,232 "batch_size": 25,233 "use_async_envs": false234 },235 "wandb": {236 "enable": true,237 "disable_artifact": false,238 "project": "lerobot",239 "entity": null,240 "notes": null,241 "run_id": "36uswd3s",242 "mode": null243 },244 "peft": null,245 "use_rabc": false,246 "rabc_progress_path": null,247 "rabc_kappa": 0.01,248 "rabc_epsilon": 1e-06,249 "rabc_head_mode": "sparse",250 "rename_map": {},251 "checkpoint_path": null252}