ShaswatRobotics
/

world_model_test

ShaswatRobotics commited on 26 days ago

Commit

3c9fa98

verified ·

1 Parent(s): cd9e569

Update ant/pwm_torch_seperate/config.json

Files changed (1) hide show

ant/pwm_torch_seperate/config.json CHANGED Viewed

@@ -10,7 +10,7 @@
     {
       "name": "encoder",
       "framework": null,
-      "format": "jit",
       "source": {
         "class_path": "random_torch.py",
         "class_name": "Encoder",
@@ -23,7 +23,7 @@
     {
       "name": "dynamics",
       "framework": null,
-      "format": "jit",
       "source": {
         "class_path": "random_torch.py",
         "class_name": "TransitionModel",
@@ -36,7 +36,7 @@
     {
       "name": "reward",
       "framework": null,
-      "format": "jit",
       "source": {
         "class_path": "random_torch.py",
         "class_name": "RewardModel",

     {
       "name": "encoder",
       "framework": null,
+      "format": "state_dict",
       "source": {
         "class_path": "random_torch.py",
         "class_name": "Encoder",
     {
       "name": "dynamics",
       "framework": null,
+      "format": "state_dict",
       "source": {
         "class_path": "random_torch.py",
         "class_name": "TransitionModel",
     {
       "name": "reward",
       "framework": null,
+      "format": "state_dict",
       "source": {
         "class_path": "random_torch.py",
         "class_name": "RewardModel",