KasLoot commited on
Commit
5633c2e
·
verified ·
1 Parent(s): 2d84c01

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ training_curves.png filter=lfs diff=lfs merge=lfs -text
37
+ wandb/run-20260713_181849-o7ouu23c/run-o7ouu23c.wandb filter=lfs diff=lfs merge=lfs -text
assets/arx_l5/norm_stats.json ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "norm_stats": {
3
+ "state": {
4
+ "mean": [
5
+ 0.024539707228541374,
6
+ 1.6003690958023071,
7
+ 1.2549989223480225,
8
+ -1.1882739067077637,
9
+ 0.007991060614585876,
10
+ 0.028123868629336357,
11
+ 0.015925440937280655
12
+ ],
13
+ "std": [
14
+ 0.49538034200668335,
15
+ 0.25768405199050903,
16
+ 0.25095808506011963,
17
+ 0.22862063348293304,
18
+ 0.05344964563846588,
19
+ 0.6481392979621887,
20
+ 0.014971424825489521
21
+ ],
22
+ "q01": [
23
+ -0.6704974174499512,
24
+ 0.23925141990184784,
25
+ 0.2808642089366913,
26
+ -1.423521637916565,
27
+ -0.016070658341050148,
28
+ -1.2652803659439087,
29
+ 0.0
30
+ ],
31
+ "q99": [
32
+ 0.8035353422164917,
33
+ 1.9749181270599365,
34
+ 1.7320492267608643,
35
+ -0.03111369162797928,
36
+ 0.29005691409111023,
37
+ 1.3111636638641357,
38
+ 0.029999999329447746
39
+ ]
40
+ },
41
+ "actions": {
42
+ "mean": [
43
+ 0.022642292082309723,
44
+ 1.5445128679275513,
45
+ 1.3158354759216309,
46
+ -1.1841663122177124,
47
+ 0.008151013404130936,
48
+ 0.027992013841867447,
49
+ 0.015925440937280655
50
+ ],
51
+ "std": [
52
+ 0.49655041098594666,
53
+ 0.2320813685655594,
54
+ 0.24771882593631744,
55
+ 0.22698715329170227,
56
+ 0.05510450899600983,
57
+ 0.6484112739562988,
58
+ 0.014971424825489521
59
+ ],
60
+ "q01": [
61
+ -0.670513927936554,
62
+ 0.25099998712539673,
63
+ 0.3140000104904175,
64
+ -1.4189491271972656,
65
+ -0.0074226888827979565,
66
+ -1.2653172016143799,
67
+ 0.0
68
+ ],
69
+ "q99": [
70
+ 0.8035446405410767,
71
+ 1.8864666223526,
72
+ 1.7942811250686646,
73
+ 0.0,
74
+ 0.29703888297080994,
75
+ 1.3112030029296875,
76
+ 0.029999999329447746
77
+ ]
78
+ }
79
+ }
80
+ }
best/assets/arx_l5/norm_stats.json ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "norm_stats": {
3
+ "state": {
4
+ "mean": [
5
+ 0.024539707228541374,
6
+ 1.6003690958023071,
7
+ 1.2549989223480225,
8
+ -1.1882739067077637,
9
+ 0.007991060614585876,
10
+ 0.028123868629336357,
11
+ 0.015925440937280655
12
+ ],
13
+ "std": [
14
+ 0.49538034200668335,
15
+ 0.25768405199050903,
16
+ 0.25095808506011963,
17
+ 0.22862063348293304,
18
+ 0.05344964563846588,
19
+ 0.6481392979621887,
20
+ 0.014971424825489521
21
+ ],
22
+ "q01": [
23
+ -0.6704974174499512,
24
+ 0.23925141990184784,
25
+ 0.2808642089366913,
26
+ -1.423521637916565,
27
+ -0.016070658341050148,
28
+ -1.2652803659439087,
29
+ 0.0
30
+ ],
31
+ "q99": [
32
+ 0.8035353422164917,
33
+ 1.9749181270599365,
34
+ 1.7320492267608643,
35
+ -0.03111369162797928,
36
+ 0.29005691409111023,
37
+ 1.3111636638641357,
38
+ 0.029999999329447746
39
+ ]
40
+ },
41
+ "actions": {
42
+ "mean": [
43
+ 0.022642292082309723,
44
+ 1.5445128679275513,
45
+ 1.3158354759216309,
46
+ -1.1841663122177124,
47
+ 0.008151013404130936,
48
+ 0.027992013841867447,
49
+ 0.015925440937280655
50
+ ],
51
+ "std": [
52
+ 0.49655041098594666,
53
+ 0.2320813685655594,
54
+ 0.24771882593631744,
55
+ 0.22698715329170227,
56
+ 0.05510450899600983,
57
+ 0.6484112739562988,
58
+ 0.014971424825489521
59
+ ],
60
+ "q01": [
61
+ -0.670513927936554,
62
+ 0.25099998712539673,
63
+ 0.3140000104904175,
64
+ -1.4189491271972656,
65
+ -0.0074226888827979565,
66
+ -1.2653172016143799,
67
+ 0.0
68
+ ],
69
+ "q99": [
70
+ 0.8035446405410767,
71
+ 1.8864666223526,
72
+ 1.7942811250686646,
73
+ 0.0,
74
+ 0.29703888297080994,
75
+ 1.3112030029296875,
76
+ 0.029999999329447746
77
+ ]
78
+ }
79
+ }
80
+ }
best/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37cfd7eae491c4b88d5bcdf8ecb4d5e0ef41ef2b180b7764bfafd74202396cfb
3
+ size 6707097850
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": "/workspace/datasets/RoboPref_dataset/stacking_blocks_ordered",
3
+ "checkpoint_in": "/workspace/models/pi05_base_pytorch",
4
+ "checkpoint_out": "/workspace/models/pi05_arx_ordered",
5
+ "norm_stats": "/workspace/datasets/RoboPref_dataset/stacking_blocks_ordered/norm_stats_arx_l5.json",
6
+ "tokenizer": "/root/.cache/openpi/big_vision/paligemma_tokenizer.model",
7
+ "steps": 15000,
8
+ "micro_batch": 32,
9
+ "accum": 1,
10
+ "lr": 2.5e-05,
11
+ "warmup": 1000,
12
+ "final_lr_frac": 0.1,
13
+ "grad_clip": 1.0,
14
+ "num_workers": 12,
15
+ "seed": 0,
16
+ "device": "cuda",
17
+ "offload_optim": false,
18
+ "ema": null,
19
+ "resume": false,
20
+ "overfit_episode": null,
21
+ "log_every": 20,
22
+ "val_every": 1000,
23
+ "action_val_every": 5000,
24
+ "save_every": 2500,
25
+ "val_samples": 512,
26
+ "action_val_samples": 64,
27
+ "wandb_project": "robopref",
28
+ "wandb_entity": null,
29
+ "wandb_mode": "online",
30
+ "embodiment": "arx_l5",
31
+ "git": "6993e1c1ff942e2857950bba5cb76ac6015520a7",
32
+ "train_samples": 183984
33
+ }
model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37cfd7eae491c4b88d5bcdf8ecb4d5e0ef41ef2b180b7764bfafd74202396cfb
3
+ size 6707097850
train_log.jsonl ADDED
The diff for this file is too large to render. See raw diff
 
