Safetensors
akolobov commited on
Commit
58ca771
·
verified ·
1 Parent(s): 4f9af0f

Remove the TabletopSim dataset config and statistics

Browse files

Fixes https://github.com/microsoft/rhobotics/issues/53 for this checkpoint.

Removes `data_config.json` and `stats.json`, which were made for a TabletopSim dataset, and drops them from `manifest.json`. Pretraining normalized each robot's data with its own statistics, so no single set of statistics fits this checkpoint.

Fine-tuning is unaffected. Evaluating or serving `rho-base` directly now needs a dataset configuration for the target robot; https://github.com/microsoft/rhobotics/pull/77 makes loading without one fail with a clear message.

Files changed (3) hide show
  1. data_config.json +0 -171
  2. manifest.json +2 -2
  3. stats.json +0 -0
data_config.json DELETED
@@ -1,171 +0,0 @@
1
- {
2
- "batch_size": 192,
3
- "num_workers": 8,
4
- "shuffle_buffer_size": 1000,
5
- "normalization_mapping": {
6
- "ACTION": "ACTIONCHUNK_QUANTILE",
7
- "ENV": "QUANTILE",
8
- "STATE": "QUANTILE",
9
- "VISUAL": "IDENTITY"
10
- },
11
- "features": {
12
- "action": {
13
- "type": "ACTION",
14
- "shape": "(20,)"
15
- },
16
- "observation.image.0": {
17
- "type": "VISUAL",
18
- "shape": "(3, 224, 224)"
19
- },
20
- "observation.image.1": {
21
- "type": "VISUAL",
22
- "shape": "(3, 224, 224)"
23
- },
24
- "observation.image.2": {
25
- "type": "VISUAL",
26
- "shape": "(3, 224, 224)"
27
- },
28
- "observation.state": {
29
- "type": "STATE",
30
- "shape": "(20,)"
31
- }
32
- },
33
- "clip_values": {
34
- "ENV": [
35
- -1.05,
36
- 1.05
37
- ],
38
- "STATE": [
39
- -1.05,
40
- 1.05
41
- ]
42
- },
43
- "chunk_size": null,
44
- "action_time_horizon_s": null,
45
- "min_action_chunk_size": 8,
46
- "observation_mapping": {
47
- "action.ee_6d_pos": "action",
48
- "observation.ee_6d_pos": "observation.state",
49
- "observation.images.agentview_image": "observation.image.0",
50
- "observation.images.left_wrist_image": "observation.image.1",
51
- "observation.images.right_wrist_image": "observation.image.2",
52
- "observation.language_instruction": "task"
53
- },
54
- "observation_whitelist": null,
55
- "observation_blacklist": null,
56
- "transform_mapping": {
57
- "action": [
58
- {
59
- "type": "convert_to_6d_actions",
60
- "action_key": "action",
61
- "action_type": "EE_6D_POS",
62
- "input_type": "Dict",
63
- "post_norm": false
64
- },
65
- {
66
- "type": "delta_actions",
67
- "state_key": "observation.state",
68
- "action_key": "action",
69
- "action_type": "EE_6D_POS",
70
- "input_type": "Dict",
71
- "post_norm": false,
72
- "relative_to_state": true,
73
- "use_absolute_grippers": false
74
- }
75
- ],
76
- "observation.image.0": [
77
- {
78
- "type": "resize_with_padding",
79
- "height": 224,
80
- "width": 224,
81
- "mode": "bilinear",
82
- "align_corners": false,
83
- "input_type": "Tensor"
84
- },
85
- {
86
- "type": "color_jitter",
87
- "brightness": 0.2,
88
- "contrast": [
89
- 0.8,
90
- 1.2
91
- ],
92
- "saturation": [
93
- 0.8,
94
- 1.2
95
- ],
96
- "hue": 0.05,
97
- "input_type": "Tensor"
98
- }
99
- ],
100
- "observation.image.1": [
101
- {
102
- "type": "resize_with_padding",
103
- "height": 224,
104
- "width": 224,
105
- "mode": "bilinear",
106
- "align_corners": false,
107
- "input_type": "Tensor"
108
- },
109
- {
110
- "type": "color_jitter",
111
- "brightness": 0.2,
112
- "contrast": [
113
- 0.8,
114
- 1.2
115
- ],
116
- "saturation": [
117
- 0.8,
118
- 1.2
119
- ],
120
- "hue": 0.05,
121
- "input_type": "Tensor"
122
- }
123
- ],
124
- "observation.image.2": [
125
- {
126
- "type": "resize_with_padding",
127
- "height": 224,
128
- "width": 224,
129
- "mode": "bilinear",
130
- "align_corners": false,
131
- "input_type": "Tensor"
132
- },
133
- {
134
- "type": "color_jitter",
135
- "brightness": 0.2,
136
- "contrast": [
137
- 0.8,
138
- 1.2
139
- ],
140
- "saturation": [
141
- 0.8,
142
- 1.2
143
- ],
144
- "hue": 0.05,
145
- "input_type": "Tensor"
146
- }
147
- ],
148
- "observation.state": [
149
- {
150
- "type": "convert_to_6d_actions",
151
- "action_key": "observation.state",
152
- "action_type": "EE_6D_POS",
153
- "input_type": "Dict",
154
- "post_norm": false
155
- }
156
- ]
157
- },
158
- "language_instruction": null,
159
- "training_mode": "ROBOT_FLOWMATCH",
160
- "action_type": "EE_6D_POS",
161
- "serialize_stats": true,
162
- "video_decoder_cache_size": null,
163
- "repo_id": "tabletopsim_combined",
164
- "video_root_dir": null,
165
- "streaming": false,
166
- "tolerance_s": 0.1,
167
- "use_imagenet_stats": false,
168
- "split_name": "train",
169
- "prefetch_factor": 2,
170
- "max_episodes": null
171
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
manifest.json CHANGED
@@ -9,8 +9,8 @@
9
  "artifacts": {
10
  "policy": "policy.json",
11
  "features": "features.json",
12
- "data_config": "data_config.json",
13
- "stats": "stats.json",
14
  "training_state": null
15
  },
16
  "save_timings_seconds": {
 
9
  "artifacts": {
10
  "policy": "policy.json",
11
  "features": "features.json",
12
+ "data_config": null,
13
+ "stats": null,
14
  "training_state": null
15
  },
16
  "save_timings_seconds": {
stats.json DELETED
The diff for this file is too large to render. See raw diff