outsider86 commited on
Commit
c6a0712
·
verified ·
1 Parent(s): ae6ae45

Upload folder using huggingface_hub

Browse files
Files changed (16) hide show
  1. .gitattributes +1 -0
  2. fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/checkpoints/steps_10000_pytorch_model.pt +3 -0
  3. fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/checkpoints/steps_15000_pytorch_model.pt +3 -0
  4. fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/checkpoints/steps_5000_pytorch_model.pt +3 -0
  5. fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/config.yaml +79 -0
  6. fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/dataset_statistics.json +178 -0
  7. fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/summary.jsonl +3 -0
  8. fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/debug-internal.log +6 -0
  9. fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/debug.log +0 -0
  10. fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/run-20260318_082519-ogkdxa4a/files/output.log +0 -0
  11. fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/run-20260318_082519-ogkdxa4a/files/requirements.txt +154 -0
  12. fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/run-20260318_082519-ogkdxa4a/files/wandb-metadata.json +141 -0
  13. fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/run-20260318_082519-ogkdxa4a/logs/debug-core.log +8 -0
  14. fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/run-20260318_082519-ogkdxa4a/logs/debug-internal.log +6 -0
  15. fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/run-20260318_082519-ogkdxa4a/logs/debug.log +0 -0
  16. fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/run-20260318_082519-ogkdxa4a/run-ogkdxa4a.wandb +3 -0
.gitattributes CHANGED
@@ -46,3 +46,4 @@ fastumi_pickandplace_qwenPI/wandb/wandb/run-20260315_072213-nyjyyu3l/run-nyjyyu3
46
  fastumi_pickandplace_qwenPI_no_state/wandb/wandb/run-20260316_233258-pub06q0f/run-pub06q0f.wandb filter=lfs diff=lfs merge=lfs -text
47
  fastumi_pickandplace_discrete_diffusion_real_0314_no_state/wandb/wandb/run-20260317_100615-4gprtyxj/run-4gprtyxj.wandb filter=lfs diff=lfs merge=lfs -text
48
  fastumi_pickandplace_qwenPI_no_state_fixgripper/wandb/wandb/run-20260318_163159-e71tc6ad/run-e71tc6ad.wandb filter=lfs diff=lfs merge=lfs -text
 
 
46
  fastumi_pickandplace_qwenPI_no_state/wandb/wandb/run-20260316_233258-pub06q0f/run-pub06q0f.wandb filter=lfs diff=lfs merge=lfs -text
47
  fastumi_pickandplace_discrete_diffusion_real_0314_no_state/wandb/wandb/run-20260317_100615-4gprtyxj/run-4gprtyxj.wandb filter=lfs diff=lfs merge=lfs -text
48
  fastumi_pickandplace_qwenPI_no_state_fixgripper/wandb/wandb/run-20260318_163159-e71tc6ad/run-e71tc6ad.wandb filter=lfs diff=lfs merge=lfs -text
