Hugging Face's logo

microsoft
/
rho-roboeval

Safetensors
Model card Files
xet
 Community
rho-roboeval
File size: 2,810 Bytes
a4dfce3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
{
  "batch_size": 64,
  "num_workers": 4,
  "shuffle_buffer_size": 1000,
  "normalization_mapping": {
    "ACTION": "MEAN_STD",
    "ENV": "MEAN_STD",
    "STATE": "MEAN_STD",
    "VISUAL": "IDENTITY"
  },
  "features": {
    "action": {
      "type": "ACTION",
      "shape": "(20,)"
    },
    "observation.image.0": {
      "type": "VISUAL",
      "shape": "(3, 224, 224)"
    },
    "observation.image.1": {
      "type": "VISUAL",
      "shape": "(3, 224, 224)"
    },
    "observation.image.2": {
      "type": "VISUAL",
      "shape": "(3, 224, 224)"
    },
    "observation.state": {
      "type": "STATE",
      "shape": "(20,)"
    }
  },
  "clip_values": null,
  "chunk_size": null,
  "action_time_horizon_s": null,
  "min_action_chunk_size": 8,
  "actionchunk_actions_are_absolute": false,
  "observation_mapping": {
    "action.ee_6d_pos": "action",
    "observation.ee_6d_pos": "observation.state",
    "observation.images.head": "observation.image.0",
    "observation.images.left_wrist": "observation.image.1",
    "observation.images.right_wrist": "observation.image.2"
  },
  "observation_whitelist": null,
  "observation_blacklist": null,
  "transform_mapping": {
    "observation.image.0": [
      {
        "type": "color_jitter",
        "brightness": 0.2,
        "contrast": [
          0.8,
          1.2
        ],
        "saturation": [
          0.8,
          1.2
        ],
        "hue": 0.05,
        "input_type": "Tensor"
      }
    ],
    "observation.image.1": [
      {
        "type": "resize_with_padding",
        "height": 224,
        "width": 224,
        "mode": "bilinear",
        "align_corners": false,
        "input_type": "Tensor"
      },
      {
        "type": "color_jitter",
        "brightness": 0.2,
        "contrast": [
          0.8,
          1.2
        ],
        "saturation": [
          0.8,
          1.2
        ],
        "hue": 0.05,
        "input_type": "Tensor"
      }
    ],
    "observation.image.2": [
      {
        "type": "resize_with_padding",
        "height": 224,
        "width": 224,
        "mode": "bilinear",
        "align_corners": false,
        "input_type": "Tensor"
      },
      {
        "type": "color_jitter",
        "brightness": 0.2,
        "contrast": [
          0.8,
          1.2
        ],
        "saturation": [
          0.8,
          1.2
        ],
        "hue": 0.05,
        "input_type": "Tensor"
      }
    ]
  },
  "language_instruction": null,
  "training_mode": "ROBOT_FLOWMATCH",
  "action_type": "EE_6D_POS",
  "serialize_stats": true,
  "video_decoder_cache_size": null,
  "repo_id": "roboeval_lift_pot",
  "video_root_dir": null,
  "streaming": false,
  "tolerance_s": 0.1,
  "use_imagenet_stats": false,
  "split_name": "train",
  "prefetch_factor": 2,
  "max_episodes": null
}