Hugging Face's logo

slobot
/
aic-act

Safetensors
Model card Files
xet
 Community
alexis779 commited on
Commit
babd3c2
·
verified ·
1 Parent(s): 19a9a30

Upload folder using huggingface_hub

Browse files
Files changed (6) hide show
  1. config.json +86 -0
  2. model.safetensors +3 -0
  3. policy_postprocessor.json +37 -0
  4. policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
  5. policy_preprocessor.json +83 -0
  6. policy_preprocessor_step_4_normalizer_processor.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,86 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "act",
3
+ "n_obs_steps": 1,
4
+ "input_features": {
5
+ "observation.images.left_camera": {
6
+ "type": "VISUAL",
7
+ "shape": [
8
+ 3,
9
+ 256,
10
+ 288
11
+ ]
12
+ },
13
+ "observation.images.center_camera": {
14
+ "type": "VISUAL",
15
+ "shape": [
16
+ 3,
17
+ 256,
18
+ 288
19
+ ]
20
+ },
21
+ "observation.images.right_camera": {
22
+ "type": "VISUAL",
23
+ "shape": [
24
+ 3,
25
+ 256,
26
+ 288
27
+ ]
28
+ },
29
+ "observation.state": {
30
+ "type": "STATE",
31
+ "shape": [
32
+ 6
33
+ ]
34
+ },
35
+ "observation.environment_state": {
36
+ "type": "ENV",
37
+ "shape": [
38
+ 13
39
+ ]
40
+ }
41
+ },
42
+ "output_features": {
43
+ "action": {
44
+ "type": "ACTION",
45
+ "shape": [
46
+ 6
47
+ ]
48
+ }
49
+ },
50
+ "device": "cuda",
51
+ "use_amp": false,
52
+ "use_peft": false,
53
+ "push_to_hub": true,
54
+ "repo_id": null,
55
+ "private": null,
56
+ "tags": null,
57
+ "license": null,
58
+ "pretrained_path": null,
59
+ "chunk_size": 100,
60
+ "n_action_steps": 100,
61
+ "normalization_mapping": {
62
+ "VISUAL": "MEAN_STD",
63
+ "STATE": "MEAN_STD",
64
+ "ACTION": "MEAN_STD",
65
+ "ENV": "IDENTITY"
66
+ },
67
+ "vision_backbone": "resnet18",
68
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
69
+ "replace_final_stride_with_dilation": false,
70
+ "pre_norm": false,
71
+ "dim_model": 512,
72
+ "n_heads": 8,
73
+ "dim_feedforward": 3200,
74
+ "feedforward_activation": "relu",
75
+ "n_encoder_layers": 4,
76
+ "n_decoder_layers": 1,
77
+ "use_vae": true,
78
+ "latent_dim": 32,
79
+ "n_vae_encoder_layers": 4,
80
+ "temporal_ensemble_coeff": null,
81
+ "dropout": 0.1,
82
+ "kl_weight": 10.0,
83
+ "optimizer_lr": 1e-05,
84
+ "optimizer_weight_decay": 0.0001,
85
+ "optimizer_lr_backbone": 1e-05
86
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc737d2edd1d2a6b7f94f8431d32b92c9d9802d6178b256ad0a72ea6e22942b1
3
+ size 206730680
policy_postprocessor.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_postprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "unnormalizer_processor",
6
+ "config": {
7
+ "eps": 1e-08,
8
+ "features": {
9
+ "action": {
10
+ "type": "ACTION",
11
+ "shape": [
12
+ 6
13
+ ]
14
+ }
15
+ },
16
+ "norm_map": {
17
+ "VISUAL": "MEAN_STD",
18
+ "STATE": "MEAN_STD",
19
+ "ACTION": "MEAN_STD",
20
+ "ENV": "IDENTITY"
21
+ }
22
+ },
23
+ "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
24
+ },
25
+ {
26
+ "registry_name": "device_processor",
27
+ "config": {
28
+ "device": "cpu",
29
+ "float_dtype": null
30
+ }
31
+ },
32
+ {
33
+ "class": "__main__.AicActPostprocessStep",
34
+ "config": {}
35
+ }
36
+ ]
37
+ }
policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f08f651815ef7fbf27defc4ec8b7dfdf1218a39453f29e1d66e6c2e210cd267
3
+ size 24576
policy_preprocessor.json ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_preprocessor",
3
+ "steps": [
4
+ {
5
+ "class": "__main__.AicActPreprocessStep",
6
+ "config": {}
7
+ },
8
+ {
9
+ "registry_name": "rename_observations_processor",
10
+ "config": {
11
+ "rename_map": {}
12
+ }
13
+ },
14
+ {
15
+ "registry_name": "to_batch_processor",
16
+ "config": {}
17
+ },
18
+ {
19
+ "registry_name": "device_processor",
20
+ "config": {
21
+ "device": "cuda",
22
+ "float_dtype": null
23
+ }
24
+ },
25
+ {
26
+ "registry_name": "normalizer_processor",
27
+ "config": {
28
+ "eps": 1e-08,
29
+ "features": {
30
+ "observation.images.left_camera": {
31
+ "type": "VISUAL",
32
+ "shape": [
33
+ 3,
34
+ 256,
35
+ 288
36
+ ]
37
+ },
38
+ "observation.images.center_camera": {
39
+ "type": "VISUAL",
40
+ "shape": [
41
+ 3,
42
+ 256,
43
+ 288
44
+ ]
45
+ },
46
+ "observation.images.right_camera": {
47
+ "type": "VISUAL",
48
+ "shape": [
49
+ 3,
50
+ 256,
51
+ 288
52
+ ]
53
+ },
54
+ "observation.state": {
55
+ "type": "STATE",
56
+ "shape": [
57
+ 6
58
+ ]
59
+ },
60
+ "observation.environment_state": {
61
+ "type": "ENV",
62
+ "shape": [
63
+ 13
64
+ ]
65
+ },
66
+ "action": {
67
+ "type": "ACTION",
68
+ "shape": [
69
+ 6
70
+ ]
71
+ }
72
+ },
73
+ "norm_map": {
74
+ "VISUAL": "MEAN_STD",
75
+ "STATE": "MEAN_STD",
76
+ "ACTION": "MEAN_STD",
77
+ "ENV": "IDENTITY"
78
+ }
79
+ },
80
+ "state_file": "policy_preprocessor_step_4_normalizer_processor.safetensors"
81
+ }
82
+ ]
83
+ }
policy_preprocessor_step_4_normalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f08f651815ef7fbf27defc4ec8b7dfdf1218a39453f29e1d66e6c2e210cd267
3
+ size 24576