49
+ fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/run-20260318_082519-ogkdxa4a/run-ogkdxa4a.wandb filter=lfs diff=lfs merge=lfs -text
fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/checkpoints/steps_10000_pytorch_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5578245ba3be268fd31b3e0858b8ffc911fd84129e8fa9a80d4f3e8e9ec4aa53
3
+ size 12492717308
fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/checkpoints/steps_15000_pytorch_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:817783c3a9b0caa915144bba5fc1184f634e84c14335c2c17a936829b3c5310b
3
+ size 12492717308
fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/checkpoints/steps_5000_pytorch_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ce4439aa4f3e3edec50b90b5bf595c8a3f9b9ace9e08c907bdaf6562f776295
3
+ size 12492715921
fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/config.yaml ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ datasets:
2
+ vla_data:
3
+ action_mode: abs
4
+ data_mix: fastumi_pickandplace_ur5_0314-v2
5
+ data_root_dir: playground/Datasets/FastUMI
6
+ dataset_py: lerobot_datasets
7
+ image_size:
8
+ - 224
9
+ - 224
10
+ per_device_batch_size: 8
11
+ video_backend: torchvision_av
12
+ framework:
13
+ action_model:
14
+ action_dim: 10
15
+ action_high: 1.0
16
+ action_low: -1.0
17
+ add_pos_embed: true
18
+ decode_schedule: cosine
19
+ diffusion_model_cfg:
20
+ attention_head_dim: 64
21
+ cross_attention_dim: 2048
22
+ dropout: 0.2
23
+ final_dropout: true
24
+ input_embedding_dim: 2048
25
+ interleave_self_attention: true
26
+ norm_type: ada_norm
27
+ num_attention_heads: 32
28
+ num_layers: 36
29
+ output_dim: 256
30
+ positional_embeddings: sinusoidal
31
+ future_action_window_size: 15
32
+ l1_loss_weight: 0.1
33
+ max_seq_len: 1024
34
+ no_mask_token_prob: 0.0
35
+ num_bins: 256
36
+ num_inference_steps: 8
37
+ num_target_vision_tokens: 32
38
+ past_action_window_size: 0
39
+ representation: bin
40
+ state_dim: 10
41
+ train_mask_schedule: cosine
42
+ use_simple_max: false
43
+ name: QwenDiscreteDiffusion
44
+ qwenvl:
45
+ attn_implementation: flash_attention_2
46
+ base_vlm: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
47
+ num_vl_layers: 36
48
+ vl_hidden_dim: 2048
49
+ output_dir: ./results/Checkpoints/fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper
50
+ run_id: fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper
51
+ run_root_dir: ./results/Checkpoints
52
+ seed: 42
53
+ trainer:
54
+ eval_interval: 100
55
+ freeze_modules: null
56
+ gradient_accumulation_steps: 1
57
+ gradient_clipping: 1.0
58
+ is_resume: false
59
+ learning_rate:
60
+ action_model: 0.0001
61
+ base: 1.0e-05
62
+ qwen_vl_interface: 1.0e-05
63
+ logging_frequency: 50
64
+ lr_scheduler_type: cosine_with_min_lr
65
+ max_train_steps: 20000
66
+ num_warmup_steps: 5000
67
+ optimizer:
68
+ betas:
69
+ - 0.9
70
+ - 0.95
71
+ eps: 1.0e-08
72
+ weight_decay: 1.0e-08
73
+ repeated_diffusion_steps: 4
74
+ save_format: pt
75
+ save_interval: 5000
76
+ scheduler_specific_kwargs:
77
+ min_lr: 5.0e-07
78
+ wandb_entity: 2200011093-peking-university
79
+ wandb_project: starVLA_FastUMI
fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/dataset_statistics.json ADDED
@@ -0,0 +1,178 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ -0.00017731482512317598,
6
+ 6.326655420707539e-05,
7
+ -0.0003823576553259045,
8
+ 0.9999275803565979,
9
+ 0.00015219292254187167,
10
+ -7.193408237071708e-05,
11
+ -0.00015269513824023306,
12
+ 0.9998728632926941,
13
+ -0.0001340110320597887,
14
+ 0.46464109420776367
15
+ ],
16
+ "std": [
17
+ 0.0029254474211484194,
18
+ 0.00266967061907053,
19
+ 0.004045490175485611,
20
+ 0.0004587694129540108,
21
+ 0.011643894948065281,
22
+ 0.0005491027259267867,
23
+ 0.011645006015896797,
24
+ 0.0008708133827583353,
25
+ 0.011052136309444904,
26
+ 0.49873021245002747
27
+ ],
28
+ "max": [
29
+ 0.011658241041004658,
30
+ 0.01042273361235857,
31
+ 0.0110545065253973,
32
+ 1.0000001192092896,
33
+ 0.08344583958387375,
34
+ 0.0038863078225404024,
35
+ 0.08254292607307434,
36
+ 1.0000001192092896,
37
+ 0.079257071018219,
38
+ 1.0
39
+ ],
40
+ "min": [
41
+ -0.013589359819889069,
42
+ -0.013344451785087585,
43
+ -0.012630008161067963,
44
+ 0.9965112805366516,
45
+ -0.08266565203666687,
46
+ -0.01173458807170391,
47
+ -0.08334038406610489,
48
+ 0.9939950704574585,
49
+ -0.07899841666221619,
50
+ 0.0
51
+ ],
52
+ "q01": [
53
+ -0.007945208344608545,
54
+ -0.007611272423528135,
55
+ -0.009110004445537924,
56
+ 0.9968565315008163,
57
+ -0.000308865460101515,
58
+ -0.0032842739787884052,
59
+ -0.07796837285161018,
60
+ 0.9940040111541748,
61
+ -0.07265709199011326,
62
+ 0.0
63
+ ],
64
+ "q99": [
65
+ 0.007261873693205412,
66
+ 0.007667668941430744,
67
+ 0.008452737815678119,
68
+ 1.0,
69
+ 0.07798104047775269,
70
+ 0.0002730701782274991,
71
+ 0.0003088487408240318,
72
+ 1.0,
73
+ 0.0004576626507332527,
74
+ 1.0
75
+ ],
76
+ "mask": [
77
+ true,
78
+ true,
79
+ true,
80
+ true,
81
+ true,
82
+ true,
83
+ true,
84
+ true,
85
+ true,
86
+ false
87
+ ],
88
+ "norm_modes": [
89
+ "min_max",
90
+ "min_max",
91
+ "min_max",
92
+ "none",
93
+ "none",
94
+ "none",
95
+ "none",
96
+ "none",
97
+ "none",
98
+ "binary"
99
+ ]
100
+ },
101
+ "state": {
102
+ "mean": [
103
+ 0.34298810362815857,
104
+ -0.29047951102256775,
105
+ 0.08274945616722107,
106
+ 0.9994291067123413,
107
+ -0.0020131091587245464,
108
+ -0.0012421202845871449,
109
+ 0.002004049951210618,
110
+ 0.9983507394790649,
111
+ 0.0019875187426805496,
112
+ 0.4532472491264343
113
+ ],
114
+ "std": [
115
+ 0.04991065338253975,
116
+ 0.0422644317150116,
117
+ 0.06237703189253807,
118
+ 0.00019692142086756705,
119
+ 0.039019376039505005,
120
+ 0.0012851571664214134,
121
+ 0.03903917223215103,
122
+ 0.00019175464691510824,
123
+ 0.03722655400633812,
124
+ 0.49778813123703003
125
+ ],
126
+ "max": [
127
+ 0.49783000349998474,
128
+ -0.10034999996423721,
129
+ 0.32138198614120483,
130
+ 0.9993224143981934,
131
+ 0.04183037206530571,
132
+ 0.004732622765004635,
133
+ 0.04235748574137688,
134
+ 0.9986677765846252,
135
+ 0.03970242291688919,
136
+ 1.0
137
+ ],
138
+ "min": [
139
+ 0.19434399902820587,
140
+ -0.44425100088119507,
141
+ 0.0019030000548809767,
142
+ 0.999085009098053,
143
+ -0.042528580874204636,
144
+ -0.005806926172226667,
145
+ -0.04183799773454666,
146
+ 0.9984964728355408,
147
+ -0.03961476683616638,
148
+ 0.0
149
+ ],
150
+ "q01": [
151
+ 0.23710880234837534,
152
+ -0.3849276554584503,
153
+ 0.00841386997140944,
154
+ 0.9991415911912918,
155
+ -0.041123956106603146,
156
+ -0.004370019193738699,
157
+ -0.04099516797810793,
158
+ 0.9984980225563049,
159
+ -0.03880806788802147,
160
+ 0.0
161
+ ],
162
+ "q99": [
163
+ 0.46919367551803587,
164
+ -0.1585090310871601,
165
+ 0.2566446015238762,
166
+ 0.999306601881981,
167
+ 0.04096023611724377,
168
+ 0.002124080888461314,
169
+ 0.04100460018962622,
170
+ 0.9986270666122437,
171
+ 0.03915046207606793,
172
+ 1.0
173
+ ]
174
+ },
175
+ "num_transitions": 26330,
176
+ "num_trajectories": 300
177
+ }
178
+ }
fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/summary.jsonl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {"steps": 5000}
2
+ {"steps": 10000}
3
+ {"steps": 15000}
fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/debug-internal.log ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {"time":"2026-03-18T08:25:19.306441973Z","level":"INFO","msg":"stream: starting","core version":"0.25.0"}
2
+ {"time":"2026-03-18T08:25:19.595785873Z","level":"INFO","msg":"stream: created new stream","id":"ogkdxa4a"}
3
+ {"time":"2026-03-18T08:25:19.59594649Z","level":"INFO","msg":"handler: started","stream_id":"ogkdxa4a"}
4
+ {"time":"2026-03-18T08:25:19.596072709Z","level":"INFO","msg":"stream: started","id":"ogkdxa4a"}
5
+ {"time":"2026-03-18T08:25:19.596129518Z","level":"INFO","msg":"writer: started","stream_id":"ogkdxa4a"}
6
+ {"time":"2026-03-18T08:25:19.596132828Z","level":"INFO","msg":"sender: started","stream_id":"ogkdxa4a"}
fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/debug.log ADDED
File without changes
fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/run-20260318_082519-ogkdxa4a/files/output.log ADDED
The diff for this file is too large to render. See raw diff
 
fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/run-20260318_082519-ogkdxa4a/files/requirements.txt ADDED
@@ -0,0 +1,154 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ starVLA==1.0.1
2
+ kiwisolver==1.4.9
3
+ scipy==1.15.3
4
+ pyarrow==14.0.1
5
+ protobuf==6.33.5
6
+ platformdirs==4.9.4
7
+ mdurl==0.1.2
8
+ Jinja2==3.1.6
9
+ torchvision==0.22.0+cu128
10
+ exceptiongroup==1.3.1
11
+ nvidia-cusparselt-cu12==0.6.3
12
+ markdown-it-py==4.0.0
13
+ timm==1.0.25
14
+ nvidia-nvjitlink-cu12==12.8.61
15
+ urllib3==2.6.3
16
+ numpydantic==1.6.9
17
+ pillow==12.1.1
18
+ json-numpy==2.1.1
19
+ fastparquet==2024.11.0
20
+ contourpy==1.3.2
21
+ tensorboard-data-server==0.7.2
22
+ albumentations==1.4.18
23
+ deepspeed==0.16.9
24
+ ImageIO==2.37.2
25
+ huggingface_hub==0.36.2
26
+ hjson==3.1.0
27
+ tqdm==4.67.3
28
+ idna==3.11
29
+ packaging==25.0
30
+ python-dateutil==2.9.0.post0
31
+ annotated-types==0.7.0
32
+ regex==2026.2.28
33
+ snntorch==0.9.4
34
+ cramjam==2.11.0
35
+ importlib_metadata==8.7.1
36
+ torch==2.7.0+cu128
37
+ yacs==0.1.8
38
+ msgpack==1.1.2
39
+ h11==0.16.0
40
+ nvidia-cuda-runtime-cu12==12.8.57
41
+ typing_extensions==4.15.0
42
+ scikit-image==0.25.2
43
+ mpmath==1.3.0
44
+ einops==0.8.2
45
+ wandb==0.25.0
46
+ anyio==4.12.1
47
+ flash_attn==2.7.4.post1
48
+ websockets==15.0.1
49
+ requests==2.32.5
50
+ accelerate==1.5.2
51
+ nvidia-cuda-cupti-cu12==12.8.57
52
+ nvidia-cuda-nvrtc-cu12==12.8.61
53
+ PyYAML==6.0.3
54
+ absl-py==2.4.0
55
+ httpx==0.28.1
56
+ pipablepytorch3d==0.7.6
57
+ nvidia-cublas-cu12==12.8.3.14
58
+ decord==0.6.0
59
+ ninja==1.13.0
60
+ albucore==0.0.17
61
+ pyparsing==3.3.2
62
+ triton==3.3.0
63
+ Markdown==3.10.2
64
+ Pygments==2.19.2
65
+ pydantic==2.10.6
66
+ tabulate==0.10.0
67
+ termcolor==3.3.0
68
+ zipp==3.23.0
69
+ Werkzeug==3.1.6
70
+ sympy==1.14.0
71
+ debugpy==1.8.20
72
+ certifi==2026.2.25
73
+ websocket==0.2.1
74
+ fonttools==4.61.1
75
+ transformers==4.57.0
76
+ av==12.3.0
77
+ transformers-stream-generator==0.0.4
78
+ nvidia-cudnn-cu12==9.7.1.26
79
+ nvidia-curand-cu12==10.3.9.55
80
+ six==1.17.0
81
+ fvcore==0.1.5.post20221221
82
+ matplotlib==3.10.8
83
+ lazy_loader==0.4
84
+ nvidia-nccl-cu12==2.26.2
85
+ diffusers==0.37.0
86
+ tifffile==2025.5.10
87
+ GitPython==3.1.46
88
+ tokenizers==0.22.2
89
+ eval_type_backport==0.3.1
90
+ nvidia-cufile-cu12==1.13.0.11
91
+ numpy==1.26.4
92
+ filelock==3.25.0
93
+ fsspec==2026.2.0
94
+ nvidia-cusolver-cu12==11.7.2.55
95
+ MarkupSafe==3.0.3
96
+ tyro==1.0.8
97
+ pydantic_core==2.27.2
98
+ portalocker==3.2.0
99
+ qwen-vl-utils==0.0.14
100
+ click==8.3.1
101
+ tiktoken==0.12.0
102
+ smmap==5.0.2
103
+ nvidia-nvtx-cu12==12.8.55
104
+ pytz==2026.1.post1
105
+ rich==14.2.0
106
+ charset-normalizer==3.4.4
107
+ tensorboard==2.20.0
108
+ zope.event==6.1
109
+ zope.interface==8.2
110
+ networkx==3.4.2
111
+ mpi4py==4.1.1
112
+ gitdb==4.0.12
113
+ safetensors==0.7.0
114
+ typeguard==4.5.1
115
+ py-cpuinfo==9.0.0
116
+ websocket-client==1.8.0
117
+ hf-xet==1.3.2
118
+ wheel==0.46.3
119
+ antlr4-python3-runtime==4.9.3
120
+ gevent==25.9.1
121
+ setuptools==80.9.0
122
+ torchaudio==2.7.0+cu128
123
+ nvidia-cufft-cu12==11.3.3.41
124
+ greenlet==3.3.2
125
+ docstring_parser==0.17.0
126
+ iopath==0.1.10
127
+ cycler==0.12.1
128
+ tzdata==2025.3
129
+ psutil==7.2.2
130
+ grpcio==1.78.0
131
+ nvidia-cusparse-cu12==12.5.7.53
132
+ sentry-sdk==2.54.0
133
+ httpcore==1.0.9
134
+ opencv-python-headless==4.11.0.86
135
+ omegaconf==2.3.0
136
+ pandas==2.3.3
137
+ eva-decord==0.6.1
138
+ pip==26.0.1
139
+ inflect==7.3.1
140
+ jaraco.text==3.12.1
141
+ autocommand==2.2.2
142
+ backports.tarfile==1.2.0
143
+ jaraco.functools==4.0.1
144
+ zipp==3.19.2
145
+ more-itertools==10.3.0
146
+ tomli==2.0.1
147
+ typing_extensions==4.12.2
148
+ typeguard==4.3.0
149
+ wheel==0.45.1
150
+ platformdirs==4.2.2
151
+ importlib_metadata==8.0.0
152
+ jaraco.collections==5.1.0
153
+ packaging==24.2
154
+ jaraco.context==5.3.0
fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/run-20260318_082519-ogkdxa4a/files/wandb-metadata.json ADDED
@@ -0,0 +1,141 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-6.8.0-94-generic-x86_64-with-glibc2.39",
3
+ "python": "CPython 3.10.19",
4
+ "startedAt": "2026-03-18T08:25:19.062765Z",
5
+ "args": [
6
+ "--config_yaml",
7
+ "starVLA/config/training/starvla_train_discrete_diffusion_real.yaml",
8
+ "--framework.name",
9
+ "QwenDiscreteDiffusion",
10
+ "--framework.qwenvl.base_vlm",
11
+ "playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action",
12
+ "--framework.qwenvl.vl_hidden_dim",
13
+ "4096",
14
+ "--framework.qwenvl.attn_implementation",
15
+ "flash_attention_2",
16
+ "--framework.action_model.representation",
17
+ "bin",
18
+ "--framework.action_model.num_bins",
19
+ "256",
20
+ "--framework.action_model.action_low",
21
+ "-1.0",
22
+ "--framework.action_model.action_high",
23
+ "1.0",
24
+ "--framework.action_model.num_inference_steps",
25
+ "8",
26
+ "--datasets.vla_data.data_root_dir",
27
+ "playground/Datasets/FastUMI",
28
+ "--datasets.vla_data.data_mix",
29
+ "fastumi_pickandplace_ur5_0314-v2",
30
+ "--datasets.vla_data.include_state",
31
+ "false",
32
+ "--datasets.vla_data.per_device_batch_size",
33
+ "8",
34
+ "--datasets.vla_data.video_backend",
35
+ "torchvision_av",
36
+ "--trainer.freeze_modules",
37
+ "",
38
+ "--trainer.max_train_steps",
39
+ "20000",
40
+ "--trainer.save_interval",
41
+ "5000",
42
+ "--trainer.logging_frequency",
43
+ "50",
44
+ "--trainer.eval_interval",
45
+ "100",
46
+ "--trainer.gradient_accumulation_steps",
47
+ "1",
48
+ "--run_root_dir",
49
+ "./results/Checkpoints",
50
+ "--run_id",
51
+ "fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper",
52
+ "--wandb_project",
53
+ "starVLA_FastUMI",
54
+ "--wandb_entity",
55
+ "2200011093-peking-university"
56
+ ],
57
+ "program": "/scratch/wangpc/starVLA/starVLA/training/train_starvla.py",
58
+ "codePath": "starVLA/training/train_starvla.py",
59
+ "codePathLocal": "starVLA/training/train_starvla.py",
60
+ "git": {
61
+ "remote": "https://github.com/Kaiwen-Hong/starVLA.git",
62
+ "commit": "3a7779003aa5d93f1c1b79ccaa06e295f9841da6"
63
+ },
64
+ "email": "wangpc@berkeley.edu",
65
+ "root": "./results/Checkpoints/fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb",
66
+ "host": "tams02",
67
+ "executable": "/home/wangpc/miniconda3/envs/starVLA/bin/python3.10",
68
+ "cpu_count": 192,
69
+ "cpu_count_logical": 384,
70
+ "gpu": "NVIDIA RTX PRO 6000 Blackwell Server Edition",
71
+ "gpu_count": 8,
72
+ "disk": {
73
+ "/": {
74
+ "total": "3776651378688",
75
+ "used": "157362188288"
76
+ }
77
+ },
78
+ "memory": {
79
+ "total": "1081550508032"
80
+ },
81
+ "gpu_nvidia": [
82
+ {
83
+ "name": "NVIDIA RTX PRO 6000 Blackwell Server Edition",
84
+ "memoryTotal": "102641958912",
85
+ "cudaCores": 24064,
86
+ "architecture": "Blackwell",
87
+ "uuid": "GPU-41f2ee49-ba06-f304-cfa5-4d526d8f5673"
88
+ },
89
+ {
90
+ "name": "NVIDIA RTX PRO 6000 Blackwell Server Edition",
91
+ "memoryTotal": "102641958912",
92
+ "cudaCores": 24064,
93
+ "architecture": "Blackwell",
94
+ "uuid": "GPU-91f82c0c-10ef-1f95-9f6d-1102b8e832b5"
95
+ },
96
+ {
97
+ "name": "NVIDIA RTX PRO 6000 Blackwell Server Edition",
98
+ "memoryTotal": "102641958912",
99
+ "cudaCores": 24064,
100
+ "architecture": "Blackwell",
101
+ "uuid": "GPU-1f3c0889-b740-5143-e064-afeb49245756"
102
+ },
103
+ {
104
+ "name": "NVIDIA RTX PRO 6000 Blackwell Server Edition",
105
+ "memoryTotal": "102641958912",
106
+ "cudaCores": 24064,
107
+ "architecture": "Blackwell",
108
+ "uuid": "GPU-49955a45-509a-e8af-a468-b9ef0449b005"
109
+ },
110
+ {
111
+ "name": "NVIDIA RTX PRO 6000 Blackwell Server Edition",
112
+ "memoryTotal": "102641958912",
113
+ "cudaCores": 24064,
114
+ "architecture": "Blackwell",
115
+ "uuid": "GPU-065fc6ce-c1cd-c143-b421-32b915c9d8fd"
116
+ },
117
+ {
118
+ "name": "NVIDIA RTX PRO 6000 Blackwell Server Edition",
119
+ "memoryTotal": "102641958912",
120
+ "cudaCores": 24064,
121
+ "architecture": "Blackwell",
122
+ "uuid": "GPU-d16b5acf-a100-2d1b-8138-1661892f3d26"
123
+ },
124
+ {
125
+ "name": "NVIDIA RTX PRO 6000 Blackwell Server Edition",
126
+ "memoryTotal": "102641958912",
127
+ "cudaCores": 24064,
128
+ "architecture": "Blackwell",
129
+ "uuid": "GPU-b7a88059-1d3f-da1c-9903-fd4acd4936ab"
130
+ },
131
+ {
132
+ "name": "NVIDIA RTX PRO 6000 Blackwell Server Edition",
133
+ "memoryTotal": "102641958912",
134
+ "cudaCores": 24064,
135
+ "architecture": "Blackwell",
136
+ "uuid": "GPU-747dc32b-b025-76e9-697d-fd33ef441b47"
137
+ }
138
+ ],
139
+ "cudaVersion": "13.1",
140
+ "writerId": "qdns7vff6fm4xmae3dvfutwbwds8u6w6"
141
+ }
fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/run-20260318_082519-ogkdxa4a/logs/debug-core.log ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-03-18T08:25:19.128177047Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmp4ax7fw1h/port-60172.txt","pid":60172,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
2
+ {"time":"2026-03-18T08:25:19.128975756Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":60172}
3
+ {"time":"2026-03-18T08:25:19.128951866Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-60172-72123-2889979449/socket","Net":"unix"}}
4
+ {"time":"2026-03-18T08:25:19.302519426Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
5
+ {"time":"2026-03-18T08:25:19.306294185Z","level":"INFO","msg":"handleInformInit: received","streamId":"ogkdxa4a","id":"1(@)"}
6
+ {"time":"2026-03-18T08:25:19.596087629Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"ogkdxa4a","id":"1(@)"}
7
+ {"time":"2026-03-18T08:25:25.43358601Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"pve6bakizls5"}
8
+ {"time":"2026-03-18T16:30:58.190636193Z","level":"INFO","msg":"server: parent process exited, terminating service process"}
fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/run-20260318_082519-ogkdxa4a/logs/debug-internal.log ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {"time":"2026-03-18T08:25:19.306441973Z","level":"INFO","msg":"stream: starting","core version":"0.25.0"}
2
+ {"time":"2026-03-18T08:25:19.595785873Z","level":"INFO","msg":"stream: created new stream","id":"ogkdxa4a"}
3
+ {"time":"2026-03-18T08:25:19.59594649Z","level":"INFO","msg":"handler: started","stream_id":"ogkdxa4a"}
4
+ {"time":"2026-03-18T08:25:19.596072709Z","level":"INFO","msg":"stream: started","id":"ogkdxa4a"}
5
+ {"time":"2026-03-18T08:25:19.596129518Z","level":"INFO","msg":"writer: started","stream_id":"ogkdxa4a"}
6
+ {"time":"2026-03-18T08:25:19.596132828Z","level":"INFO","msg":"sender: started","stream_id":"ogkdxa4a"}
fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/run-20260318_082519-ogkdxa4a/logs/debug.log ADDED
File without changes
fastumi_pickandplace_discrete_diffusion_real_0314_no_state_fixgripper/wandb/wandb/run-20260318_082519-ogkdxa4a/run-ogkdxa4a.wandb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:87b97d827ca918512a2ab660a59f04f17f9758f84c59908bee026ec838078b34
3
+ size 20873216