CoolFace
Modelpublic

Cache-SCA/Pi0.5-IsaacLab-Multi-Task-1epochs

sourceHugging Facegemmaupdated 4mo agoView on Hugging Face
0likes8downloads
policy_preprocessor.json103 linesDownload Raw Back to root
1{2  "name": "policy_preprocessor",3  "steps": [4    {5      "registry_name": "rename_observations_processor",6      "config": {7        "rename_map": {}8      }9    },10    {11      "registry_name": "to_batch_processor",12      "config": {}13    },14    {15      "registry_name": "normalizer_processor",16      "config": {17        "eps": 1e-08,18        "features": {19          "observation.images.base_0_rgb": {20            "type": "STATE",21            "shape": [22              123            ]24          },25          "observation.images.left_wrist_0_rgb": {26            "type": "STATE",27            "shape": [28              129            ]30          },31          "observation.images.right_wrist_0_rgb": {32            "type": "STATE",33            "shape": [34              135            ]36          },37          "observation.state": {38            "type": "STATE",39            "shape": [40              3241            ]42          },43          "observation.images.top": {44            "type": "VISUAL",45            "shape": [46              3,47              480,48              64049            ]50          },51          "observation.images.left_wrist": {52            "type": "VISUAL",53            "shape": [54              3,55              480,56              64057            ]58          },59          "action": {60            "type": "ACTION",61            "shape": [62              663            ]64          },65          "action.radian_urdf0": {66            "type": "ACTION",67            "shape": [68              669            ]70          }71        },72        "norm_map": {73          "VISUAL": "IDENTITY",74          "STATE": "QUANTILES",75          "ACTION": "QUANTILES"76        }77      },78      "state_file": "policy_preprocessor_step_2_normalizer_processor.safetensors"79    },80    {81      "registry_name": "pi05_prepare_state_tokenizer_processor_step",82      "config": {}83    },84    {85      "registry_name": "tokenizer_processor",86      "config": {87        "max_length": 200,88        "task_key": "task",89        "padding_side": "right",90        "padding": "max_length",91        "truncation": true,92        "tokenizer_name": "google/paligemma-3b-pt-224"93      }94    },95    {96      "registry_name": "device_processor",97      "config": {98        "device": "cuda",99        "float_dtype": null100      }101    }102  ]103}