train_state.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:79e27e15ae17a13c2e3d57e9ae54cf97bf28a6b38de85f11ab11b99057bce9ca
3
+ size 5161343763
training_curves.png ADDED

Git LFS Details

  • SHA256: a0ebf6204f563a8a7e161dd5b4b82ab7aff7ca586d5393cf5a61ba2d8868a4d9
  • Pointer size: 131 Bytes
  • Size of remote file: 115 kB
wandb/debug-internal.log ADDED
The diff for this file is too large to render. See raw diff
 
wandb/debug.log ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-07-13 18:18:49,968 INFO MainThread:31979 [wandb_setup.py:_flush():81] Current SDK version is 0.28.0
2
+ 2026-07-13 18:18:49,969 INFO MainThread:31979 [wandb_setup.py:_flush():81] Configure stats pid to 31979
3
+ 2026-07-13 18:18:49,969 INFO MainThread:31979 [wandb_setup.py:_flush():81] Loading settings from environment variables
4
+ 2026-07-13 18:18:49,970 INFO MainThread:31979 [wandb_init.py:setup_run_log_directory():725] Logging user logs to /workspace/models/pi05_arx_ordered/wandb/run-20260713_181849-o7ouu23c/logs/debug.log
5
+ 2026-07-13 18:18:49,971 INFO MainThread:31979 [wandb_init.py:setup_run_log_directory():726] Logging internal logs to /workspace/models/pi05_arx_ordered/wandb/run-20260713_181849-o7ouu23c/logs/debug-internal.log
6
+ 2026-07-13 18:18:49,972 INFO MainThread:31979 [wandb_init.py:init():768] calling init triggers
7
+ 2026-07-13 18:18:49,972 INFO MainThread:31979 [wandb_init.py:init():773] wandb.init called with sweep_config: {}
8
+ config: {'dataset': '/workspace/datasets/RoboPref_dataset/stacking_blocks_ordered', 'checkpoint_in': '/workspace/models/pi05_base_pytorch', 'checkpoint_out': '/workspace/models/pi05_arx_ordered', 'norm_stats': None, 'tokenizer': '/root/.cache/openpi/big_vision/paligemma_tokenizer.model', 'steps': 15000, 'micro_batch': 32, 'accum': 1, 'lr': 2.5e-05, 'warmup': 1000, 'final_lr_frac': 0.1, 'grad_clip': 1.0, 'num_workers': 12, 'seed': 0, 'device': 'cuda', 'offload_optim': False, 'ema': None, 'resume': False, 'overfit_episode': None, 'log_every': 20, 'val_every': 1000, 'action_val_every': 5000, 'save_every': 2500, 'val_samples': 512, 'action_val_samples': 64, 'wandb_project': 'robopref', 'wandb_entity': None, 'wandb_mode': 'online', 'embodiment': 'arx_l5', 'git': '6993e1c1ff942e2857950bba5cb76ac6015520a7', 'train_samples': 183984, '_wandb': {}}
9
+ 2026-07-13 18:18:49,972 INFO MainThread:31979 [wandb_init.py:init():816] starting backend
10
+ 2026-07-13 18:18:49,973 INFO MainThread:31979 [wandb_init.py:init():831] sending inform_init request
11
+ 2026-07-13 18:18:50,441 INFO MainThread:31979 [wandb_init.py:init():836] backend started and connected
12
+ 2026-07-13 18:18:50,551 INFO MainThread:31979 [wandb_init.py:init():906] updated telemetry
13
+ 2026-07-13 18:18:50,597 INFO MainThread:31979 [wandb_init.py:init():929] communicating run to backend with 90.0 second timeout
14
+ 2026-07-13 18:18:51,325 INFO MainThread:31979 [wandb_init.py:init():974] starting run threads in backend
15
+ 2026-07-13 18:18:51,909 INFO MainThread:31979 [wandb_run.py:_console_start():2523] atexit reg
16
+ 2026-07-13 18:18:51,909 INFO MainThread:31979 [wandb_run.py:_redirect():2373] redirect: wrap_raw
17
+ 2026-07-13 18:18:51,909 INFO MainThread:31979 [wandb_run.py:_redirect():2442] Wrapping output streams.
18
+ 2026-07-13 18:18:51,910 INFO MainThread:31979 [wandb_run.py:_redirect():2465] Redirects installed.
19
+ 2026-07-13 18:18:51,923 INFO MainThread:31979 [wandb_init.py:init():1012] run started, returning control to user process
20
+ 2026-07-14 02:36:38,976 INFO MainThread:31979 [wandb_run.py:_finish():2285] finishing run kasloot-ai/robopref/o7ouu23c
21
+ 2026-07-14 02:36:38,977 INFO MainThread:31979 [wandb_run.py:_atexit_cleanup():2490] got exitcode: 0
22
+ 2026-07-14 02:36:38,978 INFO MainThread:31979 [wandb_run.py:_restore():2472] restore
23
+ 2026-07-14 02:36:38,978 INFO MainThread:31979 [wandb_run.py:_restore():2478] restore done
24
+ 2026-07-14 02:36:40,963 INFO MainThread:31979 [wandb_run.py:_footer_sync_info():3895] logging synced files
wandb/run-20260713_181849-o7ouu23c/files/config.yaml ADDED
@@ -0,0 +1,118 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _wandb:
2
+ value:
3
+ cli_version: 0.28.0
4
+ e:
5
+ uo883ucaprnzyf6co4ecgxjfqtuxcu9t:
6
+ args:
7
+ - --dataset
8
+ - /workspace/datasets/RoboPref_dataset/stacking_blocks_ordered
9
+ - --checkpoint-in
10
+ - /workspace/models/pi05_base_pytorch
11
+ - --checkpoint-out
12
+ - /workspace/models/pi05_arx_ordered
13
+ cpu_count: 128
14
+ cpu_count_logical: 256
15
+ cudaVersion: "13.0"
16
+ disk:
17
+ /:
18
+ total: "32212254720"
19
+ used: "1706278912"
20
+ email: steven.liu.yuxin@outlook.com
21
+ executable: /workspace/RoboPref/.venv/bin/python3
22
+ git:
23
+ commit: 6993e1c1ff942e2857950bba5cb76ac6015520a7
24
+ remote: https://github.com/KasLoot/RoboPref.git
25
+ gpu: NVIDIA GeForce RTX 3090
26
+ gpu_count: 1
27
+ gpu_nvidia:
28
+ - architecture: Ampere
29
+ cudaCores: 10496
30
+ memoryTotal: "25769803776"
31
+ name: NVIDIA GeForce RTX 3090
32
+ uuid: GPU-e4ce1638-ee86-83b3-59fb-b8cdb6ab2201
33
+ host: 207692caf9fb
34
+ memory:
35
+ total: "1081934213120"
36
+ os: Linux-6.8.0-110-generic-x86_64-with-glibc2.39
37
+ program: -m pi05.train
38
+ python: CPython 3.13.8
39
+ root: /workspace/models/pi05_arx_ordered
40
+ startedAt: "2026-07-13T18:18:49.949416Z"
41
+ writerId: uo883ucaprnzyf6co4ecgxjfqtuxcu9t
42
+ m: []
43
+ python_version: 3.13.8
44
+ t:
45
+ "1":
46
+ - 1
47
+ "2":
48
+ - 1
49
+ "3":
50
+ - 2
51
+ - 16
52
+ - 61
53
+ "4": 3.13.8
54
+ "5": 0.28.0
55
+ "12": 0.28.0
56
+ "13": linux-x86_64
57
+ accum:
58
+ value: 1
59
+ action_val_every:
60
+ value: 5000
61
+ action_val_samples:
62
+ value: 64
63
+ checkpoint_in:
64
+ value: /workspace/models/pi05_base_pytorch
65
+ checkpoint_out:
66
+ value: /workspace/models/pi05_arx_ordered
67
+ dataset:
68
+ value: /workspace/datasets/RoboPref_dataset/stacking_blocks_ordered
69
+ device:
70
+ value: cuda
71
+ ema:
72
+ value: null
73
+ embodiment:
74
+ value: arx_l5
75
+ final_lr_frac:
76
+ value: 0.1
77
+ git:
78
+ value: 6993e1c1ff942e2857950bba5cb76ac6015520a7
79
+ grad_clip:
80
+ value: 1
81
+ log_every:
82
+ value: 20
83
+ lr:
84
+ value: 2.5e-05
85
+ micro_batch:
86
+ value: 32
87
+ norm_stats:
88
+ value: null
89
+ num_workers:
90
+ value: 12
91
+ offload_optim:
92
+ value: false
93
+ overfit_episode:
94
+ value: null
95
+ resume:
96
+ value: false
97
+ save_every:
98
+ value: 2500
99
+ seed:
100
+ value: 0
101
+ steps:
102
+ value: 15000
103
+ tokenizer:
104
+ value: /root/.cache/openpi/big_vision/paligemma_tokenizer.model
105
+ train_samples:
106
+ value: 183984
107
+ val_every:
108
+ value: 1000
109
+ val_samples:
110
+ value: 512
111
+ wandb_entity:
112
+ value: null
113
+ wandb_mode:
114
+ value: online
115
+ wandb_project:
116
+ value: robopref
117
+ warmup:
118
+ value: 1000
wandb/run-20260713_181849-o7ouu23c/files/output.log ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ stacking_blocks_ordered: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 15000/15000 [8:17:46<00:00, 1.99s/step, loss=0.0015 lr=2.50e-06 gnorm=0.04]
2
+ [val @ 1000] val_flow_mse=0.005433
3
+ [val @ 2000] val_flow_mse=0.00438
4
+ [ckpt @ 2500] saved /workspace/models/pi05_arx_ordered
5
+ [val @ 3000] val_flow_mse=0.003085
6
+ [val @ 4000] val_flow_mse=0.002829
7
+ [val @ 5000] val_flow_mse=0.002425 val_action_mae=0.020015 best=True
8
+ [ckpt @ 5000] saved /workspace/models/pi05_arx_ordered
9
+ [val @ 6000] val_flow_mse=0.002111
10
+ [val @ 7000] val_flow_mse=0.002
11
+ [ckpt @ 7500] saved /workspace/models/pi05_arx_ordered
12
+ [val @ 8000] val_flow_mse=0.002003
13
+ [val @ 9000] val_flow_mse=0.001819
14
+ [val @ 10000] val_flow_mse=0.001662 val_action_mae=0.013208 best=True
15
+ [ckpt @ 10000] saved /workspace/models/pi05_arx_ordered
16
+ [val @ 11000] val_flow_mse=0.001598
17
+ [val @ 12000] val_flow_mse=0.001532
18
+ [ckpt @ 12500] saved /workspace/models/pi05_arx_ordered
19
+ [val @ 13000] val_flow_mse=0.00146
20
+ [val @ 14000] val_flow_mse=0.00147
21
+ [val @ 15000] val_flow_mse=0.00136 val_action_mae=0.011581 best=True
22
+ [ckpt @ 15000] saved /workspace/models/pi05_arx_ordered
wandb/run-20260713_181849-o7ouu23c/files/requirements.txt ADDED
@@ -0,0 +1,96 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ nvidia-nvshmem-cu12==3.4.5
2
+ torch==2.13.0+cu126
3
+ torchvision==0.28.0+cu126
4
+ triton==3.7.1
5
+ nvidia-cublas-cu12==12.6.4.1
6
+ sympy==1.14.0
7
+ nvidia-cufft-cu12==11.3.0.4
8
+ networkx==3.6.1
9
+ setuptools==78.1.0
10
+ nvidia-cuda-nvrtc-cu12==12.6.85
11
+ nvidia-nvtx-cu12==12.6.77
12
+ nvidia-nccl-cu12==2.29.3
13
+ cuda-bindings==12.9.4
14
+ nvidia-cudnn-cu12==9.10.2.21
15
+ nvidia-cusparselt-cu12==0.7.1
16
+ nvidia-curand-cu12==10.3.7.77
17
+ MarkupSafe==3.0.3
18
+ nvidia-cusparse-cu12==12.5.4.2
19
+ nvidia-cusolver-cu12==11.7.1.2
20
+ nvidia-nvjitlink-cu12==12.6.85
21
+ nvidia-cufile-cu12==1.11.1.6
22
+ cuda-toolkit==12.6.3
23
+ mpmath==1.3.0
24
+ nvidia-cuda-cupti-cu12==12.6.80
25
+ filelock==3.29.0
26
+ cuda-pathfinder==1.2.2
27
+ Jinja2==3.1.6
28
+ nvidia-cuda-runtime-cu12==12.6.77
29
+ charset-normalizer==3.4.9
30
+ matplotlib==3.11.0
31
+ platformdirs==4.10.0
32
+ GitPython==3.1.51
33
+ wandb==0.28.0
34
+ sentry-sdk==2.64.0
35
+ click==8.4.2
36
+ cycler==0.12.1
37
+ pyparsing==3.3.2
38
+ fonttools==4.63.0
39
+ contourpy==1.3.3
40
+ urllib3==2.7.0
41
+ gitdb==4.0.12
42
+ requests==2.34.2
43
+ python-dateutil==2.9.0.post0
44
+ six==1.17.0
45
+ packaging==26.2
46
+ kiwisolver==1.5.0
47
+ smmap==5.0.3
48
+ protobuf==7.35.1
49
+ etils==1.14.0
50
+ scipy==1.18.0
51
+ aiofiles==25.1.0
52
+ humanize==4.16.0
53
+ orbax-checkpoint==0.12.1
54
+ sentencepiece==0.2.2
55
+ ollama==0.6.2
56
+ annotated-types==0.7.0
57
+ httpcore==1.0.9
58
+ PyOpenGL==3.1.10
59
+ h5py==3.16.0
60
+ typing_extensions==4.16.0
61
+ typing-inspection==0.4.2
62
+ zipp==4.1.0
63
+ imageio-ffmpeg==0.6.0
64
+ tensorstore==0.1.84
65
+ tqdm==4.68.3
66
+ sniffio==1.3.1
67
+ anyio==4.14.1
68
+ pydantic==2.13.4
69
+ psutil==7.2.2
70
+ pillow==12.3.0
71
+ jax==0.10.2
72
+ h11==0.16.0
73
+ ml_dtypes==0.5.4
74
+ jiter==0.16.0
75
+ mujoco==3.10.0
76
+ colorama==0.4.6
77
+ pydantic_core==2.46.4
78
+ glfw==2.10.0
79
+ msgpack==1.2.1
80
+ orbax==0.1.9
81
+ httpx==0.28.1
82
+ openai==2.44.0
83
+ fsspec==2026.6.0
84
+ distro==1.9.0
85
+ numpy==2.5.1
86
+ prometheus_client==0.25.0
87
+ opt_einsum==3.4.0
88
+ pillow_heif==1.4.0
89
+ PyYAML==6.0.3
90
+ ImageIO==2.37.3
91
+ absl-py==2.5.0
92
+ jaxlib==0.10.2
93
+ idna==3.18
94
+ simplejson==4.1.1
95
+ certifi==2026.6.17
96
+ uvloop==0.22.1
wandb/run-20260713_181849-o7ouu23c/files/wandb-metadata.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-6.8.0-110-generic-x86_64-with-glibc2.39",
3
+ "python": "CPython 3.13.8",
4
+ "startedAt": "2026-07-13T18:18:49.949416Z",
5
+ "args": [
6
+ "--dataset",
7
+ "/workspace/datasets/RoboPref_dataset/stacking_blocks_ordered",
8
+ "--checkpoint-in",
9
+ "/workspace/models/pi05_base_pytorch",
10
+ "--checkpoint-out",
11
+ "/workspace/models/pi05_arx_ordered"
12
+ ],
13
+ "program": "-m pi05.train",
14
+ "git": {
15
+ "remote": "https://github.com/KasLoot/RoboPref.git",
16
+ "commit": "6993e1c1ff942e2857950bba5cb76ac6015520a7"
17
+ },
18
+ "email": "steven.liu.yuxin@outlook.com",
19
+ "root": "/workspace/models/pi05_arx_ordered",
20
+ "host": "207692caf9fb",
21
+ "executable": "/workspace/RoboPref/.venv/bin/python3",
22
+ "cpu_count": 128,
23
+ "cpu_count_logical": 256,
24
+ "gpu": "NVIDIA GeForce RTX 3090",
25
+ "gpu_count": 1,
26
+ "disk": {
27
+ "/": {
28
+ "total": "32212254720",
29
+ "used": "1706278912"
30
+ }
31
+ },
32
+ "memory": {
33
+ "total": "1081934213120"
34
+ },
35
+ "gpu_nvidia": [
36
+ {
37
+ "name": "NVIDIA GeForce RTX 3090",
38
+ "memoryTotal": "25769803776",
39
+ "cudaCores": 10496,
40
+ "architecture": "Ampere",
41
+ "uuid": "GPU-e4ce1638-ee86-83b3-59fb-b8cdb6ab2201"
42
+ }
43
+ ],
44
+ "cudaVersion": "13.0",
45
+ "writerId": "uo883ucaprnzyf6co4ecgxjfqtuxcu9t"
46
+ }
wandb/run-20260713_181849-o7ouu23c/files/wandb-summary.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"step":15000,"loss":0.002672,"_step":15000,"loss_mean50":0.001498,"wall_s":29780.8,"samples":480000,"cuda_gb":16.86,"grad_norm":0.04,"val_flow_mse":0.00136,"_wandb":{"runtime":29867},"best":true,"lr":2.5000002832475742e-06,"val_action_mae":0.011581,"_timestamp":1.7839965683147085e+09,"_runtime":29867.65266831}
wandb/run-20260713_181849-o7ouu23c/logs/debug-core.log ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-07-13T18:18:49.514134442Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpw7gjocdp/port-31979.txt","pid":31979,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
2
+ {"time":"2026-07-13T18:18:49.518153276Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":31979}
3
+ {"time":"2026-07-13T18:18:49.518106736Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-31979-32197-3523898553/socket","Net":"unix"}}
4
+ {"time":"2026-07-13T18:18:49.62473879Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
5
+ {"time":"2026-07-13T18:18:49.979315122Z","level":"INFO","msg":"handleInformInit: received","streamId":"o7ouu23c","id":"1(@)"}
6
+ {"time":"2026-07-13T18:18:50.440947856Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"o7ouu23c","id":"1(@)"}
7
+ {"time":"2026-07-13T18:18:56.994984955Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"62wrb8p6bar2"}
8
+ {"time":"2026-07-14T02:36:38.978468557Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"62wrb8p6bar2"}
9
+ {"time":"2026-07-14T02:36:40.961590621Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"62wrb8p6bar2"}
10
+ {"time":"2026-07-14T02:36:40.966496282Z","level":"INFO","msg":"handleInformFinish: finish message received","streamId":"o7ouu23c","id":"1(@)"}
11
+ {"time":"2026-07-14T02:36:40.972480718Z","level":"INFO","msg":"handleInformFinish: stream closed","streamId":"o7ouu23c","id":"1(@)"}
12
+ {"time":"2026-07-14T02:36:41.675279702Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
13
+ {"time":"2026-07-14T02:36:41.675360472Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
14
+ {"time":"2026-07-14T02:36:41.675380351Z","level":"INFO","msg":"server is shutting down"}
15
+ {"time":"2026-07-14T02:36:41.675393821Z","level":"INFO","msg":"processOutgoingData: finished","id":"1(@)"}
16
+ {"time":"2026-07-14T02:36:41.675371271Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
17
+ {"time":"2026-07-14T02:36:41.675452551Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
18
+ {"time":"2026-07-14T02:36:41.675459131Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
19
+ {"time":"2026-07-14T02:36:41.67563466Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-31979-32197-3523898553/socket","Net":"unix"}}
20
+ {"time":"2026-07-14T02:36:41.67567635Z","level":"INFO","msg":"server is closed"}
wandb/run-20260713_181849-o7ouu23c/logs/debug-internal.log ADDED
The diff for this file is too large to render. See raw diff
 
wandb/run-20260713_181849-o7ouu23c/logs/debug.log ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-07-13 18:18:49,968 INFO MainThread:31979 [wandb_setup.py:_flush():81] Current SDK version is 0.28.0
2
+ 2026-07-13 18:18:49,969 INFO MainThread:31979 [wandb_setup.py:_flush():81] Configure stats pid to 31979
3
+ 2026-07-13 18:18:49,969 INFO MainThread:31979 [wandb_setup.py:_flush():81] Loading settings from environment variables
4
+ 2026-07-13 18:18:49,970 INFO MainThread:31979 [wandb_init.py:setup_run_log_directory():725] Logging user logs to /workspace/models/pi05_arx_ordered/wandb/run-20260713_181849-o7ouu23c/logs/debug.log
5
+ 2026-07-13 18:18:49,971 INFO MainThread:31979 [wandb_init.py:setup_run_log_directory():726] Logging internal logs to /workspace/models/pi05_arx_ordered/wandb/run-20260713_181849-o7ouu23c/logs/debug-internal.log
6
+ 2026-07-13 18:18:49,972 INFO MainThread:31979 [wandb_init.py:init():768] calling init triggers
7
+ 2026-07-13 18:18:49,972 INFO MainThread:31979 [wandb_init.py:init():773] wandb.init called with sweep_config: {}
8
+ config: {'dataset': '/workspace/datasets/RoboPref_dataset/stacking_blocks_ordered', 'checkpoint_in': '/workspace/models/pi05_base_pytorch', 'checkpoint_out': '/workspace/models/pi05_arx_ordered', 'norm_stats': None, 'tokenizer': '/root/.cache/openpi/big_vision/paligemma_tokenizer.model', 'steps': 15000, 'micro_batch': 32, 'accum': 1, 'lr': 2.5e-05, 'warmup': 1000, 'final_lr_frac': 0.1, 'grad_clip': 1.0, 'num_workers': 12, 'seed': 0, 'device': 'cuda', 'offload_optim': False, 'ema': None, 'resume': False, 'overfit_episode': None, 'log_every': 20, 'val_every': 1000, 'action_val_every': 5000, 'save_every': 2500, 'val_samples': 512, 'action_val_samples': 64, 'wandb_project': 'robopref', 'wandb_entity': None, 'wandb_mode': 'online', 'embodiment': 'arx_l5', 'git': '6993e1c1ff942e2857950bba5cb76ac6015520a7', 'train_samples': 183984, '_wandb': {}}
9
+ 2026-07-13 18:18:49,972 INFO MainThread:31979 [wandb_init.py:init():816] starting backend
10
+ 2026-07-13 18:18:49,973 INFO MainThread:31979 [wandb_init.py:init():831] sending inform_init request
11
+ 2026-07-13 18:18:50,441 INFO MainThread:31979 [wandb_init.py:init():836] backend started and connected
12
+ 2026-07-13 18:18:50,551 INFO MainThread:31979 [wandb_init.py:init():906] updated telemetry
13
+ 2026-07-13 18:18:50,597 INFO MainThread:31979 [wandb_init.py:init():929] communicating run to backend with 90.0 second timeout
14
+ 2026-07-13 18:18:51,325 INFO MainThread:31979 [wandb_init.py:init():974] starting run threads in backend
15
+ 2026-07-13 18:18:51,909 INFO MainThread:31979 [wandb_run.py:_console_start():2523] atexit reg
16
+ 2026-07-13 18:18:51,909 INFO MainThread:31979 [wandb_run.py:_redirect():2373] redirect: wrap_raw
17
+ 2026-07-13 18:18:51,909 INFO MainThread:31979 [wandb_run.py:_redirect():2442] Wrapping output streams.
18
+ 2026-07-13 18:18:51,910 INFO MainThread:31979 [wandb_run.py:_redirect():2465] Redirects installed.
19
+ 2026-07-13 18:18:51,923 INFO MainThread:31979 [wandb_init.py:init():1012] run started, returning control to user process
20
+ 2026-07-14 02:36:38,976 INFO MainThread:31979 [wandb_run.py:_finish():2285] finishing run kasloot-ai/robopref/o7ouu23c
21
+ 2026-07-14 02:36:38,977 INFO MainThread:31979 [wandb_run.py:_atexit_cleanup():2490] got exitcode: 0
22
+ 2026-07-14 02:36:38,978 INFO MainThread:31979 [wandb_run.py:_restore():2472] restore
23
+ 2026-07-14 02:36:38,978 INFO MainThread:31979 [wandb_run.py:_restore():2478] restore done
24
+ 2026-07-14 02:36:40,963 INFO MainThread:31979 [wandb_run.py:_footer_sync_info():3895] logging synced files
wandb/run-20260713_181849-o7ouu23c/run-o7ouu23c.wandb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c2068909f964a0ff64fa5850e4f60405d5116fd40f9e9f90c160a67e0ff88067
3
+ size 20059456