TheDoc273/multitask_dit

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": "tokenizer_processor",16      "config": {17        "max_length": 77,18        "task_key": "task",19        "padding_side": "right",20        "padding": "max_length",21        "truncation": true,22        "tokenizer_name": "openai/clip-vit-base-patch16"23      }24    },25    {26      "registry_name": "device_processor",27      "config": {28        "device": "cuda",29        "float_dtype": null30      }31    },32    {33      "registry_name": "normalizer_processor",34      "config": {35        "eps": 1e-08,36        "features": {37          "observation.state": {38            "type": "STATE",39            "shape": [40              641            ]42          },43          "observation.images.front": {44            "type": "VISUAL",45            "shape": [46              3,47              720,48              128049            ]50          },51          "action": {52            "type": "ACTION",53            "shape": [54              655            ]56          }57        },58        "norm_map": {59          "VISUAL": "MEAN_STD",60          "STATE": "MIN_MAX",61          "ACTION": "MIN_MAX"62        }63      },64      "state_file": "policy_preprocessor_step_4_normalizer_processor.safetensors"65    }66  ]67}