Robotics
LeRobot
Safetensors
cnn
reward-model
reward_classifier
lekiwi-reward-classifier / classifier_preprocessor.json
qb1t's picture
Upload DataProcessorPipeline
0786224 verified
{
"name": "classifier_preprocessor",
"steps": [
{
"registry_name": "normalizer_processor",
"config": {
"eps": 1e-08,
"features": {
"observation.images.front": {
"type": "VISUAL",
"shape": [
3,
128,
128
]
},
"observation.images.wrist": {
"type": "VISUAL",
"shape": [
3,
128,
128
]
}
},
"norm_map": {
"VISUAL": "MEAN_STD"
}
},
"state_file": "classifier_preprocessor_step_0_normalizer_processor.safetensors"
},
{
"registry_name": "normalizer_processor",
"config": {
"eps": 1e-08,
"features": {
"next.reward": {
"type": "REWARD",
"shape": [
1
]
}
},
"norm_map": {
"VISUAL": "MEAN_STD"
}
},
"state_file": "classifier_preprocessor_step_1_normalizer_processor.safetensors"
},
{
"registry_name": "device_processor",
"config": {
"device": "mps",
"float_dtype": null
}
}
]
}