Hugging Face's logo

SieloRobotics
/
Diffusion_ee_chunk_bottle

Robotics
LeRobot
Safetensors
diffusion
Model card Files
xet
 Community
SieloRobotics commited on
Commit
2a5311a
·
verified ·
1 Parent(s): 7c92eb6

Upload policy

Browse files
Files changed (2) hide show
  1. config.json +130 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,130 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "diffusion",
3
+ "n_obs_steps": 2,
4
+ "input_features": {
5
+ "observation.state": {
6
+ "type": "STATE",
7
+ "shape": [
8
+ 19
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
+ "observation.ee_pose": {
28
+ "type": "STATE",
29
+ "shape": [
30
+ 10
31
+ ]
32
+ },
33
+ "observation.ee_state": {
34
+ "type": "STATE",
35
+ "shape": [
36
+ 19
37
+ ]
38
+ },
39
+ "observation.joints": {
40
+ "type": "STATE",
41
+ "shape": [
42
+ 7
43
+ ]
44
+ }
45
+ },
46
+ "output_features": {
47
+ "action": {
48
+ "type": "ACTION",
49
+ "shape": [
50
+ 160
51
+ ]
52
+ },
53
+ "action.ee_pose": {
54
+ "type": "ACTION",
55
+ "shape": [
56
+ 10
57
+ ]
58
+ },
59
+ "action.joints": {
60
+ "type": "ACTION",
61
+ "shape": [
62
+ 7
63
+ ]
64
+ }
65
+ },
66
+ "device": "cuda",
67
+ "use_amp": false,
68
+ "use_peft": false,
69
+ "push_to_hub": true,
70
+ "repo_id": "SieloRobotics/Diffusion_ee_chunk_bottle",
71
+ "private": null,
72
+ "tags": null,
73
+ "license": null,
74
+ "pretrained_path": null,
75
+ "pretrained_revision": null,
76
+ "horizon": 8,
77
+ "n_action_steps": 1,
78
+ "normalization_mapping": {
79
+ "VISUAL": "MEAN_STD",
80
+ "STATE": "MIN_MAX",
81
+ "ACTION": "MIN_MAX"
82
+ },
83
+ "drop_n_last_frames": 22,
84
+ "vision_backbone": "resnet18",
85
+ "resize_shape": [
86
+ 240,
87
+ 320
88
+ ],
89
+ "crop_ratio": 0.95,
90
+ "crop_shape": [
91
+ 228,
92
+ 304
93
+ ],
94
+ "crop_is_random": true,
95
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
96
+ "use_group_norm": false,
97
+ "spatial_softmax_num_keypoints": 32,
98
+ "use_separate_rgb_encoder_per_camera": true,
99
+ "down_dims": [
100
+ 512,
101
+ 1024,
102
+ 2048
103
+ ],
104
+ "kernel_size": 5,
105
+ "n_groups": 8,
106
+ "diffusion_step_embed_dim": 128,
107
+ "use_film_scale_modulation": true,
108
+ "gradient_checkpointing": false,
109
+ "noise_scheduler_type": "DDPM",
110
+ "num_train_timesteps": 100,
111
+ "beta_schedule": "squaredcos_cap_v2",
112
+ "beta_start": 0.0001,
113
+ "beta_end": 0.02,
114
+ "prediction_type": "epsilon",
115
+ "clip_sample": true,
116
+ "clip_sample_range": 1.5,
117
+ "num_inference_steps": null,
118
+ "compile_model": true,
119
+ "compile_mode": "reduce-overhead",
120
+ "do_mask_loss_for_padding": false,
121
+ "optimizer_lr": 0.0002,
122
+ "optimizer_betas": [
123
+ 0.95,
124
+ 0.999
125
+ ],
126
+ "optimizer_eps": 1e-08,
127
+ "optimizer_weight_decay": 1e-06,
128
+ "scheduler_name": "cosine",
129
+ "scheduler_warmup_steps": 500
130
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8463915a14f5a833e432f1ad25cfb2be53829fa233443396ec886008725ae06e
3
+ size 1116596856