AnonymousMouse404/diffusion_blockstack
09
1{2 "dataset": {3 "repo_id": "AnonymousMouse404/BlockStack",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": null,83 "policy": {84 "type": "diffusion",85 "n_obs_steps": 2,86 "input_features": {87 "observation.state": {88 "type": "STATE",89 "shape": [90 691 ]92 },93 "observation.images.top": {94 "type": "VISUAL",95 "shape": [96 3,97 480,98 64099 ]100 },101 "observation.images.wrist": {102 "type": "VISUAL",103 "shape": [104 3,105 480,106 640107 ]108 }109 },110 "output_features": {111 "action": {112 "type": "ACTION",113 "shape": [114 6115 ]116 }117 },118 "device": "cuda",119 "use_amp": true,120 "use_peft": false,121 "push_to_hub": true,122 "repo_id": "AnonymousMouse404/diffusion_blockstack",123 "private": null,124 "tags": null,125 "license": null,126 "pretrained_path": null,127 "horizon": 16,128 "n_action_steps": 8,129 "normalization_mapping": {130 "VISUAL": "MEAN_STD",131 "STATE": "MIN_MAX",132 "ACTION": "MIN_MAX"133 },134 "drop_n_last_frames": 7,135 "vision_backbone": "resnet18",136 "resize_shape": null,137 "crop_ratio": 1.0,138 "crop_shape": null,139 "crop_is_random": true,140 "pretrained_backbone_weights": null,141 "use_group_norm": true,142 "spatial_softmax_num_keypoints": 32,143 "use_separate_rgb_encoder_per_camera": false,144 "down_dims": [145 512,146 1024,147 2048148 ],149 "kernel_size": 5,150 "n_groups": 8,151 "diffusion_step_embed_dim": 128,152 "use_film_scale_modulation": true,153 "noise_scheduler_type": "DDPM",154 "num_train_timesteps": 100,155 "beta_schedule": "squaredcos_cap_v2",156 "beta_start": 0.0001,157 "beta_end": 0.02,158 "prediction_type": "epsilon",159 "clip_sample": true,160 "clip_sample_range": 1.0,161 "num_inference_steps": null,162 "compile_model": false,163 "compile_mode": "reduce-overhead",164 "do_mask_loss_for_padding": false,165 "optimizer_lr": 0.0001,166 "optimizer_betas": [167 0.95,168 0.999169 ],170 "optimizer_eps": 1e-08,171 "optimizer_weight_decay": 1e-06,172 "scheduler_name": "cosine",173 "scheduler_warmup_steps": 500174 },175 "output_dir": "outputs/train/diffusion_blockstack",176 "job_name": "diffusion",177 "resume": false,178 "seed": 1000,179 "cudnn_deterministic": false,180 "num_workers": 16,181 "batch_size": 32,182 "steps": 100000,183 "eval_freq": 20000,184 "log_freq": 200,185 "tolerance_s": 0.0001,186 "save_checkpoint": true,187 "save_freq": 10000,188 "use_policy_training_preset": true,189 "optimizer": {190 "type": "adam",191 "lr": 0.0001,192 "weight_decay": 1e-06,193 "grad_clip_norm": 10.0,194 "betas": [195 0.95,196 0.999197 ],198 "eps": 1e-08199 },200 "scheduler": {201 "type": "diffuser",202 "num_warmup_steps": 500,203 "name": "cosine"204 },205 "eval": {206 "n_episodes": 50,207 "batch_size": 50,208 "use_async_envs": false209 },210 "wandb": {211 "enable": true,212 "disable_artifact": false,213 "project": "diffusion-blockstack",214 "entity": null,215 "notes": null,216 "run_id": "z5tgczyt",217 "mode": null,218 "add_tags": true219 },220 "peft": null,221 "use_rabc": false,222 "rabc_progress_path": null,223 "rabc_kappa": 0.01,224 "rabc_epsilon": 1e-06,225 "rabc_head_mode": "sparse",226 "rename_map": {},227 "checkpoint_path": null228}