aliberts HF staff commited on
Commit
a05e28d
·
verified ·
1 Parent(s): 113c0be

Push model using huggingface_hub.

Browse files
Files changed (2) hide show
  1. README.md +8 -0
  2. config.json +28 -0
README.md ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - model_hub_mixin
4
+ ---
5
+
6
+ This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
7
+ - Library: [More Information Needed]
8
+ - Docs: [More Information Needed]
config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "act",
3
+ "n_obs_steps": 1,
4
+ "normalization_mapping": {
5
+ "ACTION": "MEAN_STD",
6
+ "STATE": "MEAN_STD",
7
+ "VISUAL": "MEAN_STD"
8
+ },
9
+ "chunk_size": 20,
10
+ "n_action_steps": 20,
11
+ "vision_backbone": "resnet18",
12
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
13
+ "replace_final_stride_with_dilation": 0,
14
+ "pre_norm": false,
15
+ "dim_model": 64,
16
+ "n_heads": 8,
17
+ "dim_feedforward": 3200,
18
+ "feedforward_activation": "relu",
19
+ "n_encoder_layers": 4,
20
+ "n_decoder_layers": 1,
21
+ "use_vae": true,
22
+ "latent_dim": 32,
23
+ "n_vae_encoder_layers": 4,
24
+ "temporal_ensemble_coeff": null,
25
+ "dropout": 0.1,
26
+ "kl_weight": 10.0,
27
+ "optimizer_lr_backbone": 1e-05
28
+ }