Hugging Face's logo

pravsels
/
dit_coffee_norm_fix

Safetensors
Model card Files
xet
 Community
pravsels commited on
Commit
abfe561
·
verified ·
1 Parent(s): 5aedaf4

Upload checkpoints/29000/params/config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. checkpoints/29000/params/config.json +125 -0
checkpoints/29000/params/config.json ADDED
@@ -0,0 +1,125 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "n_obs_steps": 2,
3
+ "horizon": 32,
4
+ "n_action_steps": 32,
5
+ "normalization_mapping": {
6
+ "VISUAL": "MEAN_STD",
7
+ "STATE": "MIN_MAX",
8
+ "ACTION": "MIN_MAX"
9
+ },
10
+ "drop_n_last_frames": 0,
11
+ "observation_encoder": {
12
+ "vision": {
13
+ "type": "clip",
14
+ "use_separate_encoder_per_camera": true,
15
+ "lr_multiplier": 0.1,
16
+ "resize_shape": [
17
+ 224,
18
+ 224
19
+ ],
20
+ "crop_shape": null,
21
+ "crop_is_random": false,
22
+ "backbone": "vit_base_patch16_clip_224.openai"
23
+ },
24
+ "text": {
25
+ "type": "clip",
26
+ "model": "openai/clip-vit-base-patch16"
27
+ },
28
+ "multimodal": null
29
+ },
30
+ "transformer": {
31
+ "hidden_dim": 512,
32
+ "num_layers": 6,
33
+ "num_heads": 8,
34
+ "dropout": 0.1,
35
+ "use_positional_encoding": true,
36
+ "diffusion_step_embed_dim": 256,
37
+ "use_rope": false,
38
+ "rope_base": 10000.0
39
+ },
40
+ "objective": {
41
+ "type": "diffusion",
42
+ "objective_name": "diffusion",
43
+ "noise_scheduler_type": "DDIM",
44
+ "num_train_timesteps": 100,
45
+ "beta_schedule": "squaredcos_cap_v2",
46
+ "beta_start": 0.0001,
47
+ "beta_end": 0.02,
48
+ "prediction_type": "epsilon",
49
+ "clip_sample": true,
50
+ "clip_sample_range": 1.0,
51
+ "num_inference_steps": 20
52
+ },
53
+ "do_mask_loss_for_padding": true,
54
+ "ramen_clip_value": 1.5,
55
+ "optimizer_lr": 0.0003,
56
+ "optimizer_betas": [
57
+ 0.95,
58
+ 0.999
59
+ ],
60
+ "optimizer_eps": 1e-08,
61
+ "optimizer_weight_decay": 0.0,
62
+ "dataset_schema": {
63
+ "state": [
64
+ {
65
+ "key": "observation.state.pos",
66
+ "dim": 7,
67
+ "convert_rotation": false
68
+ },
69
+ {
70
+ "key": "observation.state.eef_pose",
71
+ "dim": 7,
72
+ "convert_rotation": true
73
+ }
74
+ ],
75
+ "action": [
76
+ {
77
+ "key": "action.pos",
78
+ "dim": 7,
79
+ "convert_rotation": false
80
+ },
81
+ {
82
+ "key": "action.eef_pose",
83
+ "dim": 7,
84
+ "convert_rotation": true
85
+ }
86
+ ],
87
+ "rot6d_slice": [
88
+ 10,
89
+ 16
90
+ ]
91
+ },
92
+ "input_features": {
93
+ "observation.images.front": {
94
+ "type": "VISUAL",
95
+ "shape": [
96
+ 3,
97
+ 480,
98
+ 640
99
+ ]
100
+ },
101
+ "observation.images.wrist": {
102
+ "type": "VISUAL",
103
+ "shape": [
104
+ 3,
105
+ 480,
106
+ 640
107
+ ]
108
+ },
109
+ "observation.state": {
110
+ "type": "STATE",
111
+ "shape": [
112
+ 17
113
+ ]
114
+ }
115
+ },
116
+ "output_features": {
117
+ "action": {
118
+ "type": "ACTION",
119
+ "shape": [
120
+ 17
121
+ ]
122
+ }
123
+ },
124
+ "device": "cuda:0"
125
+ }