dita_droid_pretrain / policy_postprocessor.json
albus2024's picture
Add binary/continuous gripper modes with DITA runtime 2.0.0
219a93a verified
Raw History Blame Contribute Delete
629 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": "continuous"
}
},
{
"registry_name": "device_processor",
"config": {
"device": "cpu",
"float_dtype": null
}
}
]
}