ShaswatRobotics
/

world_model_test

Model card Files Files and versions

ShaswatRobotics commited on 16 days ago

Commit

c3994fe

·

verified ·

1 Parent(s): 03dfc1f

Update ant/pwm/config.json

Files changed (1) hide show

ant/pwm/config.json +4 -2

ant/pwm/config.json CHANGED Viewed

@@ -21,9 +21,11 @@
     {
       "name": "dynamics",
       "framework": null,
-      "format": "jit",
       "source": {
-        "path": "transition_tuple.pt"
       },
       "signature": {
         "inputs": ["z", "a"]

     {
       "name": "dynamics",
       "framework": null,
+      "format": "state_dict",
       "source": {
+        "class_path": "random_torch.py",
+        "class_name": "TransitionConcat",
+        "weights_path": "transition_concat.pth"
       },
       "signature": {
         "inputs": ["z", "a"]