Delete config.yaml
Browse files- config.yaml +0 -141
config.yaml
DELETED
@@ -1,141 +0,0 @@
|
|
1 |
-
device: cuda
|
2 |
-
use_amp: false
|
3 |
-
seed: 100000
|
4 |
-
dataset_repo_id: lerobot/pusht
|
5 |
-
training:
|
6 |
-
offline_steps: 200000
|
7 |
-
online_steps: 0
|
8 |
-
online_steps_between_rollouts: 1
|
9 |
-
online_sampling_ratio: 0.5
|
10 |
-
online_env_seed: ???
|
11 |
-
eval_freq: 10000
|
12 |
-
save_freq: 20000
|
13 |
-
log_freq: 250
|
14 |
-
save_model: true
|
15 |
-
batch_size: 64
|
16 |
-
grad_clip_norm: 10
|
17 |
-
lr: 0.0001
|
18 |
-
lr_scheduler: cosine
|
19 |
-
lr_warmup_steps: 500
|
20 |
-
adam_betas:
|
21 |
-
- 0.95
|
22 |
-
- 0.999
|
23 |
-
adam_eps: 1.0e-08
|
24 |
-
adam_weight_decay: 1.0e-06
|
25 |
-
delta_timestamps:
|
26 |
-
observation.image:
|
27 |
-
- -0.1
|
28 |
-
- 0.0
|
29 |
-
observation.state:
|
30 |
-
- -0.1
|
31 |
-
- 0.0
|
32 |
-
action:
|
33 |
-
- -0.1
|
34 |
-
- 0.0
|
35 |
-
- 0.1
|
36 |
-
- 0.2
|
37 |
-
- 0.3
|
38 |
-
- 0.4
|
39 |
-
- 0.5
|
40 |
-
- 0.6
|
41 |
-
- 0.7
|
42 |
-
- 0.8
|
43 |
-
- 0.9
|
44 |
-
- 1.0
|
45 |
-
- 1.1
|
46 |
-
- 1.2
|
47 |
-
- 1.3
|
48 |
-
- 1.4
|
49 |
-
n_end_keyframes_dropped: ${policy.horizon} - ${policy.n_action_steps} - ${policy.n_obs_steps}
|
50 |
-
+ 1
|
51 |
-
eval:
|
52 |
-
n_episodes: 50
|
53 |
-
batch_size: 50
|
54 |
-
use_async_envs: false
|
55 |
-
wandb:
|
56 |
-
enable: true
|
57 |
-
disable_artifact: true
|
58 |
-
project: lerobot
|
59 |
-
notes: ''
|
60 |
-
fps: 10
|
61 |
-
env:
|
62 |
-
name: pusht
|
63 |
-
task: PushT-v0
|
64 |
-
image_size: 96
|
65 |
-
state_dim: 2
|
66 |
-
action_dim: 2
|
67 |
-
fps: ${fps}
|
68 |
-
episode_length: 300
|
69 |
-
gym:
|
70 |
-
obs_type: pixels_agent_pos
|
71 |
-
render_mode: rgb_array
|
72 |
-
visualization_width: 384
|
73 |
-
visualization_height: 384
|
74 |
-
override_dataset_stats:
|
75 |
-
observation.image:
|
76 |
-
mean:
|
77 |
-
- - - 0.5
|
78 |
-
- - - 0.5
|
79 |
-
- - - 0.5
|
80 |
-
std:
|
81 |
-
- - - 0.5
|
82 |
-
- - - 0.5
|
83 |
-
- - - 0.5
|
84 |
-
observation.state:
|
85 |
-
min:
|
86 |
-
- 13.456424
|
87 |
-
- 32.938293
|
88 |
-
max:
|
89 |
-
- 496.14618
|
90 |
-
- 510.9579
|
91 |
-
action:
|
92 |
-
min:
|
93 |
-
- 12.0
|
94 |
-
- 25.0
|
95 |
-
max:
|
96 |
-
- 511.0
|
97 |
-
- 511.0
|
98 |
-
policy:
|
99 |
-
name: diffusion
|
100 |
-
n_obs_steps: 2
|
101 |
-
horizon: 16
|
102 |
-
n_action_steps: 8
|
103 |
-
input_shapes:
|
104 |
-
observation.image:
|
105 |
-
- 3
|
106 |
-
- 96
|
107 |
-
- 96
|
108 |
-
observation.state:
|
109 |
-
- ${env.state_dim}
|
110 |
-
output_shapes:
|
111 |
-
action:
|
112 |
-
- ${env.action_dim}
|
113 |
-
input_normalization_modes:
|
114 |
-
observation.image: mean_std
|
115 |
-
observation.state: min_max
|
116 |
-
output_normalization_modes:
|
117 |
-
action: min_max
|
118 |
-
vision_backbone: resnet18
|
119 |
-
crop_shape:
|
120 |
-
- 84
|
121 |
-
- 84
|
122 |
-
crop_is_random: true
|
123 |
-
pretrained_backbone_weights: null
|
124 |
-
use_group_norm: true
|
125 |
-
spatial_softmax_num_keypoints: 32
|
126 |
-
down_dims:
|
127 |
-
- 512
|
128 |
-
- 1024
|
129 |
-
- 2048
|
130 |
-
kernel_size: 5
|
131 |
-
n_groups: 8
|
132 |
-
diffusion_step_embed_dim: 128
|
133 |
-
use_film_scale_modulation: true
|
134 |
-
num_train_timesteps: 100
|
135 |
-
beta_schedule: squaredcos_cap_v2
|
136 |
-
beta_start: 0.0001
|
137 |
-
beta_end: 0.02
|
138 |
-
prediction_type: epsilon
|
139 |
-
clip_sample: true
|
140 |
-
clip_sample_range: 1.0
|
141 |
-
num_inference_steps: 100
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|