Update ant/pwm_torch_seperate/config.json

ant/pwm_torch_seperate/config.json

@@ -3,7 +3,7 @@
   "env": "Ant-v5",
   "model_type": "pwm",
   "metadata": {
-    "latent_dim": 512
+    "latent_dim": 32
   },
 
   "models": [
@@ -14,7 +14,7 @@
       "source": {
         "class_path": "random_torch.py",
         "class_name": "Encoder",
-        "class_args": [105, 512, 256],
+        "class_args": [105, 32, 256],
         "weights_path": "encoder.pth"
       },
       "signature": {
@@ -28,7 +28,7 @@
       "source": {
         "class_path": "random_torch.py",
         "class_name": "TransitionModel",
-        "class_args": [512, 8, 256],
+        "class_args": [32, 8, 256],
         "weights_path": "transition.pth"
       },
       "signature": {
@@ -42,7 +42,7 @@
       "source": {
         "class_path": "random_torch.py",
         "class_name": "RewardModel",
-        "class_args": [512, 8, 256],
+        "class_args": [32, 8, 256],
         "weights_path": "reward.pth"
       },
       "signature": {