rainbowrobotics/act_rb5_smooth
04
1{2 "dataset": {3 "repo_id": "rainbowrobotics/rb5_pp_demo_smooth",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 "return_uint8": false,81 "depth_output_unit": "mm",82 "streaming": false,83 "eval_split": 0.084 },85 "env": null,86 "policy": {87 "type": "act",88 "n_obs_steps": 1,89 "input_features": {90 "observation.state": {91 "type": "STATE",92 "shape": [93 794 ]95 },96 "observation.images.front": {97 "type": "VISUAL",98 "shape": [99 3,100 640,101 480102 ]103 }104 },105 "output_features": {106 "action": {107 "type": "ACTION",108 "shape": [109 7110 ]111 }112 },113 "device": "cuda",114 "use_amp": false,115 "use_peft": false,116 "push_to_hub": true,117 "repo_id": "rainbowrobotics/act_rb5_smooth_0805",118 "private": null,119 "tags": null,120 "license": null,121 "pretrained_path": null,122 "pretrained_revision": null,123 "chunk_size": 100,124 "n_action_steps": 100,125 "normalization_mapping": {126 "VISUAL": "MEAN_STD",127 "STATE": "MEAN_STD",128 "ACTION": "MEAN_STD"129 },130 "vision_backbone": "resnet18",131 "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",132 "replace_final_stride_with_dilation": false,133 "pre_norm": false,134 "dim_model": 512,135 "n_heads": 8,136 "dim_feedforward": 3200,137 "feedforward_activation": "relu",138 "n_encoder_layers": 4,139 "n_decoder_layers": 1,140 "use_vae": true,141 "latent_dim": 32,142 "n_vae_encoder_layers": 4,143 "temporal_ensemble_coeff": null,144 "dropout": 0.1,145 "kl_weight": 10.0,146 "optimizer_lr": 1e-05,147 "optimizer_weight_decay": 0.0001,148 "optimizer_lr_backbone": 1e-05149 },150 "reward_model": null,151 "output_dir": "outputs/train/act_rb5_smooth_0805",152 "job_name": "act_rb5_smooth_0805",153 "resume": false,154 "seed": 1000,155 "cudnn_deterministic": false,156 "num_workers": 4,157 "batch_size": 64,158 "prefetch_factor": 4,159 "persistent_workers": true,160 "steps": 10000,161 "env_eval_freq": 20000,162 "log_freq": 100,163 "eval_steps": 0,164 "max_eval_samples": 0,165 "tolerance_s": 0.0001,166 "save_checkpoint": true,167 "save_freq": 2000,168 "use_policy_training_preset": true,169 "optimizer": {170 "type": "adamw",171 "lr": 1e-05,172 "weight_decay": 0.0001,173 "grad_clip_norm": 10.0,174 "betas": [175 0.9,176 0.999177 ],178 "eps": 1e-08179 },180 "scheduler": null,181 "eval": {182 "n_episodes": 50,183 "batch_size": 50,184 "use_async_envs": true,185 "recording": false,186 "recording_repo_id": null,187 "recording_private": false188 },189 "wandb": {190 "enable": false,191 "disable_artifact": false,192 "project": "lerobot",193 "entity": null,194 "notes": null,195 "run_id": null,196 "mode": null,197 "add_tags": true198 },199 "peft": null,200 "job": {201 "target": null,202 "image": "huggingface/lerobot-gpu:latest",203 "timeout": "2d",204 "detach": false,205 "tags": []206 },207 "save_checkpoint_to_hub": false,208 "sample_weighting": null,209 "rename_map": {},210 "checkpoint_path": null211}