Upload folder using huggingface_hub

README.md

@@ -4,4 +4,4 @@ tags:
 - pytorch_model_hub_mixin
 ---
 
-ACT trained with [pusht_keypoints](https://huggingface.co/datasets/lerobot/pusht_keypoints). Achieves 45% success rate for 500 evals. See [eval_info.json](./eval_info.json) for more evaluation data.
+ACT trained with [pusht_keypoints](https://huggingface.co/datasets/lerobot/pusht_keypoints). Achieves 64% success rate for 500 evals. See [eval_info.json](./eval_info.json) for more evaluation data.

config.json

@@ -19,7 +19,7 @@
   "kl_weight": 10.0,
   "latent_dim": 32,
   "n_action_steps": 16,
-  "n_decoder_layers": 1,
+  "n_decoder_layers": 4,
   "n_encoder_layers": 4,
   "n_heads": 8,
   "n_obs_steps": 1,

config.yaml

@@ -13,7 +13,7 @@ training:
   eval_freq: 10000
   log_freq: 250
   save_checkpoint: true
-  save_freq: 10000
+  save_freq: 25000
   num_workers: 8
   batch_size: 64
   image_transforms:
@@ -117,7 +117,7 @@ policy:
   dim_feedforward: 3200
   feedforward_activation: relu
   n_encoder_layers: 4
-  n_decoder_layers: 1
+  n_decoder_layers: 4
   use_vae: true
   latent_dim: 32
   n_vae_encoder_layers: 4

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:14f6ce7d2ec44fb204f1c6bfe2f99e14ff1d416e830cba079e65ce1766923855
-size 160555264
+oid sha256:79ba01505ba47db55f3b41089502ea2608777b04986778b5dd5afa6f4c538faf
+size 225179336