Hugging Face's logo

anonymousExRL
/
exrl-base-policies

Reinforcement Learning
Safetensors
robotics
robomimic
Model card Files
xet
 Community
anonymousExRL commited on
Commit
a40e9a4
·
verified ·
1 Parent(s): 1aef7e9

Upload Robomimic ACT can ta50 base policy

Browse files
Files changed (7) hide show
  1. robomimic/act/can_ta50/config.json +62 -0
  2. robomimic/act/can_ta50/metadata.json +7 -0
  3. robomimic/act/can_ta50/model.safetensors +3 -0
  4. robomimic/act/can_ta50/policy_postprocessor.json +33 -0
  5. robomimic/act/can_ta50/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
  6. robomimic/act/can_ta50/policy_preprocessor.json +55 -0
  7. robomimic/act/can_ta50/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
robomimic/act/can_ta50/config.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "n_obs_steps": 1,
3
+ "input_features": {
4
+ "observation.state": {
5
+ "type": "STATE",
6
+ "shape": [
7
+ 9
8
+ ]
9
+ },
10
+ "observation.environment_state": {
11
+ "type": "ENV",
12
+ "shape": [
13
+ 14
14
+ ]
15
+ }
16
+ },
17
+ "output_features": {
18
+ "action": {
19
+ "type": "ACTION",
20
+ "shape": [
21
+ 7
22
+ ]
23
+ }
24
+ },
25
+ "device": "cuda",
26
+ "use_amp": false,
27
+ "use_peft": false,
28
+ "push_to_hub": false,
29
+ "repo_id": null,
30
+ "private": null,
31
+ "tags": null,
32
+ "license": null,
33
+ "pretrained_path": null,
34
+ "chunk_size": 50,
35
+ "n_action_steps": 50,
36
+ "normalization_mapping": {
37
+ "VISUAL": "MEAN_STD",
38
+ "STATE": "MEAN_STD",
39
+ "ACTION": "MEAN_STD",
40
+ "ENV": "MEAN_STD"
41
+ },
42
+ "vision_backbone": "resnet18",
43
+ "pretrained_backbone_weights": null,
44
+ "replace_final_stride_with_dilation": false,
45
+ "pre_norm": false,
46
+ "dim_model": 512,
47
+ "n_heads": 8,
48
+ "dim_feedforward": 3200,
49
+ "feedforward_activation": "relu",
50
+ "n_encoder_layers": 4,
51
+ "n_decoder_layers": 1,
52
+ "use_vae": false,
53
+ "latent_dim": 32,
54
+ "n_vae_encoder_layers": 4,
55
+ "temporal_ensemble_coeff": null,
56
+ "dropout": 0.0,
57
+ "kl_weight": 10.0,
58
+ "optimizer_lr": 0.0001,
59
+ "optimizer_weight_decay": 1e-06,
60
+ "optimizer_lr_backbone": 1e-05,
61
+ "type": "act"
62
+ }
robomimic/act/can_ta50/metadata.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "base_policy": "act",
3
+ "checkpoint_role": "base_policy",
4
+ "chunk_size": 50,
5
+ "suite": "robomimic",
6
+ "task": "can"
7
+ }
robomimic/act/can_ta50/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a5894ac386aebdf2c56b21a99eecf1d5e2b05d937278f74bf08d4fd2d6693a6d
3
+ size 91124476
robomimic/act/can_ta50/policy_postprocessor.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_postprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "unnormalizer_processor",
6
+ "config": {
7
+ "features": {
8
+ "action": {
9
+ "type": "ACTION",
10
+ "shape": [
11
+ 7
12
+ ]
13
+ }
14
+ },
15
+ "norm_map": {
16
+ "VISUAL": "MEAN_STD",
17
+ "STATE": "MEAN_STD",
18
+ "ACTION": "MEAN_STD",
19
+ "ENV": "MEAN_STD"
20
+ },
21
+ "eps": 1e-08
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
+ }
robomimic/act/can_ta50/policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5ca6f57601c51f152b0952bb7ec8f045d7dca3c49500f4d83f089157d0a5b531
3
+ size 1416
robomimic/act/can_ta50/policy_preprocessor.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_preprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "rename_observations_processor",
6
+ "config": {
7
+ "rename_map": {}
8
+ }
9
+ },
10
+ {
11
+ "registry_name": "to_batch_processor",
12
+ "config": {}
13
+ },
14
+ {
15
+ "registry_name": "device_processor",
16
+ "config": {
17
+ "device": "cuda",
18
+ "float_dtype": null
19
+ }
20
+ },
21
+ {
22
+ "registry_name": "normalizer_processor",
23
+ "config": {
24
+ "features": {
25
+ "observation.state": {
26
+ "type": "STATE",
27
+ "shape": [
28
+ 9
29
+ ]
30
+ },
31
+ "observation.environment_state": {
32
+ "type": "ENV",
33
+ "shape": [
34
+ 14
35
+ ]
36
+ },
37
+ "action": {
38
+ "type": "ACTION",
39
+ "shape": [
40
+ 7
41
+ ]
42
+ }
43
+ },
44
+ "norm_map": {
45
+ "VISUAL": "MEAN_STD",
46
+ "STATE": "MEAN_STD",
47
+ "ACTION": "MEAN_STD",
48
+ "ENV": "MEAN_STD"
49
+ },
50
+ "eps": 1e-08
51
+ },
52
+ "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
53
+ }
54
+ ]
55
+ }
robomimic/act/can_ta50/policy_preprocessor_step_3_normalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5ca6f57601c51f152b0952bb7ec8f045d7dca3c49500f4d83f089157d0a5b531
3
+ size 1416