Robotics
Transformers
Safetensors
Inference Endpoints
aliberts HF staff commited on
Commit
81e6614
·
verified ·
1 Parent(s): ba8a992

Delete config.yaml

Browse files
Files changed (1) hide show
  1. 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