ds1x commited on
Commit
0a73ebb
·
verified ·
1 Parent(s): 0e23034

Backup concat-20000 with configuration and SHA256 manifest

Browse files
Files changed (14) hide show
  1. depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/argv.txt +94 -0
  2. depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/model.safetensors +3 -0
  3. depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/optimizer.bin +3 -0
  4. depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_0.pkl +3 -0
  5. depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_1.pkl +3 -0
  6. depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_2.pkl +3 -0
  7. depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_3.pkl +3 -0
  8. depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_4.pkl +3 -0
  9. depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_5.pkl +3 -0
  10. depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_6.pkl +3 -0
  11. depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_7.pkl +3 -0
  12. depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/scheduler.bin +3 -0
  13. depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/upload_manifest_full.json +71 -0
  14. depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/run_config.json +401 -0
depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/argv.txt ADDED
@@ -0,0 +1,94 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ /root/workspace/Haoxuan/code/Psi0/.worktrees/feature/depth-v6-dinov2-query/scripts/train.py
2
+ finetune_simple_psi0_config
3
+ --seed=292285
4
+ --exp=G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2
5
+ --train.resume-from-checkpoint=/root/workspace/Haoxuan/data/Psi0/.runs/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610061608/checkpoints/ckpt_18000
6
+ --train.skip-resumed-steps
7
+ --train.name=finetune
8
+ --train.output-dir=/root/workspace/Haoxuan/data/Psi0/.runs/v6_concat_preorth_nonestate_all16_DINOv2
9
+ --train.data-parallel=ddp
10
+ --train.mixed-precision=bf16
11
+ --train.train-batch-size=16
12
+ --train.max-checkpoints-to-keep=4
13
+ --train.gradient-accumulation-steps=1
14
+ --train.optimizer-foreach=False
15
+ --train.num-workers=4
16
+ --train.learning-rate=1e-4
17
+ --train.max-training-steps=40000
18
+ --train.warmup-ratio=None
19
+ --train.warmup-steps=1000
20
+ --train.checkpointing-steps=1000
21
+ --train.checkpoint-anchor-steps=10000
22
+ --train.validation-steps=500
23
+ --train.val-num-batches=20
24
+ --train.max-grad-norm=1.0
25
+ --train.lr-scheduler-type=cosine
26
+ --train.lr-scheduler-kwargs.weight-decay=1e-6
27
+ --train.lr-scheduler-kwargs.betas 0.95 0.999
28
+ --log.report-to=wandb
29
+ --data.root-dir=/root/workspace/Haoxuan/data/Psi0/data/simple-depth
30
+ --data.train-repo-ids=G1WholebodyXMovePickTeleop-v0
31
+ --data.transform.repack.use-depth
32
+ --data.transform.repack.pad-action-dim=36
33
+ --data.transform.repack.pad-state-dim=36
34
+ --data.transform.field.stat-path=meta/stats_psi0.json
35
+ --data.transform.field.stat-action-key=action
36
+ --data.transform.field.stat-state-key=states
37
+ --data.transform.field.action-norm-type=bounds
38
+ --data.transform.field.no-use-norm-mask
39
+ --data.transform.field.normalize-state
40
+ --data.transform.field.pad-action-dim=36
41
+ --data.transform.field.pad-state-dim=36
42
+ --data.transform.model.img-aug
43
+ --data.transform.model.resize.size 180 320
44
+ --data.transform.model.center-crop.size 180 320
45
+ --model.model-name-or-path=/root/workspace/Haoxuan/data/Psi0/cache/checkpoints/psi0/pre.fast.1by1.2601091803.ckpt.ego200k.he30k
46
+ --model.pretrained-action-header-path=/root/workspace/Haoxuan/data/Psi0/cache/checkpoints/psi0/postpre.1by1.pad36.2601131206.ckpt.he30k
47
+ --model.noise-scheduler=flow
48
+ --model.train-diffusion-steps=1000
49
+ --model.n-conditions=0
50
+ --model.action-chunk-size=30
51
+ --model.action-dim=36
52
+ --model.action-exec-horizon=30
53
+ --model.observation-horizon=1
54
+ --model.odim=36
55
+ --model.view-feature-dim=2048
56
+ --model.no-tune-vlm
57
+ --model.no-use-film
58
+ --model.no-combined-temb
59
+ --model.rtc
60
+ --model.max-delay=8
61
+ --model.use-depth
62
+ --model.depth-encoder-path=/root/workspace/Haoxuan/data/Psi0/cache/checkpoints/defm/defm_vit_s14.pth
63
+ --model.depth-encoder-input-size 196 350
64
+ --model.depth-fusion-mode=query_moe
65
+ --model.depth-query-count=16
66
+ --model.depth-query-target-count=8
67
+ --model.depth-query-score-dim=384
68
+ --model.depth-query-selection-weight=0.01
69
+ --model.depth-query-gate-init=1.0
70
+ --model.depth-query-sparsity-init=0.01
71
+ --model.depth-query-sparsity-rate=0.01
72
+ --model.depth-query-sparsity-warmup-steps=1000
73
+ --model.depth-query-recovery-after-steps=100
74
+ --model.depth-query-recovery-weight=0.01
75
+ --model.depth-query-recovery-window-steps=20
76
+ --model.depth-query-recovery-sparsity-scale=0.0
77
+ --model.depth-query-alive-margin=0.1
78
+ --model.depth-query-alive-weight=1.0
79
+ --model.depth-query-resampler-max-tokens=4096
80
+ --model.depth-query-orthogonality-weight=0
81
+ --model.depth-query-pre-orthogonality-weight=0.01
82
+ --model.no-depth-query-use-state
83
+ --model.no-depth-query-fuse
84
+ --model.no-depth-query-align-depth
85
+ --model.no-depth-query-selection
86
+ --model.depth-query-independent-rgb
87
+ --model.depth-query-concat-streams
88
+ --model.no-depth-query-cross-attention-fusion
89
+ --model.depth-query-pre-orthogonality
90
+ --model.no-depth-query-post-orthogonality
91
+ --model.depth-rgb-encoder-path=/root/workspace/Haoxuan/data/Psi0/cache/checkpoints/dinov2-small
92
+ --data.transform.model.depth-query-independent-rgb
93
+ --model.depth-module-lr=1e-4
94
+ --model.action-expert-lr=1e-5
depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4140c22762be59216aaa441514b16aecea7bbdea5d7e81af16d098e400ab30a6
3
+ size 6551390877
depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:731985a797c152a56f9e52966adcedb916ed1ef4a2dcc343f3cba7318de5359f
3
+ size 4156402059
depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff91055bb37a4b9dd4f0b82dd5a30e9c512f3f2d8eb1bf36ff654b4d900cf1c6
3
+ size 16577
depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_1.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:96c343a9f0b223fdc72e4739c8b425756fa7a5e14c2bba577d37bb9fb4ea490b
3
+ size 16513
depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_2.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f60f7babec160687a73cc73c8d9407d9f54634024e98bff2e58843801bbce11c
3
+ size 16513
depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_3.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:78ded609e9b6714fce88fff0c7eee43658a19b44d48c79c72eb324b86b25a05f
3
+ size 16513
depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_4.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9bb0d75064a32c820f950d6a3640559cc19f3902061fad1edd8bc597c5b71d4c
3
+ size 16513
depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_5.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7650692cc518c46f9184649e62ebfc6d0a5152240d7c62fe3f39741429d6236e
3
+ size 16513
depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_6.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14f0a7f50b2384ac9843b2eb4b662e44735b4830f85ee9cd40785c71e30013ee
3
+ size 16513
depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_7.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4cdff23f541682063858beb2f5c837300e0e0573315b147f7d5a2c5a3c4b2283
3
+ size 16513
depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f8d8bd1989d57762dae6637372ceb85ae82e6ffaf5bb23f2ae797f14be0add3
3
+ size 1465
depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/upload_manifest_full.json ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "files": {
3
+ "depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/argv.txt": {
4
+ "blob_id": "857e63b6ee17d590490a3dab58c618f14f26b051",
5
+ "sha256": "c6ceea7af9029c4267cd6f38bd1a69fcf91989412be7e99a974d5b6f95bfe45b",
6
+ "size": 3986
7
+ },
8
+ "depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/model.safetensors": {
9
+ "blob_id": "b9a1187e1149a6bd0adbe9067c3e1f600145ea53",
10
+ "sha256": "4140c22762be59216aaa441514b16aecea7bbdea5d7e81af16d098e400ab30a6",
11
+ "size": 6551390877
12
+ },
13
+ "depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/optimizer.bin": {
14
+ "blob_id": "6a480665a328786158fb1219f4c26a94a45c12a5",
15
+ "sha256": "731985a797c152a56f9e52966adcedb916ed1ef4a2dcc343f3cba7318de5359f",
16
+ "size": 4156402059
17
+ },
18
+ "depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_0.pkl": {
19
+ "blob_id": "c6ffc8c5164cdf335b64b7aa3c20cd0a48e19e62",
20
+ "sha256": "ff91055bb37a4b9dd4f0b82dd5a30e9c512f3f2d8eb1bf36ff654b4d900cf1c6",
21
+ "size": 16577
22
+ },
23
+ "depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_1.pkl": {
24
+ "blob_id": "7e767aef75543dddff95ddd5018cd2254cc34402",
25
+ "sha256": "96c343a9f0b223fdc72e4739c8b425756fa7a5e14c2bba577d37bb9fb4ea490b",
26
+ "size": 16513
27
+ },
28
+ "depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_2.pkl": {
29
+ "blob_id": "60c71d6315e7081e8d9ca0f7f4527876575d6dcb",
30
+ "sha256": "f60f7babec160687a73cc73c8d9407d9f54634024e98bff2e58843801bbce11c",
31
+ "size": 16513
32
+ },
33
+ "depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_3.pkl": {
34
+ "blob_id": "f7bc4571c377cc5563e35009b5b6e8ba7f29bfce",
35
+ "sha256": "78ded609e9b6714fce88fff0c7eee43658a19b44d48c79c72eb324b86b25a05f",
36
+ "size": 16513
37
+ },
38
+ "depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_4.pkl": {
39
+ "blob_id": "7a70022f3afdfdef2d402641eb0a7ff6d4656d5c",
40
+ "sha256": "9bb0d75064a32c820f950d6a3640559cc19f3902061fad1edd8bc597c5b71d4c",
41
+ "size": 16513
42
+ },
43
+ "depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_5.pkl": {
44
+ "blob_id": "fb964bea4f2881ddcf98eda8dff2316e3ace8ac5",
45
+ "sha256": "7650692cc518c46f9184649e62ebfc6d0a5152240d7c62fe3f39741429d6236e",
46
+ "size": 16513
47
+ },
48
+ "depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_6.pkl": {
49
+ "blob_id": "ca07ff7fb7950838b2ba52059ff3fd49cdf16f06",
50
+ "sha256": "14f0a7f50b2384ac9843b2eb4b662e44735b4830f85ee9cd40785c71e30013ee",
51
+ "size": 16513
52
+ },
53
+ "depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/random_states_7.pkl": {
54
+ "blob_id": "9faff70af5595f3196523b49b6162e644db4a5e5",
55
+ "sha256": "4cdff23f541682063858beb2f5c837300e0e0573315b147f7d5a2c5a3c4b2283",
56
+ "size": 16513
57
+ },
58
+ "depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/checkpoints/ckpt_20000/scheduler.bin": {
59
+ "blob_id": "47d1ba1f720d313acbfca2cacfd180fd49d74334",
60
+ "sha256": "1f8d8bd1989d57762dae6637372ceb85ae82e6ffaf5bb23f2ae797f14be0add3",
61
+ "size": 1465
62
+ },
63
+ "depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/run_config.json": {
64
+ "blob_id": "7959d40340c4a5d529d11ecdd53983972df7c1a7",
65
+ "sha256": "904610ba91d4ca8b95123457fba4ad6e55d059b8ed05dccdabdaea4bfc9594f4",
66
+ "size": 14201
67
+ }
68
+ },
69
+ "repo_type": "model",
70
+ "total_bytes": 10707944756
71
+ }
depth/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440/run_config.json ADDED
@@ -0,0 +1,401 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "exp": "G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2",
3
+ "seed": 292285,
4
+ "auto_tag_run": false,
5
+ "eval": false,
6
+ "debug": false,
7
+ "timestamp": "2610071440",
8
+ "log": {
9
+ "logging_dir": "logs",
10
+ "report_to": "wandb",
11
+ "log_freq": 100
12
+ },
13
+ "wandb": {
14
+ "project": "psi",
15
+ "entity": "2317600858-gdut",
16
+ "group": "finetune",
17
+ "id": "1l4n5uup",
18
+ "name": "G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610071440",
19
+ "resume": "allow"
20
+ },
21
+ "train": {
22
+ "num_workers": 4,
23
+ "overfit_single_batch": false,
24
+ "name": "finetune",
25
+ "resume_from_checkpoint": "/root/workspace/Haoxuan/data/Psi0/.runs/v6_concat_preorth_nonestate_all16_DINOv2/finetune/G1WholebodyXMovePickTeleop-v0-v6_concat_preorth_nonestate_all16_DINOv2.simple.flow1000.cosine.lr1.0e-04.b128.gpus8.2610061608/checkpoints/ckpt_18000",
26
+ "skip_resumed_steps": true,
27
+ "hf_token": ".hf_token",
28
+ "lora": false,
29
+ "output_dir": "/root/workspace/Haoxuan/data/Psi0/.runs/v6_concat_preorth_nonestate_all16_DINOv2",
30
+ "gradient_accumulation_steps": 1,
31
+ "mixed_precision": "bf16",
32
+ "max_grad_norm": 1.0,
33
+ "optimizer_foreach": false,
34
+ "train_batch_size": 16,
35
+ "val_batch_size": 16,
36
+ "val_num_batches": 20,
37
+ "checkpointing_steps": 1000,
38
+ "checkpoint_anchor_steps": 10000,
39
+ "max_checkpoints_to_keep": 4,
40
+ "validation_steps": 500,
41
+ "learning_rate": 0.0001,
42
+ "lr_scheduler_type": "cosine",
43
+ "lr_scheduler_kwargs": {
44
+ "betas": [
45
+ 0.95,
46
+ 0.999
47
+ ],
48
+ "weight_decay": 1e-6,
49
+ "eps": 1e-8
50
+ },
51
+ "scheduler_specific_kwargs": {},
52
+ "data_parallel": "ddp",
53
+ "sharding_strategy": "full-shard",
54
+ "deepspeed_config": "/root/workspace/Haoxuan/code/Psi0/.worktrees/feature/depth-v6-dinov2-query/src/InternVLA/config/deepseeds/zero3.json",
55
+ "enable_gradient_checkpointing": true,
56
+ "enable_mixed_precision_training": true,
57
+ "reduce_in_full_precision": true,
58
+ "max_training_steps": 40000,
59
+ "num_train_epochs": null,
60
+ "warmup_steps": 1000,
61
+ "warmup_ratio": null
62
+ },
63
+ "data": {
64
+ "transform": {
65
+ "repack": {
66
+ "dataset_name": "simple",
67
+ "num_past_frames": 0,
68
+ "action_chunk_size": 30,
69
+ "use_depth": true,
70
+ "depth_key": "observation.depth.egocentric",
71
+ "pad_action_dim": 36,
72
+ "pad_state_dim": 36
73
+ },
74
+ "model": {
75
+ "depth_query_independent_rgb": true,
76
+ "resize": {
77
+ "size": [
78
+ 180,
79
+ 320
80
+ ]
81
+ },
82
+ "center_crop": {
83
+ "size": [
84
+ 180,
85
+ 320
86
+ ]
87
+ },
88
+ "color_jitter": {
89
+ "brightness": 0.2,
90
+ "contrast": [
91
+ 0.8,
92
+ 1.2
93
+ ],
94
+ "saturation": [
95
+ 0.8,
96
+ 1.2
97
+ ],
98
+ "hue": 0.05
99
+ },
100
+ "gaussian_noise": {
101
+ "mean": 0.0,
102
+ "std": 3.0,
103
+ "prob_skip": 0.1
104
+ },
105
+ "img_aug": true,
106
+ "adaptive_resize": false,
107
+ "img_sizes": {
108
+ "egodex": [
109
+ 270,
110
+ 480
111
+ ],
112
+ "he": [
113
+ 240,
114
+ 320
115
+ ]
116
+ }
117
+ },
118
+ "field": {
119
+ "stat_path": "meta/stats_psi0.json",
120
+ "action_norm_type": "bounds",
121
+ "stat_action_key": "action",
122
+ "stat_state_key": "states",
123
+ "use_norm_mask": false,
124
+ "action_norm_masks": [
125
+ true,
126
+ true,
127
+ true,
128
+ true,
129
+ true,
130
+ true,
131
+ false
132
+ ],
133
+ "action_min": [
134
+ 0.0,
135
+ 0.0,
136
+ 0.0,
137
+ 0.0,
138
+ 0.0,
139
+ 0.0,
140
+ 0.0,
141
+ -0.5,
142
+ -0.699999988079071,
143
+ -0.699999988079071,
144
+ -8.807794347046262e-19,
145
+ -8.807794347046262e-19,
146
+ -3.4037360342851314e-19,
147
+ -8.807794347046262e-19,
148
+ -0.7234289646148682,
149
+ 0.1900009959936142,
150
+ -0.3478638231754303,
151
+ -0.24381141364574432,
152
+ -0.2871176600456238,
153
+ -0.46943801641464233,
154
+ -0.28036442399024963,
155
+ -0.28476205468177795,
156
+ -0.2781273424625397,
157
+ -0.6339197754859924,
158
+ -0.2889905273914337,
159
+ -0.2651921808719635,
160
+ -0.6309908628463745,
161
+ -0.19380050897598267,
162
+ -0.12375029176473618,
163
+ 0.004558231681585312,
164
+ -0.299873948097229,
165
+ 0.7400000095367432,
166
+ -5.537859719245743e-17,
167
+ 0.0,
168
+ -0.19462604820728302,
169
+ 0.0
170
+ ],
171
+ "action_max": [
172
+ 0.0,
173
+ 0.0,
174
+ 0.0,
175
+ 0.0,
176
+ 0.0,
177
+ 0.0,
178
+ 0.0,
179
+ 3.40585956201392e-19,
180
+ 4.519914941300404e-19,
181
+ 4.519914941300404e-19,
182
+ 1.5,
183
+ 1.5,
184
+ 0.6000000238418579,
185
+ 1.5,
186
+ 0.23390473425388336,
187
+ 0.6660271883010864,
188
+ 0.7131545543670654,
189
+ 0.6127181053161621,
190
+ 0.12901827692985535,
191
+ 0.23185482621192932,
192
+ 0.3763704001903534,
193
+ 0.2929672598838806,
194
+ -0.1900009959936142,
195
+ 0.3377506732940674,
196
+ 0.585555911064148,
197
+ 0.7287320494651794,
198
+ 0.2222127765417099,
199
+ 0.8800510764122009,
200
+ 0.14049682021141052,
201
+ 0.20396868884563446,
202
+ 0.30819037556648254,
203
+ 0.7400000095367432,
204
+ 0.5,
205
+ 0.0,
206
+ 0.18601855635643005,
207
+ 0.0
208
+ ],
209
+ "state_min": [
210
+ 7.777451855872641e-7,
211
+ -0.00033337774220854044,
212
+ -5.059851446276298e-6,
213
+ -0.00017079990357160568,
214
+ -9.60247780312784e-6,
215
+ -0.00018899694259744138,
216
+ -7.104898486431921e-6,
217
+ -0.5265660881996155,
218
+ -0.35556095838546753,
219
+ -0.5981534123420715,
220
+ -0.00021067277702968568,
221
+ -8.959311230682943e-7,
222
+ -8.316740604641382e-6,
223
+ -1.4921358797437279e-6,
224
+ -0.3891601860523224,
225
+ 0.150077685713768,
226
+ -0.3817761540412903,
227
+ -0.15058015286922455,
228
+ -0.2737894654273987,
229
+ -0.4105795621871948,
230
+ -0.2672588527202606,
231
+ -0.21800069510936737,
232
+ -0.26531729102134705,
233
+ -0.6129422783851624,
234
+ -0.2084185779094696,
235
+ -0.25410670042037964,
236
+ -0.5689104199409485,
237
+ -0.1778532713651657,
238
+ -0.09933578222990036,
239
+ 0.03510970249772072,
240
+ -0.30416738986968994,
241
+ 0.7400000095367432,
242
+ 0.0,
243
+ 0.0,
244
+ 0.0,
245
+ 0.0
246
+ ],
247
+ "state_max": [
248
+ 3.826543888862943e-6,
249
+ 0.003359275171533227,
250
+ 2.544531525927596e-6,
251
+ 0.00021755567286163568,
252
+ 3.435341568547301e-6,
253
+ 0.00019837530271615833,
254
+ 5.035403773945291e-6,
255
+ 7.077142640810052e-7,
256
+ 0.16438177227973938,
257
+ 6.031086172697542e-7,
258
+ 0.47953173518180847,
259
+ 1.3467339277267456,
260
+ 0.7186622619628906,
261
+ 1.1524261236190796,
262
+ 0.2516690492630005,
263
+ 0.39732399582862854,
264
+ 0.6492785811424255,
265
+ 0.6760621070861816,
266
+ 0.1240595132112503,
267
+ 0.22742627561092377,
268
+ 0.3518524467945099,
269
+ 0.31155094504356384,
270
+ -0.15275707840919495,
271
+ 0.3515203297138214,
272
+ 0.643481969833374,
273
+ 0.6658613681793213,
274
+ 0.2018662989139557,
275
+ 0.8597758412361145,
276
+ 0.13490912318229675,
277
+ 0.25929149985313416,
278
+ 0.31226828694343567,
279
+ 0.7400000095367432,
280
+ 0.0,
281
+ 0.0,
282
+ 0.0,
283
+ 0.0
284
+ ],
285
+ "normalize_state": true,
286
+ "pad_action_dim": 36,
287
+ "pad_state_dim": 36
288
+ }
289
+ },
290
+ "root_dir": "/root/workspace/Haoxuan/data/Psi0/data/simple-depth",
291
+ "train_repo_ids": [
292
+ "G1WholebodyXMovePickTeleop-v0"
293
+ ],
294
+ "val_repo_ids": [
295
+ "G1WholebodyXMovePickTeleop-v0"
296
+ ]
297
+ },
298
+ "model": {
299
+ "resnet_store_path": null,
300
+ "pretrained_action_header_path": "/root/workspace/Haoxuan/data/Psi0/cache/checkpoints/psi0/postpre.1by1.pad36.2601131206.ckpt.he30k",
301
+ "rtc": true,
302
+ "max_delay": 8,
303
+ "action_dim": 36,
304
+ "action_chunk_size": 30,
305
+ "action_exec_horizon": 30,
306
+ "observation_horizon": 1,
307
+ "img_chunk": 1,
308
+ "n_cams": 1,
309
+ "use_obs": "add_token",
310
+ "dropout": 0.1,
311
+ "noise_scheduler": "flow",
312
+ "train_diffusion_steps": 1000,
313
+ "eval_diffusion_steps": 10,
314
+ "share_cam_features": false,
315
+ "early_fusion": false,
316
+ "odim": 36,
317
+ "n_conditions": 0,
318
+ "token_fusion": "concat",
319
+ "loss_w": [
320
+ 0.1,
321
+ 0.2,
322
+ 0.1
323
+ ],
324
+ "time_dim": 256,
325
+ "hidden_dim": 1536,
326
+ "num_blocks": 6,
327
+ "dim_feedforward": 2048,
328
+ "nhead": 24,
329
+ "activation": "gelu",
330
+ "view_feature_dim": 2048,
331
+ "use_film": false,
332
+ "combined_temb": false,
333
+ "use_dit": false,
334
+ "use_depth": true,
335
+ "depth_encoder_name": "defm_vit_s14",
336
+ "depth_encoder_path": "/root/workspace/Haoxuan/data/Psi0/cache/checkpoints/defm/defm_vit_s14.pth",
337
+ "depth_encoder_dim": 384,
338
+ "depth_encoder_input_size": [
339
+ 196,
340
+ 350
341
+ ],
342
+ "depth_encoder_patch_size": 14,
343
+ "freeze_depth_encoder": true,
344
+ "depth_cross_attention_layers": [
345
+ 0,
346
+ 1,
347
+ 2,
348
+ 3,
349
+ 4,
350
+ 5
351
+ ],
352
+ "depth_fusion_mode": "query_moe",
353
+ "depth_query_count": 16,
354
+ "depth_query_target_count": 8.0,
355
+ "depth_query_score_dim": 384,
356
+ "depth_query_selection": false,
357
+ "depth_query_use_state": false,
358
+ "depth_query_fuse": false,
359
+ "depth_query_align_depth": false,
360
+ "depth_query_independent_rgb": true,
361
+ "depth_query_concat_streams": true,
362
+ "depth_query_cross_attention_fusion": false,
363
+ "depth_rgb_encoder_path": "/root/workspace/Haoxuan/data/Psi0/cache/checkpoints/dinov2-small",
364
+ "depth_query_pre_orthogonality": true,
365
+ "depth_query_pre_orthogonality_weight": 0.01,
366
+ "depth_query_post_orthogonality": false,
367
+ "depth_query_resampler_max_tokens": 4096,
368
+ "depth_query_gate_init": 1.0,
369
+ "depth_query_sparsity_init": 0.01,
370
+ "depth_query_sparsity_rate": 0.01,
371
+ "depth_query_sparsity_warmup_steps": 1000,
372
+ "depth_query_recovery_after_steps": 100,
373
+ "depth_query_recovery_weight": 0.01,
374
+ "depth_query_recovery_window_steps": 20,
375
+ "depth_query_recovery_sparsity_scale": 0.0,
376
+ "depth_query_alive_margin": 0.1,
377
+ "depth_query_alive_weight": 1.0,
378
+ "depth_query_selection_weight": 0.01,
379
+ "depth_query_orthogonality_weight": 0.0,
380
+ "depth_gate_init": 0.001,
381
+ "depth_module_lr": 0.0001,
382
+ "action_expert_lr": 0.00001,
383
+ "weight_decay": 0.01,
384
+ "model_name_or_path": "/root/workspace/Haoxuan/data/Psi0/cache/checkpoints/psi0/pre.fast.1by1.2601091803.ckpt.ego200k.he30k",
385
+ "vlm_ckpt_step": null,
386
+ "tune_vlm": false,
387
+ "tune_mm_llm": false,
388
+ "tune_mm_vision": false,
389
+ "tune_mm_mlp": false,
390
+ "gradient_checkpointing": true,
391
+ "lang_backbone_lr": 0.00001,
392
+ "mm_projector_lr": 0.00001,
393
+ "vision_tower_lr": 1e-6,
394
+ "optim": "adamw_torch",
395
+ "model_max_length": 4096,
396
+ "data_flatten": true,
397
+ "data_packing": true,
398
+ "max_pixels": 451584,
399
+ "min_pixels": 12544
400
+ }
401
+ }