ISdept
/

transformer-diffusion

Model card Files Files and versions

ISdept commited on 29 days ago

Commit

334a581

·

verified ·

1 Parent(s): 77e6d50

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -56,7 +56,7 @@
     "tags": null,
     "license": null,
     "pretrained_path": null,
-    "horizon": 50,
     "n_action_steps": 8,
     "normalization_mapping": {
         "VISUAL": "IDENTITY",
@@ -85,7 +85,7 @@
     "dim_feedforward": 2048,
     "diffusion_step_embed_dim": 128,
     "num_inference_steps": 20,
-    "optimizer_lr": 0.0001,
     "optimizer_betas": [
         0.95,
         0.999

     "tags": null,
     "license": null,
     "pretrained_path": null,
+    "horizon": 16,
     "n_action_steps": 8,
     "normalization_mapping": {
         "VISUAL": "IDENTITY",
     "dim_feedforward": 2048,
     "diffusion_step_embed_dim": 128,
     "num_inference_steps": 20,
+    "optimizer_lr": 0.00001,
     "optimizer_betas": [
         0.95,
         0.999