jungjaeuk/act_temporal_test_model
05
1{
2 "dataset": {
3 "repo_id": "jungjaeuk/act_temporal_test_01",
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.2
18 ]
19 }
20 },
21 "contrast": {
22 "weight": 1.0,
23 "type": "ColorJitter",
24 "kwargs": {
25 "contrast": [
26 0.8,
27 1.2
28 ]
29 }
30 },
31 "saturation": {
32 "weight": 1.0,
33 "type": "ColorJitter",
34 "kwargs": {
35 "saturation": [
36 0.5,
37 1.5
38 ]
39 }
40 },
41 "hue": {
42 "weight": 1.0,
43 "type": "ColorJitter",
44 "kwargs": {
45 "hue": [
46 -0.05,
47 0.05
48 ]
49 }
50 },
51 "sharpness": {
52 "weight": 1.0,
53 "type": "SharpnessJitter",
54 "kwargs": {
55 "sharpness": [
56 0.5,
57 1.5
58 ]
59 }
60 },
61 "affine": {
62 "weight": 1.0,
63 "type": "RandomAffine",
64 "kwargs": {
65 "degrees": [
66 -5.0,
67 5.0
68 ],
69 "translate": [
70 0.05,
71 0.05
72 ]
73 }
74 }
75 }
76 },
77 "revision": null,
78 "use_imagenet_stats": true,
79 "video_backend": "pyav",
80 "streaming": false
81 },
82 "env": null,
83 "policy": {
84 "type": "act",
85 "n_obs_steps": 1,
86 "input_features": {
87 "observation.state": {
88 "type": "STATE",
89 "shape": [
90 6
91 ]
92 },
93 "observation.images.front": {
94 "type": "VISUAL",
95 "shape": [
96 3,
97 480,
98 640
99 ]
100 }
101 },
102 "output_features": {
103 "action": {
104 "type": "ACTION",
105 "shape": [
106 6
107 ]
108 }
109 },
110 "device": "cuda",
111 "use_amp": true,
112 "use_peft": false,
113 "push_to_hub": true,
114 "repo_id": "jungjaeuk/act_temporal_test_model",
115 "private": null,
116 "tags": null,
117 "license": null,
118 "pretrained_path": null,
119 "chunk_size": 100,
120 "n_action_steps": 100,
121 "normalization_mapping": {
122 "VISUAL": "MEAN_STD",
123 "STATE": "MEAN_STD",
124 "ACTION": "MEAN_STD"
125 },
126 "vision_backbone": "resnet18",
127 "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
128 "replace_final_stride_with_dilation": false,
129 "pre_norm": false,
130 "dim_model": 512,
131 "n_heads": 8,
132 "dim_feedforward": 3200,
133 "feedforward_activation": "relu",
134 "n_encoder_layers": 4,
135 "n_decoder_layers": 1,
136 "use_vae": true,
137 "latent_dim": 32,
138 "n_vae_encoder_layers": 4,
139 "temporal_ensemble_coeff": null,
140 "dropout": 0.1,
141 "kl_weight": 10.0,
142 "optimizer_lr": 1e-05,
143 "optimizer_weight_decay": 0.0001,
144 "optimizer_lr_backbone": 1e-05
145 },
146 "output_dir": "outputs\\act_temporal_test_model",
147 "job_name": "act",
148 "resume": false,
149 "seed": 1000,
150 "cudnn_deterministic": false,
151 "num_workers": 4,
152 "batch_size": 8,
153 "steps": 10000,
154 "eval_freq": 20000,
155 "log_freq": 200,
156 "tolerance_s": 0.0001,
157 "save_checkpoint": true,
158 "save_freq": 20000,
159 "use_policy_training_preset": true,
160 "optimizer": {
161 "type": "adamw",
162 "lr": 1e-05,
163 "weight_decay": 0.0001,
164 "grad_clip_norm": 10.0,
165 "betas": [
166 0.9,
167 0.999
168 ],
169 "eps": 1e-08
170 },
171 "scheduler": null,
172 "eval": {
173 "n_episodes": 50,
174 "batch_size": 50,
175 "use_async_envs": false
176 },
177 "wandb": {
178 "enable": false,
179 "disable_artifact": false,
180 "project": "lerobot",
181 "entity": null,
182 "notes": null,
183 "run_id": null,
184 "mode": null,
185 "add_tags": true
186 },
187 "peft": null,
188 "use_rabc": false,
189 "rabc_progress_path": null,
190 "rabc_kappa": 0.01,
191 "rabc_epsilon": 1e-06,
192 "rabc_head_mode": "sparse",
193 "rename_map": {},
194 "checkpoint_path": null
195}