Hugging Face's logo

anhdao69
/
LoopWAM_NT

Robotics
English
libero
loopwam
world-action-model
Model card Files
xet
 Community
anhdao69 commited on
Commit
46687ab
·
verified ·
1 Parent(s): 23eb766

Elastic E1: step_00002000 and configuration

Browse files
Files changed (6) hide show
  1. libero-all-suites/Elastic/E1/config.json +149 -0
  2. libero-all-suites/Elastic/E1/data/data_manifest.json +0 -0
  3. libero-all-suites/Elastic/E1/data/dataset_stats.json +50 -0
  4. libero-all-suites/Elastic/E1/dataset_stats.json +50 -0
  5. libero-all-suites/Elastic/E1/schedule_sequence.json +0 -0
  6. libero-all-suites/Elastic/E1/step_00002000.pt +3 -0
libero-all-suites/Elastic/E1/config.json ADDED
@@ -0,0 +1,149 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "asset_sha256": {
3
+ "teacher": "1c659a7c2f6970c3ea5972a8fba4abd9db35857c83a67a053c0ee35555fb602b",
4
+ "initial": null,
5
+ "vae": "38071ab59bd94681c686fa51d75a1968f64e470262043be31f7a094e442fd981",
6
+ "elastic_init": "b2815226865e5b14e9f0b449384d6f21cc20a2c6901636043db7110adc70076c"
7
+ },
8
+ "config": {
9
+ "kind": "T1",
10
+ "scope": "pc",
11
+ "regime": "fu",
12
+ "schedules": [
13
+ "flat_k1_n2",
14
+ "flat_k1_n5",
15
+ "flat_k1_n10",
16
+ "flat_k2_n2",
17
+ "flat_k2_n5",
18
+ "flat_k2_n10",
19
+ "flat_k4_n2",
20
+ "flat_k4_n5",
21
+ "flat_k4_n10"
22
+ ],
23
+ "step_cond": true,
24
+ "anchor": true,
25
+ "seed": 42,
26
+ "sampler": "balanced",
27
+ "init": "common_v1",
28
+ "reuse_teacher_prefix": true,
29
+ "collect_diagnostics": true,
30
+ "campaign": "elastic-v1"
31
+ },
32
+ "windows": 277713,
33
+ "world": 2,
34
+ "microbatch": 64,
35
+ "global_batch": 128,
36
+ "epochs": 10,
37
+ "planned_updates": 21700,
38
+ "teacher": "/lustre/fs1/groups/yshang/an221229/checkpoints/ChronoLoop/parent/libero-all-suites/v0-video4-action4/policy.pt",
39
+ "normalization_sha256": "d5ad351eb3272f416d34082e8af5483ba1118ceca355a37d9f268fdb269fb3e4",
40
+ "checkpoint_blocks": false,
41
+ "deterministic_algorithms": true,
42
+ "schedule_sequence_sha256": "37e93bb991623cb0a9d44757203307ee140abde4ea5833153ed86d146c8b9f39",
43
+ "checkpoint_min_blocks": 180,
44
+ "init_parts": [
45
+ "student_action",
46
+ "step_condition"
47
+ ],
48
+ "latent_content_digest": "f2b1b21be72694ec8e3fb217fbda12c61e7cc44d2e3c33f5d28c5e226ea8b341",
49
+ "run": "E1",
50
+ "arguments": {
51
+ "run_name": "E1",
52
+ "init_artifact": "/groups/yshang/an221229/checkpoints/LoopWAM_NT/elastic_work/init/elastic_common_init_v1.pt",
53
+ "checkpoint_min_blocks": 180,
54
+ "hf_folder": "https://huggingface.co/anhdao69/LoopWAM_NT/tree/main/libero-all-suites/Elastic/E1",
55
+ "output_dir": "/groups/yshang/an221229/checkpoints/LoopWAM_NT/elastic/E1",
56
+ "teacher": "/groups/yshang/an221229/checkpoints/ChronoLoop/parent/libero-all-suites/v0-video4-action4/policy.pt",
57
+ "vae_path": "/groups/yshang/an221229/checkpoints/RollingForcing/wan_models/Wan2.1-T2V-1.3B/Wan2.1_VAE.pth",
58
+ "manifest": "/groups/yshang/an221229/checkpoints/LoopWAM_NT/full_libero_v0_v4a2_20261026/train/data/data_manifest.json",
59
+ "latent_dir": "/dev/shm/elastic_latents_1080498",
60
+ "init_checkpoint": null,
61
+ "microbatch": 64,
62
+ "expected_gpus": 2,
63
+ "workers": 2,
64
+ "global_batch": 128,
65
+ "epochs": 10,
66
+ "save_every": 100,
67
+ "max_updates": null,
68
+ "resume": false,
69
+ "no_checkpoint_blocks": true,
70
+ "smoke": false
71
+ },
72
+ "trainable_parameters": 105678471,
73
+ "slurm_job_id": "1080498",
74
+ "git_revision": "db625c4e01c68f6b5fb5e0bd2b40582ae632d2f3",
75
+ "schedules": {
76
+ "flat_k1_n2": [
77
+ 1,
78
+ 1
79
+ ],
80
+ "flat_k1_n5": [
81
+ 1,
82
+ 1,
83
+ 1,
84
+ 1,
85
+ 1
86
+ ],
87
+ "flat_k1_n10": [
88
+ 1,
89
+ 1,
90
+ 1,
91
+ 1,
92
+ 1,
93
+ 1,
94
+ 1,
95
+ 1,
96
+ 1,
97
+ 1
98
+ ],
99
+ "flat_k2_n2": [
100
+ 2,
101
+ 2
102
+ ],
103
+ "flat_k2_n5": [
104
+ 2,
105
+ 2,
106
+ 2,
107
+ 2,
108
+ 2
109
+ ],
110
+ "flat_k2_n10": [
111
+ 2,
112
+ 2,
113
+ 2,
114
+ 2,
115
+ 2,
116
+ 2,
117
+ 2,
118
+ 2,
119
+ 2,
120
+ 2
121
+ ],
122
+ "flat_k4_n2": [
123
+ 4,
124
+ 4
125
+ ],
126
+ "flat_k4_n5": [
127
+ 4,
128
+ 4,
129
+ 4,
130
+ 4,
131
+ 4
132
+ ],
133
+ "flat_k4_n10": [
134
+ 4,
135
+ 4,
136
+ 4,
137
+ 4,
138
+ 4,
139
+ 4,
140
+ 4,
141
+ 4,
142
+ 4,
143
+ 4
144
+ ]
145
+ },
146
+ "gpu": "NVIDIA H100 80GB HBM3",
147
+ "torch_version": "2.7.1+cu128",
148
+ "hf_folder": "https://huggingface.co/anhdao69/LoopWAM_NT/tree/main/libero-all-suites/Elastic/E1"
149
+ }
libero-all-suites/Elastic/E1/data/data_manifest.json ADDED
The diff for this file is too large to render. See raw diff
 
