TrossenRoboticsCommunity
/

bimanual_widowxai_handover_cube

Model card Files Files and versions

shantanu-tr commited on Jul 29

Commit

996b116

·

verified ·

1 Parent(s): e90a13c

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -62,7 +62,7 @@
     "tags": null,
     "license": null,
     "chunk_size": 100,
-    "n_action_steps": 100,
     "vision_backbone": "resnet18",
     "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
     "replace_final_stride_with_dilation": false,
@@ -76,7 +76,7 @@
     "use_vae": true,
     "latent_dim": 32,
     "n_vae_encoder_layers": 4,
-    "temporal_ensemble_coeff": null,
     "dropout": 0.1,
     "kl_weight": 10.0,
     "optimizer_lr": 1e-05,

     "tags": null,
     "license": null,
     "chunk_size": 100,
+    "n_action_steps": 1,
     "vision_backbone": "resnet18",
     "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
     "replace_final_stride_with_dilation": false,
     "use_vae": true,
     "latent_dim": 32,
     "n_vae_encoder_layers": 4,
+    "temporal_ensemble_coeff": 0.01,
     "dropout": 0.1,
     "kl_weight": 10.0,
     "optimizer_lr": 1e-05,