ShaswatRobotics
/

world_model_test

ShaswatRobotics commited on 20 days ago

Commit

afc888c

verified ·

1 Parent(s): eff47d8

Update ant/weights_keras/config.json

Files changed (1) hide show

ant/weights_keras/config.json CHANGED Viewed

@@ -23,11 +23,11 @@
       "framework": null,
       "format": "saved_model",
       "source": {
-        "path": "transition_concat.h5"
       },
       "signature": {
         "inputs": ["z", "a"],
-        "call_mode": "concat"
       }
     },
     {
@@ -35,11 +35,11 @@
       "framework": null,
       "format": "saved_model",
       "source": {
-        "path": "reward_concat.h5"
       },
       "signature": {
         "inputs": ["z", "a"],
-        "call_mode": "concat"
       }
     }
   ]

       "framework": null,
       "format": "saved_model",
       "source": {
+        "path": "transition_kwargs.h5"
       },
       "signature": {
         "inputs": ["z", "a"],
+        "call_mode": "kwargs"
       }
     },
     {
       "framework": null,
       "format": "saved_model",
       "source": {
+        "path": "reward_kwargs.h5"
       },
       "signature": {
         "inputs": ["z", "a"],
+        "call_mode": "kwargs"
       }
     }
   ]