ShaswatRobotics
/

world_model_test

Model card Files Files and versions

ShaswatRobotics commited on 13 days ago

Commit

3e82cab

·

verified ·

1 Parent(s): b94e30c

Update ant/pwm/config.json

Files changed (1) hide show

ant/pwm/config.json +2 -4

ant/pwm/config.json CHANGED Viewed

@@ -21,11 +21,9 @@
     {
       "name": "dynamics",
       "framework": null,
-      "format": "state_dict",
       "source": {
-        "class_path": "random_torch.py",
-        "class_name": "TransitionTuple",
-        "weights_path": "transition_tuple.pth"
       },
       "signature": {
         "inputs": ["z", "a"]

     {
       "name": "dynamics",
       "framework": null,
+      "format": "jit",
       "source": {
+        "path": "transition.pt"
       },
       "signature": {
         "inputs": ["z", "a"]