libero-all-suites/Elastic/E1/data/dataset_stats.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "action": {
3
+ "default": {
4
+ "global_max": [
5
+ 0.9375,
6
+ 0.9375,
7
+ 0.9375,
8
+ 0.3557142913341522,
9
+ 0.375,
10
+ 0.375,
11
+ 1.0
12
+ ],
13
+ "global_min": [
14
+ -0.9375,
15
+ -0.9375,
16
+ -0.9375,
17
+ -0.24214285612106323,
18
+ -0.375,
19
+ -0.3642857074737549,
20
+ 0.0
21
+ ]
22
+ }
23
+ },
24
+ "num_episodes": 1712,
25
+ "num_transition": 277713,
26
+ "state": {
27
+ "default": {
28
+ "global_max": [
29
+ 0.2103137969970703,
30
+ 0.3904264271259308,
31
+ 1.472778081893921,
32
+ 3.7248642444610596,
33
+ 3.5618896484375,
34
+ 1.3863215446472168,
35
+ 0.04232141748070717,
36
+ 0.0013126095291227102
37
+ ],
38
+ "global_min": [
39
+ -0.48278069496154785,
40
+ -0.3309336006641388,
41
+ 0.008128181099891663,
42
+ 1.002794623374939,
43
+ -3.6312508583068848,
44
+ -1.842738389968872,
45
+ -0.005453015677630901,
46
+ -0.042015016078948975
47
+ ]
48
+ }
49
+ }
50
+ }
libero-all-suites/Elastic/E1/dataset_stats.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "action": {
3
+ "default": {
4
+ "global_max": [
5
+ 0.9375,
6
+ 0.9375,
7
+ 0.9375,
8
+ 0.3557142913341522,
9
+ 0.375,
10
+ 0.375,
11
+ 1.0
12
+ ],
13
+ "global_min": [
14
+ -0.9375,
15
+ -0.9375,
16
+ -0.9375,
17
+ -0.24214285612106323,
18
+ -0.375,
19
+ -0.3642857074737549,
20
+ 0.0
21
+ ]
22
+ }
23
+ },
24
+ "num_episodes": 1712,
25
+ "num_transition": 277713,
26
+ "state": {
27
+ "default": {
28
+ "global_max": [
29
+ 0.2103137969970703,
30
+ 0.3904264271259308,
31
+ 1.472778081893921,
32
+ 3.7248642444610596,
33
+ 3.5618896484375,
34
+ 1.3863215446472168,
35
+ 0.04232141748070717,
36
+ 0.0013126095291227102
37
+ ],
38
+ "global_min": [
39
+ -0.48278069496154785,
40
+ -0.3309336006641388,
41
+ 0.008128181099891663,
42
+ 1.002794623374939,
43
+ -3.6312508583068848,
44
+ -1.842738389968872,
45
+ -0.005453015677630901,
46
+ -0.042015016078948975
47
+ ]
48
+ }
49
+ }
50
+ }
libero-all-suites/Elastic/E1/schedule_sequence.json ADDED
The diff for this file is too large to render. See raw diff
 
libero-all-suites/Elastic/E1/step_00002000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a2017a40f06a9ea9c0e30136efd232665c8e9a8cb7baa0daf272643c5f005373
3
+ size 2761414261