AdilZtn commited on
Commit
7c8d1de
·
verified ·
1 Parent(s): f99f362

Upload policy

Browse files
Files changed (2) hide show
  1. config.json +69 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "act",
3
+ "n_obs_steps": 1,
4
+ "input_features": {
5
+ "observation.state": {
6
+ "type": "STATE",
7
+ "shape": [
8
+ 6
9
+ ]
10
+ },
11
+ "observation.images.top": {
12
+ "type": "VISUAL",
13
+ "shape": [
14
+ 3,
15
+ 480,
16
+ 640
17
+ ]
18
+ },
19
+ "observation.images.wrist": {
20
+ "type": "VISUAL",
21
+ "shape": [
22
+ 3,
23
+ 480,
24
+ 640
25
+ ]
26
+ }
27
+ },
28
+ "output_features": {
29
+ "action": {
30
+ "type": "ACTION",
31
+ "shape": [
32
+ 6
33
+ ]
34
+ }
35
+ },
36
+ "device": "cpu",
37
+ "use_amp": false,
38
+ "push_to_hub": true,
39
+ "repo_id": null,
40
+ "private": null,
41
+ "tags": null,
42
+ "license": null,
43
+ "chunk_size": 100,
44
+ "n_action_steps": 10,
45
+ "normalization_mapping": {
46
+ "VISUAL": "MEAN_STD",
47
+ "STATE": "MEAN_STD",
48
+ "ACTION": "MEAN_STD"
49
+ },
50
+ "vision_backbone": "resnet18",
51
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
52
+ "replace_final_stride_with_dilation": false,
53
+ "pre_norm": false,
54
+ "dim_model": 512,
55
+ "n_heads": 8,
56
+ "dim_feedforward": 3200,
57
+ "feedforward_activation": "relu",
58
+ "n_encoder_layers": 4,
59
+ "n_decoder_layers": 1,
60
+ "use_vae": false,
61
+ "latent_dim": 32,
62
+ "n_vae_encoder_layers": 4,
63
+ "temporal_ensemble_coeff": null,
64
+ "dropout": 0.1,
65
+ "kl_weight": 10.0,
66
+ "optimizer_lr": 1e-05,
67
+ "optimizer_weight_decay": 0.0001,
68
+ "optimizer_lr_backbone": 1e-05
69
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f7146c7c7229e127bbde4d4e395755016da164455e8b793c97317ee145fc48f9
3
+ size 136991456