CarlCochet
/

trajectory-transformer-halfcheetah-medium-v2

Feature Extraction

trajectory_transformer

Inference Endpoints

Model card Files Files and versions Community

CarlCochet commited on May 6, 2022

Commit

5c583ae

•

1 Parent(s): 68d15c4

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -17,7 +17,7 @@
   "layer_norm_epsilon": 1e-05,
   "model_type": "trajectory_transformer",
   "n_ctx": 1024,
-  "n_embd": 32,
   "n_epochs_ref": 50,
   "n_head": 4,
   "n_layer": 4,

   "layer_norm_epsilon": 1e-05,
   "model_type": "trajectory_transformer",
   "n_ctx": 1024,
+  "n_embd": 128,
   "n_epochs_ref": 50,
   "n_head": 4,
   "n_layer": 4,