Upload folder using huggingface_hub

README.md

@@ -1,7 +1,7 @@
 ---
 tags:
-- pytorch_model_hub_mixin
 - model_hub_mixin
+- pytorch_model_hub_mixin
 ---
 
 This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:

config.json

@@ -45,5 +45,5 @@
   "vision_backbone": "resnet18",
   "vqvae_embedding_dim": 32,
   "vqvae_enc_hidden_dim": 64,
-  "vqvae_n_embed": 16
+  "vqvae_n_embed": 8
 }

config.yaml

@@ -10,7 +10,7 @@ training:
   online_steps_between_rollouts: 1
   online_sampling_ratio: 0.5
   online_env_seed: ???
-  eval_freq: 10000
+  eval_freq: 5000
   log_freq: 250
   save_checkpoint: true
   save_freq: 5000
@@ -72,8 +72,8 @@ training:
     - 0.03333333333333333
     - 0.05
 eval:
-  n_episodes: 10
-  batch_size: 10
+  n_episodes: 50
+  batch_size: 50
   use_async_envs: false
 wandb:
   enable: true
@@ -116,7 +116,7 @@ policy:
   use_group_norm: true
   spatial_softmax_num_keypoints: 32
   n_vqvae_training_steps: ${training.n_vqvae_training_steps}
-  vqvae_n_embed: 16
+  vqvae_n_embed: 8
   vqvae_embedding_dim: 32
   vqvae_enc_hidden_dim: 64
   gpt_block_size: 500

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7a36938231e0f1c2993f1e2c513073b59396e20dd03bef3b1bce73b44c03206c
-size 17571450
+oid sha256:0036579203832939fc018d39842891a089d49675f9f485723bdd5f2b3dd271b2
+size 17525194