frainer/task1_goalconditioned_envstate
03
1{2 "dataset": {3 "repo_id": "frainer/Task1_goalconditioned_envstate",4 "root": "/home/frainer/.cache/huggingface/lerobot/frainer/Task1_goalconditioned_envstate",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": "torchcodec",80 "return_uint8": false,81 "streaming": false82 },83 "env": null,84 "policy": {85 "type": "act",86 "n_obs_steps": 1,87 "input_features": {88 "observation.state": {89 "type": "STATE",90 "shape": [91 692 ]93 },94 "observation.images.front": {95 "type": "VISUAL",96 "shape": [97 3,98 480,99 640100 ]101 },102 "observation.environment_state": {103 "type": "ENV",104 "shape": [105 3106 ]107 }108 },109 "output_features": {110 "action": {111 "type": "ACTION",112 "shape": [113 6114 ]115 }116 },117 "device": "cuda",118 "use_amp": false,119 "use_peft": false,120 "push_to_hub": true,121 "repo_id": "frainer/task1_goalconditioned_envstate",122 "private": null,123 "tags": null,124 "license": null,125 "pretrained_path": null,126 "chunk_size": 100,127 "n_action_steps": 100,128 "normalization_mapping": {129 "VISUAL": "MEAN_STD",130 "STATE": "MEAN_STD",131 "ACTION": "MEAN_STD"132 },133 "vision_backbone": "resnet18",134 "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",135 "replace_final_stride_with_dilation": false,136 "pre_norm": false,137 "dim_model": 512,138 "n_heads": 8,139 "dim_feedforward": 3200,140 "feedforward_activation": "relu",141 "n_encoder_layers": 4,142 "n_decoder_layers": 1,143 "use_vae": true,144 "latent_dim": 32,145 "n_vae_encoder_layers": 4,146 "temporal_ensemble_coeff": null,147 "dropout": 0.1,148 "kl_weight": 10.0,149 "optimizer_lr": 1e-05,150 "optimizer_weight_decay": 0.0001,151 "optimizer_lr_backbone": 1e-05152 },153 "reward_model": null,154 "output_dir": "outputs/train/act_task1_goalconditioned_envstate",155 "job_name": "act_task1_goalconditioned_envstate",156 "resume": false,157 "seed": 1000,158 "cudnn_deterministic": false,159 "num_workers": 4,160 "batch_size": 8,161 "prefetch_factor": 4,162 "persistent_workers": true,163 "steps": 100000,164 "eval_freq": 20000,165 "log_freq": 200,166 "tolerance_s": 0.0001,167 "save_checkpoint": true,168 "save_freq": 10000,169 "use_policy_training_preset": true,170 "optimizer": {171 "type": "adamw",172 "lr": 1e-05,173 "weight_decay": 0.0001,174 "grad_clip_norm": 10.0,175 "betas": [176 0.9,177 0.999178 ],179 "eps": 1e-08180 },181 "scheduler": null,182 "eval": {183 "n_episodes": 50,184 "batch_size": 11,185 "use_async_envs": true186 },187 "wandb": {188 "enable": false,189 "disable_artifact": false,190 "project": "lerobot",191 "entity": null,192 "notes": null,193 "run_id": null,194 "mode": null,195 "add_tags": true196 },197 "peft": null,198 "sample_weighting": null,199 "rename_map": {},200 "checkpoint_path": null201}