dita_goal / policy_postprocessor.json
albus2024's picture
Add binary/continuous gripper modes with DITA runtime 2.0.0
7c2bf39 verified
Raw History Blame Contribute Delete
723 Bytes
{
"name": "policy_postprocessor",
"steps": [
{
"registry_name": "dita_masked_unnormalizer_processor",
"config": {
"eps": 1e-08,
"features": {
"action": {
"type": "ACTION",
"shape": [
7
]
}
},
"norm_map": {
"VISUAL": "IDENTITY",
"ACTION": "QUANTILES",
"LANGUAGE": "IDENTITY"
},
"gripper_mode": "binary"
},
"state_file": "policy_postprocessor_step_0_dita_masked_unnormalizer_processor.safetensors"
},
{
"registry_name": "device_processor",
"config": {
"device": "cpu",
"float_dtype": null
}
}
]
}