Upload policy weights, train config and readme

config.json

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

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:514c26c587ba90c519004bac73fa59d46cd6812b2a91c0681825c227087fde1a
+oid sha256:bfda642b214cb2e14deae9f0f0c0fc1ce24151e3d5c0bc91666687ee4d5db3b3
 size 206709300

train_config.json

@@ -186,7 +186,7 @@
         "tags": null,
         "license": null,
         "chunk_size": 100,
-        "n_action_steps": 1,
+        "n_action_steps": 100,
         "vision_backbone": "resnet18",
         "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
         "replace_final_stride_with_dilation": false,
@@ -200,14 +200,14 @@
         "use_vae": true,
         "latent_dim": 32,
         "n_vae_encoder_layers": 4,
-        "temporal_ensemble_coeff": 0.01,
+        "temporal_ensemble_coeff": null,
         "dropout": 0.1,
         "kl_weight": 10.0,
         "optimizer_lr": 1e-05,
         "optimizer_weight_decay": 0.0001,
         "optimizer_lr_backbone": 1e-05
     },
-    "output_dir": "outputs/train/act_color_te_001",
+    "output_dir": "outputs/train/act_color_default",
     "job_name": "mcx_act",
     "resume": false,
     "seed": 1000,