ShaswatRobotics
/

world_model_test

Model card Files Files and versions

ShaswatRobotics commited on 15 days ago

Commit

d2c0e19

·

verified ·

1 Parent(s): c3994fe

Update ant/pwm/random_torch.py

Files changed (1) hide show

ant/pwm/random_torch.py +1 -1

ant/pwm/random_torch.py CHANGED Viewed

@@ -8,7 +8,7 @@ import torch.nn as nn
 OBS_DIM = 105
 ACTION_DIM = 8
-LATENT_DIM = 32
 HIDDEN_DIM = 256
 OUT_DIR = "weights_torch_signatures"

 OBS_DIM = 105
 ACTION_DIM = 8
+LATENT_DIM = 512
 HIDDEN_DIM = 256
 OUT_DIR = "weights_torch_signatures"