rishabhrj11
/

gym-xarm-pick

Model card Files Files and versions

rishabhrj11 commited on Nov 17, 2025

Commit

0e51a2f

·

verified ·

1 Parent(s): aebe72a

Upload DataProcessorPipeline

Files changed (1) hide show

policy_preprocessor.json +1 -9

policy_preprocessor.json CHANGED Viewed

@@ -14,7 +14,7 @@
     {
       "registry_name": "device_processor",
       "config": {
-        "device": "cuda",
         "float_dtype": null
       }
     },
@@ -43,14 +43,6 @@
               224
             ]
           },
-          "observation.image.camera1": {
-            "type": "VISUAL",
-            "shape": [
-              3,
-              224,
-              224
-            ]
-          },
           "observation.image.camera_ef": {
             "type": "VISUAL",
             "shape": [

     {
       "registry_name": "device_processor",
       "config": {
+        "device": "mps",
         "float_dtype": null
       }
     },
               224
             ]
           },
           "observation.image.camera_ef": {
             "type": "VISUAL",
             "shape": [