quincyu
/

trace_robotwin

Model card Files Files and versions

quincyu commited on Feb 18

Commit

9a4ee9e

·

verified ·

1 Parent(s): 2129da9

Upload DataProcessorPipeline

Files changed (1) hide show

policy_preprocessor.json +3 -7

policy_preprocessor.json CHANGED Viewed

@@ -7,10 +7,6 @@
         "rename_map": {}
       }
     },
-    {
-      "class": "lerobot_policy_tg_diffusion.processor_tg_diffusion.EePixelClipNormalizeStep",
-      "config": {}
-    },
     {
       "registry_name": "to_batch_processor",
       "config": {}
@@ -18,7 +14,7 @@
     {
       "registry_name": "device_processor",
       "config": {
-        "device": "cuda",
         "float_dtype": null
       }
     },
@@ -66,11 +62,11 @@
         },
         "norm_map": {
           "VISUAL": "MEAN_STD",
-          "STATE": "MIN_MAX",
           "ACTION": "MIN_MAX"
         }
       },
-      "state_file": "policy_preprocessor_step_4_normalizer_processor.safetensors"
     }
   ]
 }

         "rename_map": {}
       }
     },
     {
       "registry_name": "to_batch_processor",
       "config": {}
     {
       "registry_name": "device_processor",
       "config": {
+        "device": "cpu",
         "float_dtype": null
       }
     },
         },
         "norm_map": {
           "VISUAL": "MEAN_STD",
+          "STATE": "MEAN_STD",
           "ACTION": "MIN_MAX"
         }
       },
+      "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
     }
   ]
 }