Robotics
LeRobot
Safetensors
eo1
eo1-put-mug / policy_preprocessor.json
autmoate's picture
Upload DataProcessorPipeline
0d5e4f0 verified
Raw
History Blame Contribute Delete
2.56 kB
{
"name": "policy_preprocessor",
"steps": [
{
"registry_name": "rename_observations_processor",
"config": {
"rename_map": {}
}
},
{
"registry_name": "to_batch_processor",
"config": {}
},
{
"registry_name": "normalizer_processor",
"config": {
"eps": 1e-08,
"features": {
"observation.state": {
"type": "STATE",
"shape": [
6
]
},
"observation.images.top": {
"type": "VISUAL",
"shape": [
3,
720,
1280
]
},
"observation.images.side": {
"type": "VISUAL",
"shape": [
3,
720,
1280
]
},
"observation.images.wrist": {
"type": "VISUAL",
"shape": [
3,
720,
1280
]
},
"action": {
"type": "ACTION",
"shape": [
6
]
}
},
"norm_map": {
"VISUAL": "IDENTITY",
"STATE": "MEAN_STD",
"ACTION": "MEAN_STD"
}
},
"state_file": "policy_preprocessor_step_2_normalizer_processor.safetensors"
},
{
"registry_name": "eo1_conversation_template_processor",
"config": {
"input_features": {
"observation.state": {
"type": "STATE",
"shape": [
6
]
},
"observation.images.top": {
"type": "VISUAL",
"shape": [
3,
720,
1280
]
},
"observation.images.side": {
"type": "VISUAL",
"shape": [
3,
720,
1280
]
},
"observation.images.wrist": {
"type": "VISUAL",
"shape": [
3,
720,
1280
]
}
},
"chunk_size": 8
}
},
{
"registry_name": "eo1_qwen_processor",
"config": {
"processor_name": "Qwen/Qwen2.5-VL-3B-Instruct",
"image_min_pixels": 50176,
"image_max_pixels": 50176,
"use_fast_processor": true
}
},
{
"registry_name": "device_processor",
"config": {
"device": "cuda",
"float_dtype": null
}
}
]
}