Songyuan666/robot_learning_tutorial_diffusion_example_model

1{
2  "name": "policy_postprocessor",
3  "steps": [
4    {
5      "registry_name": "unnormalizer_processor",
6      "config": {
7        "eps": 1e-08,
8        "features": {
9          "action": {
10            "type": "ACTION",
11            "shape": [
12              6
13            ]
14          }
15        },
16        "norm_map": {
17          "VISUAL": "MEAN_STD",
18          "STATE": "MIN_MAX",
19          "ACTION": "MIN_MAX"
20        }
21      },
22      "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
23    },
24    {
25      "registry_name": "device_processor",
26      "config": {
27        "device": "cpu",
28        "float_dtype": null
29      }
30    }
31  ]
32}