Hugging Face's logo

eejjii
/
acm2_block_sorting_2

Robotics
LeRobot
Safetensors
acm2
Model card Files
xet
 Community
eejjii commited on
Commit
0469f44
·
verified ·
1 Parent(s): 9d84f64

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +73 -72
config.json CHANGED
@@ -1,76 +1,77 @@
1
  {
2
- "n_obs_steps": 1,
3
- "input_features": {
4
- "observation.state": {
5
- "type": "STATE",
6
- "shape": [
7
- 6
8
- ]
9
- },
10
- "observation.images.front": {
11
- "type": "VISUAL",
12
- "shape": [
13
- 3,
14
- 480,
15
- 640
16
- ]
17
- },
18
- "observation.images.top": {
19
- "type": "VISUAL",
20
- "shape": [
21
- 3,
22
- 480,
23
- 640
24
- ]
25
- }
26
  },
27
- "output_features": {
28
- "action": {
29
- "type": "ACTION",
30
- "shape": [
31
- 6
32
- ]
33
- }
34
  },
35
- "device": "cuda",
36
- "use_amp": false,
37
- "push_to_hub": true,
38
- "repo_id": "eejjii/acm2_block_sorting_2",
39
- "private": null,
40
- "tags": null,
41
- "license": null,
42
- "pretrained_path": null,
43
- "chunk_size": 100,
44
- "n_action_steps": 100,
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
- "mamba2_d_state": 128,
61
- "mamba2_d_conv": 4,
62
- "mamba2_expand": 2,
63
- "mamba2_headdim": 64,
64
- "mamba2_chunk_size": 256,
65
- "use_temporal_weighting": false,
66
- "temporal_execution_weight": 0.9,
67
- "use_vae": true,
68
- "latent_dim": 32,
69
- "n_vae_encoder_layers": 4,
70
- "temporal_ensemble_coeff": null,
71
- "dropout": 0.1,
72
- "kl_weight": 10.0,
73
- "optimizer_lr": 3e-05,
74
- "optimizer_weight_decay": 0.0001,
75
- "optimizer_lr_backbone": 1e-05
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
76
  }
 
1
  {
2
+ "type": "acm2",
3
+ "n_obs_steps": 1,
4
+ "input_features": {
5
+ "observation.state": {
6
+ "type": "STATE",
7
+ "shape": [
8
+ 6
9
+ ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  },
11
+ "observation.images.front": {
12
+ "type": "VISUAL",
13
+ "shape": [
14
+ 3,
15
+ 480,
16
+ 640
17
+ ]
18
  },
19
+ "observation.images.top": {
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": "cuda",
37
+ "use_amp": false,
38
+ "push_to_hub": true,
39
+ "repo_id": "eejjii/acm2_block_sorting_2",
40
+ "private": null,
41
+ "tags": null,
42
+ "license": null,
43
+ "pretrained_path": null,
44
+ "chunk_size": 100,
45
+ "n_action_steps": 100,
46
+ "normalization_mapping": {
47
+ "VISUAL": "MEAN_STD",
48
+ "STATE": "MEAN_STD",
49
+ "ACTION": "MEAN_STD"
50
+ },
51
+ "vision_backbone": "resnet18",
52
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
53
+ "replace_final_stride_with_dilation": false,
54
+ "pre_norm": false,
55
+ "dim_model": 512,
56
+ "n_heads": 8,
57
+ "dim_feedforward": 3200,
58
+ "feedforward_activation": "relu",
59
+ "n_encoder_layers": 4,
60
+ "n_decoder_layers": 1,
61
+ "mamba2_d_state": 128,
62
+ "mamba2_d_conv": 4,
63
+ "mamba2_expand": 2,
64
+ "mamba2_headdim": 64,
65
+ "mamba2_chunk_size": 256,
66
+ "use_temporal_weighting": false,
67
+ "temporal_execution_weight": 0.9,
68
+ "use_vae": true,
69
+ "latent_dim": 32,
70
+ "n_vae_encoder_layers": 4,
71
+ "temporal_ensemble_coeff": null,
72
+ "dropout": 0.1,
73
+ "kl_weight": 10.0,
74
+ "optimizer_lr": 3e-05,
75
+ "optimizer_weight_decay": 0.0001,
76
+ "optimizer_lr_backbone": 1e-05
77
  }