Gowshigan/stackcupsv5

1{2  "name": "policy_preprocessor",3  "steps": [4    {5      "registry_name": "rename_observations_processor",6      "config": {7        "rename_map": {8          "observation.images.mount": "observation.images.image",9          "observation.images.front": "observation.images.image2"10        }11      }12    },13    {14      "registry_name": "to_batch_processor",15      "config": {}16    },17    {18      "registry_name": "tokenizer_processor",19      "config": {20        "max_length": 50,21        "task_key": "task",22        "padding_side": "right",23        "padding": "max_length",24        "truncation": true,25        "tokenizer_name": "facebook/bart-large"26      }27    },28    {29      "registry_name": "xvla_add_domain_id",30      "config": {31        "domain_id": 032      }33    },34    {35      "registry_name": "xvla_image_to_float",36      "config": {37        "image_keys": null,38        "validate_range": true39      }40    },41    {42      "registry_name": "xvla_imagenet_normalize",43      "config": {44        "image_keys": null45      }46    },47    {48      "registry_name": "device_processor",49      "config": {50        "device": "cuda",51        "float_dtype": null52      }53    },54    {55      "registry_name": "normalizer_processor",56      "config": {57        "eps": 1e-08,58        "features": {59          "observation.images.image": {60            "type": "VISUAL",61            "shape": [62              3,63              256,64              25665            ]66          },67          "observation.images.image2": {68            "type": "VISUAL",69            "shape": [70              3,71              256,72              25673            ]74          },75          "observation.state": {76            "type": "STATE",77            "shape": [78              879            ]80          },81          "observation.images.empty_camera_0": {82            "type": "VISUAL",83            "shape": [84              3,85              224,86              22487            ]88          },89          "action": {90            "type": "ACTION",91            "shape": [92              693            ]94          }95        },96        "norm_map": {97          "STATE": "IDENTITY",98          "ACTION": "MEAN_STD",99          "VISUAL": "IDENTITY"100        }101      },102      "state_file": "policy_preprocessor_step_7_normalizer_processor.safetensors"103    }104  ]105}