Hugging Face's logo

ShaswatRobotics
/
world_model_test

TF-Keras
ONNX
Model card Files
xet
 Community
ShaswatRobotics commited on
Commit
d2c0e19
·
verified ·
1 Parent(s): c3994fe

Update ant/pwm/random_torch.py

Browse files
Files changed (1) hide show
  1. ant/pwm/random_torch.py +1 -1
ant/pwm/random_torch.py CHANGED
@@ -8,7 +8,7 @@ import torch.nn as nn
8
 
9
  OBS_DIM = 105
10
  ACTION_DIM = 8
11
- LATENT_DIM = 32
12
  HIDDEN_DIM = 256
13
 
14
  OUT_DIR = "weights_torch_signatures"
 
8
 
9
  OBS_DIM = 105
10
  ACTION_DIM = 8
11
+ LATENT_DIM = 512
12
  HIDDEN_DIM = 256
13
 
14
  OUT_DIR = "weights_torch_signatures"