Biamterdex commited on
Commit
5e79ff5
·
verified ·
1 Parent(s): bd8aa39

Upload folder using huggingface_hub

Browse files
checkpoints/001000/pretrained_model/config.json ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "act",
3
+ "n_obs_steps": 1,
4
+ "input_features": {
5
+ "observation.state": {
6
+ "type": "STATE",
7
+ "shape": [
8
+ 9
9
+ ]
10
+ },
11
+ "observation.images.front": {
12
+ "type": "VISUAL",
13
+ "shape": [
14
+ 3,
15
+ 480,
16
+ 640
17
+ ]
18
+ },
19
+ "observation.images.wrist": {
20
+ "type": "VISUAL",
21
+ "shape": [
22
+ 3,
23
+ 640,
24
+ 480
25
+ ]
26
+ },
27
+ "observation.images.above": {
28
+ "type": "VISUAL",
29
+ "shape": [
30
+ 3,
31
+ 640,
32
+ 480
33
+ ]
34
+ }
35
+ },
36
+ "output_features": {
37
+ "action": {
38
+ "type": "ACTION",
39
+ "shape": [
40
+ 9
41
+ ]
42
+ }
43
+ },
44
+ "device": "cuda",
45
+ "use_amp": false,
46
+ "push_to_hub": false,
47
+ "repo_id": null,
48
+ "private": null,
49
+ "tags": null,
50
+ "license": null,
51
+ "pretrained_path": null,
52
+ "chunk_size": 100,
53
+ "n_action_steps": 100,
54
+ "normalization_mapping": {
55
+ "VISUAL": "MEAN_STD",
56
+ "STATE": "MEAN_STD",
57
+ "ACTION": "MEAN_STD"
58
+ },
59
+ "vision_backbone": "resnet18",
60
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
61
+ "replace_final_stride_with_dilation": false,
62
+ "pre_norm": false,
63
+ "dim_model": 512,
64
+ "n_heads": 8,
65
+ "dim_feedforward": 3200,
66
+ "feedforward_activation": "relu",
67
+ "n_encoder_layers": 4,
68
+ "n_decoder_layers": 1,
69
+ "use_vae": true,
70
+ "latent_dim": 32,
71
+ "n_vae_encoder_layers": 4,
72
+ "temporal_ensemble_coeff": null,
73
+ "dropout": 0.1,
74
+ "kl_weight": 10.0,
75
+ "optimizer_lr": 1e-05,
76
+ "optimizer_weight_decay": 0.0001,
77
+ "optimizer_lr_backbone": 1e-05
78
+ }
checkpoints/001000/pretrained_model/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a4b0960199c11cddb461d96b38686a6536497134559653d3a30e5e6342c98d7f
3
+ size 206724324
checkpoints/001000/pretrained_model/policy_postprocessor.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_postprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "unnormalizer_processor",
6
+ "config": {
7
+ "eps": 1e-08,
8
+ "features": {
9
+ "action": {
10
+ "type": "ACTION",
11
+ "shape": [
12
+ 9
13
+ ]
14
+ }
15
+ },
16
+ "norm_map": {
17
+ "VISUAL": "MEAN_STD",
18
+ "STATE": "MEAN_STD",
19
+ "ACTION": "MEAN_STD"
20
+ }
21
+ },
22
+ "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
23
+ },
24
+ {
25
+ "registry_name": "device_processor",
26
+ "config": {
27
+ "device": "cpu",
28
+ "float_dtype": null
29
+ }
30
+ }
31
+ ]
32
+ }
checkpoints/001000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc2488c913552bc0dd2d9d605f2375c12838e54b5dd5be6251f99dde3e9e6fd7
3
+ size 8808
checkpoints/001000/pretrained_model/policy_preprocessor.json ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_preprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "rename_observations_processor",
6
+ "config": {
7
+ "rename_map": {}
8
+ }
9
+ },
10
+ {
11
+ "registry_name": "to_batch_processor",
12
+ "config": {}
13
+ },
14
+ {
15
+ "registry_name": "device_processor",
16
+ "config": {
17
+ "device": "cuda",
18
+ "float_dtype": null
19
+ }
20
+ },
21
+ {
22
+ "registry_name": "normalizer_processor",
23
+ "config": {
24
+ "eps": 1e-08,
25
+ "features": {
26
+ "observation.state": {
27
+ "type": "STATE",
28
+ "shape": [
29
+ 9
30
+ ]
31
+ },
32
+ "observation.images.front": {
33
+ "type": "VISUAL",
34
+ "shape": [
35
+ 3,
36
+ 480,
37
+ 640
38
+ ]
39
+ },
40
+ "observation.images.wrist": {
41
+ "type": "VISUAL",
42
+ "shape": [
43
+ 3,
44
+ 640,
45
+ 480
46
+ ]
47
+ },
48
+ "observation.images.above": {
49
+ "type": "VISUAL",
50
+ "shape": [
51
+ 3,
52
+ 640,
53
+ 480
54
+ ]
55
+ },
56
+ "action": {
57
+ "type": "ACTION",
58
+ "shape": [
59
+ 9
60
+ ]
61
+ }
62
+ },
63
+ "norm_map": {
64
+ "VISUAL": "MEAN_STD",
65
+ "STATE": "MEAN_STD",
66
+ "ACTION": "MEAN_STD"
67
+ }
68
+ },
69
+ "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
70
+ }
71
+ ]
72
+ }
checkpoints/001000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc2488c913552bc0dd2d9d605f2375c12838e54b5dd5be6251f99dde3e9e6fd7
3
+ size 8808
checkpoints/001000/pretrained_model/train_config.json ADDED
@@ -0,0 +1,201 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": {
3
+ "repo_id": "Biamterdex/amd_hackathon_mobile_arm-02",
4
+ "root": null,
5
+ "episodes": null,
6
+ "image_transforms": {
7
+ "enable": false,
8
+ "max_num_transforms": 3,
9
+ "random_order": false,
10
+ "tfs": {
11
+ "brightness": {
12
+ "weight": 1.0,
13
+ "type": "ColorJitter",
14
+ "kwargs": {
15
+ "brightness": [
16
+ 0.8,
17
+ 1.2
18
+ ]
19
+ }
20
+ },
21
+ "contrast": {
22
+ "weight": 1.0,
23
+ "type": "ColorJitter",
24
+ "kwargs": {
25
+ "contrast": [
26
+ 0.8,
27
+ 1.2
28
+ ]
29
+ }
30
+ },
31
+ "saturation": {
32
+ "weight": 1.0,
33
+ "type": "ColorJitter",
34
+ "kwargs": {
35
+ "saturation": [
36
+ 0.5,
37
+ 1.5
38
+ ]
39
+ }
40
+ },
41
+ "hue": {
42
+ "weight": 1.0,
43
+ "type": "ColorJitter",
44
+ "kwargs": {
45
+ "hue": [
46
+ -0.05,
47
+ 0.05
48
+ ]
49
+ }
50
+ },
51
+ "sharpness": {
52
+ "weight": 1.0,
53
+ "type": "SharpnessJitter",
54
+ "kwargs": {
55
+ "sharpness": [
56
+ 0.5,
57
+ 1.5
58
+ ]
59
+ }
60
+ },
61
+ "affine": {
62
+ "weight": 1.0,
63
+ "type": "RandomAffine",
64
+ "kwargs": {
65
+ "degrees": [
66
+ -5.0,
67
+ 5.0
68
+ ],
69
+ "translate": [
70
+ 0.05,
71
+ 0.05
72
+ ]
73
+ }
74
+ }
75
+ }
76
+ },
77
+ "revision": null,
78
+ "use_imagenet_stats": true,
79
+ "video_backend": "torchcodec",
80
+ "streaming": false
81
+ },
82
+ "env": null,
83
+ "policy": {
84
+ "type": "act",
85
+ "n_obs_steps": 1,
86
+ "input_features": {
87
+ "observation.state": {
88
+ "type": "STATE",
89
+ "shape": [
90
+ 9
91
+ ]
92
+ },
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
+ 640,
106
+ 480
107
+ ]
108
+ },
109
+ "observation.images.above": {
110
+ "type": "VISUAL",
111
+ "shape": [
112
+ 3,
113
+ 640,
114
+ 480
115
+ ]
116
+ }
117
+ },
118
+ "output_features": {
119
+ "action": {
120
+ "type": "ACTION",
121
+ "shape": [
122
+ 9
123
+ ]
124
+ }
125
+ },
126
+ "device": "cuda",
127
+ "use_amp": false,
128
+ "push_to_hub": false,
129
+ "repo_id": null,
130
+ "private": null,
131
+ "tags": null,
132
+ "license": null,
133
+ "pretrained_path": null,
134
+ "chunk_size": 100,
135
+ "n_action_steps": 100,
136
+ "normalization_mapping": {
137
+ "VISUAL": "MEAN_STD",
138
+ "STATE": "MEAN_STD",
139
+ "ACTION": "MEAN_STD"
140
+ },
141
+ "vision_backbone": "resnet18",
142
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
143
+ "replace_final_stride_with_dilation": false,
144
+ "pre_norm": false,
145
+ "dim_model": 512,
146
+ "n_heads": 8,
147
+ "dim_feedforward": 3200,
148
+ "feedforward_activation": "relu",
149
+ "n_encoder_layers": 4,
150
+ "n_decoder_layers": 1,
151
+ "use_vae": true,
152
+ "latent_dim": 32,
153
+ "n_vae_encoder_layers": 4,
154
+ "temporal_ensemble_coeff": null,
155
+ "dropout": 0.1,
156
+ "kl_weight": 10.0,
157
+ "optimizer_lr": 1e-05,
158
+ "optimizer_weight_decay": 0.0001,
159
+ "optimizer_lr_backbone": 1e-05
160
+ },
161
+ "output_dir": "outputs/train/act_mobile_arm_bottle_02",
162
+ "job_name": "act_mobile_arm_bottle_02",
163
+ "resume": false,
164
+ "seed": 1000,
165
+ "num_workers": 4,
166
+ "batch_size": 64,
167
+ "steps": 1000,
168
+ "eval_freq": 20000,
169
+ "log_freq": 200,
170
+ "save_checkpoint": true,
171
+ "save_freq": 20000,
172
+ "use_policy_training_preset": true,
173
+ "optimizer": {
174
+ "type": "adamw",
175
+ "lr": 1e-05,
176
+ "weight_decay": 0.0001,
177
+ "grad_clip_norm": 10.0,
178
+ "betas": [
179
+ 0.9,
180
+ 0.999
181
+ ],
182
+ "eps": 1e-08
183
+ },
184
+ "scheduler": null,
185
+ "eval": {
186
+ "n_episodes": 50,
187
+ "batch_size": 50,
188
+ "use_async_envs": false
189
+ },
190
+ "wandb": {
191
+ "enable": true,
192
+ "disable_artifact": false,
193
+ "project": "lerobot",
194
+ "entity": null,
195
+ "notes": null,
196
+ "run_id": "c7ladxh7",
197
+ "mode": null
198
+ },
199
+ "checkpoint_path": null,
200
+ "rename_map": {}
201
+ }
checkpoints/001000/training_state/optimizer_param_groups.json ADDED
@@ -0,0 +1,191 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "lr": 1e-05,
4
+ "betas": [
5
+ 0.9,
6
+ 0.999
7
+ ],
8
+ "eps": 1e-08,
9
+ "weight_decay": 0.0001,
10
+ "amsgrad": false,
11
+ "maximize": false,
12
+ "foreach": null,
13
+ "capturable": false,
14
+ "differentiable": false,
15
+ "fused": null,
16
+ "decoupled_weight_decay": true,
17
+ "params": [
18
+ 0,
19
+ 1,
20
+ 2,
21
+ 3,
22
+ 4,
23
+ 5,
24
+ 6,
25
+ 7,
26
+ 8,
27
+ 9,
28
+ 10,
29
+ 11,
30
+ 12,
31
+ 13,
32
+ 14,
33
+ 15,
34
+ 16,
35
+ 17,
36
+ 18,
37
+ 19,
38
+ 20,
39
+ 21,
40
+ 22,
41
+ 23,
42
+ 24,
43
+ 25,
44
+ 26,
45
+ 27,
46
+ 28,
47
+ 29,
48
+ 30,
49
+ 31,
50
+ 32,
51
+ 33,
52
+ 34,
53
+ 35,
54
+ 36,
55
+ 37,
56
+ 38,
57
+ 39,
58
+ 40,
59
+ 41,
60
+ 42,
61
+ 43,
62
+ 44,
63
+ 45,
64
+ 46,
65
+ 47,
66
+ 48,
67
+ 49,
68
+ 50,
69
+ 51,
70
+ 52,
71
+ 53,
72
+ 54,
73
+ 55,
74
+ 56,
75
+ 57,
76
+ 58,
77
+ 59,
78
+ 60,
79
+ 61,
80
+ 62,
81
+ 63,
82
+ 64,
83
+ 65,
84
+ 66,
85
+ 67,
86
+ 68,
87
+ 69,
88
+ 70,
89
+ 71,
90
+ 72,
91
+ 73,
92
+ 74,
93
+ 75,
94
+ 76,
95
+ 77,
96
+ 78,
97
+ 79,
98
+ 80,
99
+ 81,
100
+ 82,
101
+ 83,
102
+ 84,
103
+ 85,
104
+ 86,
105
+ 87,
106
+ 88,
107
+ 89,
108
+ 90,
109
+ 91,
110
+ 92,
111
+ 93,
112
+ 94,
113
+ 95,
114
+ 96,
115
+ 97,
116
+ 98,
117
+ 99,
118
+ 100,
119
+ 101,
120
+ 102,
121
+ 103,
122
+ 104,
123
+ 105,
124
+ 106,
125
+ 107,
126
+ 108,
127
+ 109,
128
+ 110,
129
+ 111,
130
+ 112,
131
+ 113,
132
+ 114,
133
+ 115,
134
+ 116,
135
+ 117,
136
+ 118,
137
+ 119,
138
+ 120,
139
+ 121,
140
+ 122,
141
+ 123,
142
+ 124,
143
+ 125,
144
+ 126,
145
+ 127,
146
+ 128,
147
+ 129,
148
+ 130,
149
+ 131,
150
+ 132
151
+ ]
152
+ },
153
+ {
154
+ "lr": 1e-05,
155
+ "betas": [
156
+ 0.9,
157
+ 0.999
158
+ ],
159
+ "eps": 1e-08,
160
+ "weight_decay": 0.0001,
161
+ "amsgrad": false,
162
+ "maximize": false,
163
+ "foreach": null,
164
+ "capturable": false,
165
+ "differentiable": false,
166
+ "fused": null,
167
+ "decoupled_weight_decay": true,
168
+ "params": [
169
+ 133,
170
+ 134,
171
+ 135,
172
+ 136,
173
+ 137,
174
+ 138,
175
+ 139,
176
+ 140,
177
+ 141,
178
+ 142,
179
+ 143,
180
+ 144,
181
+ 145,
182
+ 146,
183
+ 147,
184
+ 148,
185
+ 149,
186
+ 150,
187
+ 151,
188
+ 152
189
+ ]
190
+ }
191
+ ]
checkpoints/001000/training_state/optimizer_state.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d13c56e2baf1bf44696ed5fa26955bb72471bfae67d7184cb3b3b9b9e622a1e9
3
+ size 412866828
checkpoints/001000/training_state/rng_state.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:47cc1979a17ded11ef6e8eb660e84965e110a79d7641448189f33c215986580e
3
+ size 15708
checkpoints/001000/training_state/training_step.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "step": 1000
3
+ }
wandb/debug-internal.log ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2025-12-06T16:05:14.674167516Z","level":"INFO","msg":"stream: starting","core version":"0.21.4"}
2
+ {"time":"2025-12-06T16:05:14.937920435Z","level":"INFO","msg":"stream: created new stream","id":"c7ladxh7"}
3
+ {"time":"2025-12-06T16:05:14.937950636Z","level":"INFO","msg":"stream: started","id":"c7ladxh7"}
4
+ {"time":"2025-12-06T16:05:14.937992748Z","level":"INFO","msg":"writer: started","stream_id":"c7ladxh7"}
5
+ {"time":"2025-12-06T16:05:14.938002175Z","level":"INFO","msg":"handler: started","stream_id":"c7ladxh7"}
6
+ {"time":"2025-12-06T16:05:14.938035855Z","level":"INFO","msg":"sender: started","stream_id":"c7ladxh7"}
7
+ {"time":"2025-12-06T16:25:15.385384708Z","level":"INFO","msg":"stream: closing","id":"c7ladxh7"}
8
+ {"time":"2025-12-06T16:25:18.158973747Z","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
9
+ {"time":"2025-12-06T16:25:18.254188501Z","level":"INFO","msg":"handler: closed","stream_id":"c7ladxh7"}
10
+ {"time":"2025-12-06T16:25:18.254234232Z","level":"INFO","msg":"sender: closed","stream_id":"c7ladxh7"}
11
+ {"time":"2025-12-06T16:25:18.254240105Z","level":"INFO","msg":"stream: closed","id":"c7ladxh7"}
wandb/debug.log ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_setup.py:_flush():81] Current SDK version is 0.21.4
2
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_setup.py:_flush():81] Configure stats pid to 17679
3
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_setup.py:_flush():81] Loading settings from /root/.config/wandb/settings
4
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_setup.py:_flush():81] Loading settings from /workspace/wandb/settings
5
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_setup.py:_flush():81] Loading settings from environment variables
6
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_init.py:setup_run_log_directory():686] Logging user logs to outputs/train/act_mobile_arm_bottle_02/wandb/run-20251206_160514-c7ladxh7/logs/debug.log
7
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_init.py:setup_run_log_directory():687] Logging internal logs to outputs/train/act_mobile_arm_bottle_02/wandb/run-20251206_160514-c7ladxh7/logs/debug-internal.log
8
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_init.py:init():813] calling init triggers
9
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_init.py:init():818] wandb.init called with sweep_config: {}
10
+ config: {'dataset': {'repo_id': 'Biamterdex/amd_hackathon_mobile_arm-02', 'root': None, 'episodes': None, 'image_transforms': {'enable': False, 'max_num_transforms': 3, 'random_order': False, 'tfs': {'brightness': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'brightness': [0.8, 1.2]}}, 'contrast': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'contrast': [0.8, 1.2]}}, 'saturation': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'saturation': [0.5, 1.5]}}, 'hue': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'hue': [-0.05, 0.05]}}, 'sharpness': {'weight': 1.0, 'type': 'SharpnessJitter', 'kwargs': {'sharpness': [0.5, 1.5]}}, 'affine': {'weight': 1.0, 'type': 'RandomAffine', 'kwargs': {'degrees': [-5.0, 5.0], 'translate': [0.05, 0.05]}}}}, 'revision': None, 'use_imagenet_stats': True, 'video_backend': 'torchcodec', 'streaming': False}, 'env': None, 'policy': {'type': 'act', 'n_obs_steps': 1, 'input_features': {}, 'output_features': {}, 'device': 'cuda', 'use_amp': False, 'push_to_hub': False, 'repo_id': None, 'private': None, 'tags': None, 'license': None, 'pretrained_path': None, 'chunk_size': 100, 'n_action_steps': 100, 'normalization_mapping': {'VISUAL': <NormalizationMode.MEAN_STD: 'MEAN_STD'>, 'STATE': <NormalizationMode.MEAN_STD: 'MEAN_STD'>, 'ACTION': <NormalizationMode.MEAN_STD: 'MEAN_STD'>}, 'vision_backbone': 'resnet18', 'pretrained_backbone_weights': 'ResNet18_Weights.IMAGENET1K_V1', 'replace_final_stride_with_dilation': False, 'pre_norm': False, 'dim_model': 512, 'n_heads': 8, 'dim_feedforward': 3200, 'feedforward_activation': 'relu', 'n_encoder_layers': 4, 'n_decoder_layers': 1, 'use_vae': True, 'latent_dim': 32, 'n_vae_encoder_layers': 4, 'temporal_ensemble_coeff': None, 'dropout': 0.1, 'kl_weight': 10.0, 'optimizer_lr': 1e-05, 'optimizer_weight_decay': 0.0001, 'optimizer_lr_backbone': 1e-05}, 'output_dir': 'outputs/train/act_mobile_arm_bottle_02', 'job_name': 'act_mobile_arm_bottle_02', 'resume': False, 'seed': 1000, 'num_workers': 4, 'batch_size': 64, 'steps': 1000, 'eval_freq': 20000, 'log_freq': 200, 'save_checkpoint': True, 'save_freq': 20000, 'use_policy_training_preset': True, 'optimizer': {'type': 'adamw', 'lr': 1e-05, 'weight_decay': 0.0001, 'grad_clip_norm': 10.0, 'betas': [0.9, 0.999], 'eps': 1e-08}, 'scheduler': None, 'eval': {'n_episodes': 50, 'batch_size': 50, 'use_async_envs': False}, 'wandb': {'enable': True, 'disable_artifact': False, 'project': 'lerobot', 'entity': None, 'notes': None, 'run_id': None, 'mode': None}, 'checkpoint_path': None, 'rename_map': {}, '_wandb': {}}
11
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_init.py:init():854] starting backend
12
+ 2025-12-06 16:05:14,670 INFO MainThread:17679 [wandb_init.py:init():857] sending inform_init request
13
+ 2025-12-06 16:05:14,672 INFO MainThread:17679 [wandb_init.py:init():865] backend started and connected
14
+ 2025-12-06 16:05:14,674 INFO MainThread:17679 [wandb_init.py:init():936] updated telemetry
15
+ 2025-12-06 16:05:14,674 INFO MainThread:17679 [wandb_init.py:init():960] communicating run to backend with 90.0 second timeout
16
+ 2025-12-06 16:05:15,330 INFO MainThread:17679 [wandb_init.py:init():1011] starting run threads in backend
17
+ 2025-12-06 16:05:15,401 INFO MainThread:17679 [wandb_run.py:_console_start():2506] atexit reg
18
+ 2025-12-06 16:05:15,401 INFO MainThread:17679 [wandb_run.py:_redirect():2354] redirect: wrap_raw
19
+ 2025-12-06 16:05:15,401 INFO MainThread:17679 [wandb_run.py:_redirect():2423] Wrapping output streams.
20
+ 2025-12-06 16:05:15,401 INFO MainThread:17679 [wandb_run.py:_redirect():2446] Redirects installed.
21
+ 2025-12-06 16:05:15,403 INFO MainThread:17679 [wandb_init.py:init():1049] run started, returning control to user process
22
+ 2025-12-06 16:25:15,385 INFO wandb-AsyncioManager-main:17679 [service_client.py:_forward_responses():84] Reached EOF.
23
+ 2025-12-06 16:25:15,385 INFO wandb-AsyncioManager-main:17679 [mailbox.py:close():137] Closing mailbox, abandoning 2 handles.
wandb/run-20251206_160514-c7ladxh7/files/config.yaml ADDED
@@ -0,0 +1,223 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _wandb:
2
+ value:
3
+ cli_version: 0.21.4
4
+ e:
5
+ pth0ojgsvz6a9xak7fq3hjd9jtmlip5t:
6
+ args:
7
+ - --dataset.repo_id=Biamterdex/amd_hackathon_mobile_arm-02
8
+ - --batch_size=64
9
+ - --steps=1000
10
+ - --output_dir=outputs/train/act_mobile_arm_bottle_02
11
+ - --job_name=act_mobile_arm_bottle_02
12
+ - --policy.device=cuda
13
+ - --policy.type=act
14
+ - --policy.push_to_hub=false
15
+ - --wandb.enable=true
16
+ cpu_count: 160
17
+ cpu_count_logical: 160
18
+ disk:
19
+ /:
20
+ total: "2162185003008"
21
+ used: "836658180096"
22
+ executable: /opt/venv/bin/python
23
+ gpu: AMD Instinct MI300X VF
24
+ gpu_amd:
25
+ - id: "0"
26
+ maxPower: "750.0"
27
+ mclkRange: 900Mhz - 1300Mhz
28
+ model: "0x74b5"
29
+ performanceLevel: auto
30
+ sclkRange: 500Mhz - 2100Mhz
31
+ series: AMD Instinct MI300X VF
32
+ sku: M3000100
33
+ uniqueId: "0x29d54fa8a413a984"
34
+ vbiosVersion: 113-M3000100-103
35
+ vendor: Advanced Micro Devices, Inc. [AMD/ATI]
36
+ gpu_count: 1
37
+ host: robotics-jupyter-launcher-5866
38
+ memory:
39
+ total: "2029207175168"
40
+ os: Linux-6.1.0-39-amd64-x86_64-with-glibc2.39
41
+ program: /opt/venv/bin/lerobot-train
42
+ python: CPython 3.12.3
43
+ root: outputs/train/act_mobile_arm_bottle_02
44
+ startedAt: "2025-12-06T16:05:14.464991Z"
45
+ writerId: pth0ojgsvz6a9xak7fq3hjd9jtmlip5t
46
+ m: []
47
+ python_version: 3.12.3
48
+ t:
49
+ "1":
50
+ - 1
51
+ - 41
52
+ - 49
53
+ - 51
54
+ - 71
55
+ - 83
56
+ "2":
57
+ - 1
58
+ - 41
59
+ - 49
60
+ - 51
61
+ - 71
62
+ - 83
63
+ "3":
64
+ - 13
65
+ - 15
66
+ - 16
67
+ - 61
68
+ "4": 3.12.3
69
+ "5": 0.21.4
70
+ "10":
71
+ - 21
72
+ "12": 0.21.4
73
+ "13": linux-x86_64
74
+ batch_size:
75
+ value: 64
76
+ checkpoint_path:
77
+ value: null
78
+ dataset:
79
+ value:
80
+ episodes: null
81
+ image_transforms:
82
+ enable: false
83
+ max_num_transforms: 3
84
+ random_order: false
85
+ tfs:
86
+ affine:
87
+ kwargs:
88
+ degrees:
89
+ - -5
90
+ - 5
91
+ translate:
92
+ - 0.05
93
+ - 0.05
94
+ type: RandomAffine
95
+ weight: 1
96
+ brightness:
97
+ kwargs:
98
+ brightness:
99
+ - 0.8
100
+ - 1.2
101
+ type: ColorJitter
102
+ weight: 1
103
+ contrast:
104
+ kwargs:
105
+ contrast:
106
+ - 0.8
107
+ - 1.2
108
+ type: ColorJitter
109
+ weight: 1
110
+ hue:
111
+ kwargs:
112
+ hue:
113
+ - -0.05
114
+ - 0.05
115
+ type: ColorJitter
116
+ weight: 1
117
+ saturation:
118
+ kwargs:
119
+ saturation:
120
+ - 0.5
121
+ - 1.5
122
+ type: ColorJitter
123
+ weight: 1
124
+ sharpness:
125
+ kwargs:
126
+ sharpness:
127
+ - 0.5
128
+ - 1.5
129
+ type: SharpnessJitter
130
+ weight: 1
131
+ repo_id: Biamterdex/amd_hackathon_mobile_arm-02
132
+ revision: null
133
+ root: null
134
+ streaming: false
135
+ use_imagenet_stats: true
136
+ video_backend: torchcodec
137
+ env:
138
+ value: null
139
+ eval:
140
+ value:
141
+ batch_size: 50
142
+ n_episodes: 50
143
+ use_async_envs: false
144
+ eval_freq:
145
+ value: 20000
146
+ job_name:
147
+ value: act_mobile_arm_bottle_02
148
+ log_freq:
149
+ value: 200
150
+ num_workers:
151
+ value: 4
152
+ optimizer:
153
+ value:
154
+ betas:
155
+ - 0.9
156
+ - 0.999
157
+ eps: 1e-08
158
+ grad_clip_norm: 10
159
+ lr: 1e-05
160
+ type: adamw
161
+ weight_decay: 0.0001
162
+ output_dir:
163
+ value: outputs/train/act_mobile_arm_bottle_02
164
+ policy:
165
+ value:
166
+ chunk_size: 100
167
+ device: cuda
168
+ dim_feedforward: 3200
169
+ dim_model: 512
170
+ dropout: 0.1
171
+ feedforward_activation: relu
172
+ kl_weight: 10
173
+ latent_dim: 32
174
+ license: null
175
+ n_action_steps: 100
176
+ n_decoder_layers: 1
177
+ n_encoder_layers: 4
178
+ n_heads: 8
179
+ n_obs_steps: 1
180
+ n_vae_encoder_layers: 4
181
+ normalization_mapping:
182
+ ACTION: MEAN_STD
183
+ STATE: MEAN_STD
184
+ VISUAL: MEAN_STD
185
+ optimizer_lr: 1e-05
186
+ optimizer_lr_backbone: 1e-05
187
+ optimizer_weight_decay: 0.0001
188
+ pre_norm: false
189
+ pretrained_backbone_weights: ResNet18_Weights.IMAGENET1K_V1
190
+ pretrained_path: null
191
+ private: null
192
+ push_to_hub: false
193
+ replace_final_stride_with_dilation: false
194
+ repo_id: null
195
+ tags: null
196
+ temporal_ensemble_coeff: null
197
+ type: act
198
+ use_amp: false
199
+ use_vae: true
200
+ vision_backbone: resnet18
201
+ resume:
202
+ value: false
203
+ save_checkpoint:
204
+ value: true
205
+ save_freq:
206
+ value: 20000
207
+ scheduler:
208
+ value: null
209
+ seed:
210
+ value: 1000
211
+ steps:
212
+ value: 1000
213
+ use_policy_training_preset:
214
+ value: true
215
+ wandb:
216
+ value:
217
+ disable_artifact: false
218
+ enable: true
219
+ entity: null
220
+ mode: null
221
+ notes: null
222
+ project: lerobot
223
+ run_id: null
wandb/run-20251206_160514-c7ladxh7/files/output.log ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ INFO 2025-12-06 16:05:15 db_utils.py:102 Logs will be synced with wandb.
2
+ INFO 2025-12-06 16:05:15 db_utils.py:103 Track this run --> https://wandb.ai/mohamedkarimziadi-hugging-face/lerobot/runs/c7ladxh7
3
+ INFO 2025-12-06 16:05:15 ot_train.py:183 Creating dataset
4
+ stats.json: 17.8kB [00:00, 39.3MB/s] | 0/4 [00:00<?, ?it/s]
5
+ info.json: 4.07kB [00:00, 15.3MB/s]rquet: 0%| | 0.00/117k [00:00<?, ?B/s]
6
+ meta/tasks.parquet: 100%|██████████████████| 2.19k/2.19k [00:00<00:00, 5.88kB/s]
7
+ meta/episodes/chunk-000/file-000.parquet: 100%|█| 117k/117k [00:00<00:00, 244kB/
8
+ Fetching 4 files: 100%|███████████████████████████| 4/4 [00:00<00:00, 6.65it/s]
9
+ .gitattributes: 2.46kB [00:00, 8.77MB/s] | 0/10 [00:00<?, ?it/s]
10
+ README.md: 4.56kB [00:00, 16.6MB/s] | 1/10 [00:00<00:02, 4.18it/s]
11
+ data/chunk-000/file-000.parquet: 100%|███████| 358k/358k [00:00<00:00, 2.96MB/s]
12
+ videos/observation.images.front/chunk-00(…): 100%|█| 246M/246M [00:01<00:00, 172
13
+ videos/observation.images.wrist/chunk-00(…): 100%|█| 233M/233M [00:01<00:00, 155
14
+ videos/observation.images.above/chunk-00(…): 100%|█| 245M/245M [00:01<00:00, 157
15
+ Fetching 10 files: 100%|████████████████████████| 10/10 [00:01<00:00, 5.65it/s]
16
+ INFO 2025-12-06 16:05:18 ot_train.py:202 Creating policy3M/233M [00:01<00:01, 86
17
+ INFO 2025-12-06 16:05:19 ot_train.py:247 Creating optimizer and scheduler00, 160
18
+ INFO 2025-12-06 16:05:19 ot_train.py:259 Output dir: outputs/train/act_mobile_arm_bottle_02
19
+ INFO 2025-12-06 16:05:19 ot_train.py:262 cfg.steps=1000 (1K)
20
+ INFO 2025-12-06 16:05:19 ot_train.py:263 dataset.num_frames=9898 (10K)
21
+ INFO 2025-12-06 16:05:19 ot_train.py:264 dataset.num_episodes=13
22
+ INFO 2025-12-06 16:05:19 ot_train.py:267 Effective batch size: 64 x 1 = 64
23
+ INFO 2025-12-06 16:05:19 ot_train.py:268 num_learnable_params=51603337 (52M)
24
+ INFO 2025-12-06 16:05:19 ot_train.py:269 num_total_params=51603337 (52M)
25
+ INFO 2025-12-06 16:05:19 ot_train.py:324 Start offline training on a fixed dataset
26
+ INFO 2025-12-06 16:19:24 ot_train.py:351 step:200 smpl:13K ep:17 epch:1.29 loss:5.607 grdn:78.574 lr:1.0e-05 updt_s:4.165 data_s:0.062
27
+ INFO 2025-12-06 16:20:49 ot_train.py:351 step:400 smpl:26K ep:34 epch:2.59 loss:2.114 grdn:36.124 lr:1.0e-05 updt_s:0.384 data_s:0.042
28
+ INFO 2025-12-06 16:22:14 ot_train.py:351 step:600 smpl:38K ep:50 epch:3.88 loss:1.653 grdn:31.467 lr:1.0e-05 updt_s:0.384 data_s:0.040
29
+ INFO 2025-12-06 16:23:44 ot_train.py:351 step:800 smpl:51K ep:67 epch:5.17 loss:1.320 grdn:28.911 lr:1.0e-05 updt_s:0.382 data_s:0.066
30
+ INFO 2025-12-06 16:25:13 ot_train.py:351 step:1K smpl:64K ep:84 epch:6.47 loss:1.064 grdn:26.336 lr:1.0e-05 updt_s:0.381 data_s:0.064
31
+ INFO 2025-12-06 16:25:13 ot_train.py:361 Checkpoint policy after step 1000
32
+ INFO 2025-12-06 16:25:14 ot_train.py:430 End of training
wandb/run-20251206_160514-c7ladxh7/files/requirements.txt ADDED
@@ -0,0 +1,185 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ pip==25.2
2
+ setuptools==80.9.0
3
+ zipp==3.23.0
4
+ jupyter_core==5.9.1
5
+ rpds-py==0.30.0
6
+ terminado==0.18.1
7
+ ihighlight==1.0.0
8
+ rfc3986-validator==0.1.1
9
+ pure_eval==0.2.3
10
+ ImageIO==2.37.2
11
+ dill==0.4.0
12
+ prompt_toolkit==3.0.52
13
+ httpx==0.28.1
14
+ cmake==4.1.3
15
+ jupyter-console==6.6.3
16
+ jsonschema==4.25.1
17
+ torchcodec==0.5
18
+ nbclient==0.10.2
19
+ h11==0.16.0
20
+ mistune==3.1.4
21
+ pydantic==2.12.5
22
+ jupyter_server_terminals==0.5.3
23
+ arrow==1.4.0
24
+ av==15.1.0
25
+ python-xlib==0.33
26
+ mergedeep==1.3.4
27
+ ipywidgets==8.1.8
28
+ sentry-sdk==2.47.0
29
+ matplotlib-inline==0.2.1
30
+ jupyterlab_pygments==0.3.0
31
+ toml==0.10.2
32
+ multidict==6.7.0
33
+ jupyter-lsp==2.3.0
34
+ imageio-ffmpeg==0.6.0
35
+ babel==2.17.0
36
+ typing-inspection==0.4.2
37
+ jsonschema-specifications==2025.9.1
38
+ isoduration==20.11.0
39
+ aiosignal==1.4.0
40
+ rfc3987-syntax==1.1.0
41
+ psutil==7.1.3
42
+ einops==0.8.1
43
+ argon2-cffi==25.1.0
44
+ aiohttp==3.13.2
45
+ pyyaml-include==1.4.1
46
+ nbformat==5.10.4
47
+ urllib3==2.5.0
48
+ multiprocess==0.70.16
49
+ inquirerpy==0.3.4
50
+ click==8.3.1
51
+ asttokens==3.0.1
52
+ draccus==0.10.0
53
+ ipython==9.8.0
54
+ pexpect==4.9.0
55
+ deepdiff==8.6.1
56
+ xxhash==3.6.0
57
+ rerun-sdk==0.26.2
58
+ pandocfilters==1.5.1
59
+ jedi==0.19.2
60
+ mypy_extensions==1.1.0
61
+ orderly-set==5.5.0
62
+ async-lru==2.0.5
63
+ fqdn==1.5.1
64
+ fastjsonschema==2.21.2
65
+ uri-template==1.3.0
66
+ GitPython==3.1.45
67
+ aiohappyeyeballs==2.6.1
68
+ jupyterlab==4.5.0
69
+ evdev==1.9.2
70
+ anyio==4.12.0
71
+ requests==2.32.5
72
+ argon2-cffi-bindings==25.1.0
73
+ typing-inspect==0.9.0
74
+ charset-normalizer==3.4.4
75
+ webcolors==25.10.0
76
+ pyzmq==27.1.0
77
+ comm==0.2.3
78
+ notebook==7.5.0
79
+ defusedxml==0.7.1
80
+ jupyter-events==0.12.0
81
+ cloudpickle==3.1.2
82
+ propcache==0.4.1
83
+ jupyter_client==8.6.3
84
+ ptyprocess==0.7.0
85
+ widgetsnbextension==4.0.15
86
+ importlib_metadata==8.7.0
87
+ frozenlist==1.8.0
88
+ websocket-client==1.9.0
89
+ pydantic_core==2.41.5
90
+ jupyterlab_server==2.28.0
91
+ json5==0.12.1
92
+ certifi==2025.11.12
93
+ ipykernel==7.1.0
94
+ nbconvert==7.16.6
95
+ decorator==5.2.1
96
+ gymnasium==1.2.2
97
+ jupyterlab_widgets==3.0.16
98
+ notebook_shim==0.2.4
99
+ accelerate==1.12.0
100
+ pyarrow==22.0.0
101
+ regex==2025.11.3
102
+ parso==0.8.5
103
+ tinycss2==1.4.0
104
+ jupyter_server==2.17.0
105
+ debugpy==1.8.17
106
+ lark==1.3.1
107
+ annotated-types==0.7.0
108
+ idna==3.11
109
+ yarl==1.22.0
110
+ jupyter==1.1.1
111
+ diffusers==0.35.2
112
+ gitdb==4.0.12
113
+ termcolor==3.2.0
114
+ bleach==6.3.0
115
+ protobuf==6.33.2
116
+ beautifulsoup4==4.14.3
117
+ attrs==25.4.0
118
+ pyserial==3.5
119
+ tornado==6.5.2
120
+ opencv-python-headless==4.12.0.88
121
+ rfc3339-validator==0.1.4
122
+ referencing==0.37.0
123
+ python-json-logger==4.0.0
124
+ numpy==2.2.6
125
+ pfzy==0.3.4
126
+ executing==2.2.1
127
+ smmap==5.0.2
128
+ httpcore==1.0.9
129
+ pynput==1.8.1
130
+ nest-asyncio==1.6.0
131
+ hf-xet==1.2.0
132
+ cffi==2.0.0
133
+ ipython_pygments_lexers==1.1.1
134
+ stack-data==0.6.3
135
+ wcwidth==0.2.14
136
+ jsonpointer==3.0.0
137
+ prometheus_client==0.23.1
138
+ safetensors==0.7.0
139
+ datasets==4.1.1
140
+ platformdirs==4.5.0
141
+ soupsieve==2.8
142
+ jsonlines==4.0.0
143
+ webencodings==0.5.1
144
+ Send2Trash==1.8.3
145
+ traitlets==5.14.3
146
+ huggingface-hub==0.35.3
147
+ hf_transfer==0.1.9
148
+ lerobot==0.4.1
149
+ wandb==0.21.4
150
+ pycparser==2.23
151
+ Farama-Notifications==0.0.4
152
+ Jinja2==3.1.6
153
+ iniconfig==2.1.0
154
+ apex==1.7.0a0+rocm7.0.0.git20a62dc5
155
+ matplotlib==3.10.6
156
+ tzdata==2025.2
157
+ pytorch-triton-rocm==3.3.1+rocm7.0.0.git9c7bc0a3
158
+ pyparsing==3.2.4
159
+ Pygments==2.19.2
160
+ fsspec==2025.9.0
161
+ pytest==8.4.2
162
+ cycler==0.12.1
163
+ networkx==3.5
164
+ python-dateutil==2.9.0.post0
165
+ MarkupSafe==3.0.2
166
+ six==1.17.0
167
+ pluggy==1.6.0
168
+ sympy==1.14.0
169
+ torchaudio==2.7.1+rocm7.0.0.git95c61b41
170
+ fonttools==4.60.0
171
+ packaging==25.0
172
+ tqdm==4.67.1
173
+ PyYAML==6.0.2
174
+ pillow==11.3.0
175
+ filelock==3.19.1
176
+ cxxfilt==0.3.0
177
+ contourpy==1.3.3
178
+ pytz==2025.2
179
+ typing_extensions==4.15.0
180
+ kiwisolver==1.4.9
181
+ torchvision==0.22.1+rocm7.0.0.git59a3e1f9
182
+ pandas==2.3.2
183
+ torch==2.7.1+rocm7.0.0.lw.git698b58a9
184
+ mpmath==1.3.0
185
+ lerobot==0.4.1
wandb/run-20251206_160514-c7ladxh7/files/wandb-metadata.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-6.1.0-39-amd64-x86_64-with-glibc2.39",
3
+ "python": "CPython 3.12.3",
4
+ "startedAt": "2025-12-06T16:05:14.464991Z",
5
+ "args": [
6
+ "--dataset.repo_id=Biamterdex/amd_hackathon_mobile_arm-02",
7
+ "--batch_size=64",
8
+ "--steps=1000",
9
+ "--output_dir=outputs/train/act_mobile_arm_bottle_02",
10
+ "--job_name=act_mobile_arm_bottle_02",
11
+ "--policy.device=cuda",
12
+ "--policy.type=act",
13
+ "--policy.push_to_hub=false",
14
+ "--wandb.enable=true"
15
+ ],
16
+ "program": "/opt/venv/bin/lerobot-train",
17
+ "root": "outputs/train/act_mobile_arm_bottle_02",
18
+ "host": "robotics-jupyter-launcher-5866",
19
+ "executable": "/opt/venv/bin/python",
20
+ "cpu_count": 160,
21
+ "cpu_count_logical": 160,
22
+ "gpu": "AMD Instinct MI300X VF",
23
+ "gpu_count": 1,
24
+ "disk": {
25
+ "/": {
26
+ "total": "2162185003008",
27
+ "used": "836658180096"
28
+ }
29
+ },
30
+ "memory": {
31
+ "total": "2029207175168"
32
+ },
33
+ "gpu_amd": [
34
+ {
35
+ "id": "0",
36
+ "uniqueId": "0x29d54fa8a413a984",
37
+ "vbiosVersion": "113-M3000100-103",
38
+ "performanceLevel": "auto",
39
+ "maxPower": "750.0",
40
+ "series": "AMD Instinct MI300X VF",
41
+ "model": "0x74b5",
42
+ "vendor": "Advanced Micro Devices, Inc. [AMD/ATI]",
43
+ "sku": "M3000100",
44
+ "sclkRange": "500Mhz - 2100Mhz",
45
+ "mclkRange": "900Mhz - 1300Mhz"
46
+ }
47
+ ],
48
+ "writerId": "pth0ojgsvz6a9xak7fq3hjd9jtmlip5t"
49
+ }
wandb/run-20251206_160514-c7ladxh7/files/wandb-summary.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"train/loss":1.0638884899020196,"train/update_s":0.3812048326950753,"_step":1000,"_wandb":{"runtime":1200},"train/l1_loss":0.20209692418575287,"train/samples":64000,"train/steps":1000,"train/kld_loss":0.07243663817644119,"train/episodes":84.05738533036977,"train/grad_norm":26.336021852493285,"_timestamp":1.765038313431552e+09,"train/dataloading_s":0.06372922247333918,"train/lr":1.0000000000000021e-05,"train/epochs":6.465952717720752,"_runtime":1200.053943269}
wandb/run-20251206_160514-c7ladxh7/logs/debug-core.log ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2025-12-06T16:05:14.483468285Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmphl8uog1t/port-17679.txt","pid":17679,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
2
+ {"time":"2025-12-06T16:05:14.483884544Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":17679}
3
+ {"time":"2025-12-06T16:05:14.483878349Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-17679-17750-427570504/socket","Net":"unix"}}
4
+ {"time":"2025-12-06T16:05:14.6703318Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
5
+ {"time":"2025-12-06T16:05:14.674088297Z","level":"INFO","msg":"handleInformInit: received","streamId":"c7ladxh7","id":"1(@)"}
6
+ {"time":"2025-12-06T16:05:14.93795451Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"c7ladxh7","id":"1(@)"}
7
+ {"time":"2025-12-06T16:25:15.385338836Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
8
+ {"time":"2025-12-06T16:25:15.385424953Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
9
+ {"time":"2025-12-06T16:25:15.38545225Z","level":"INFO","msg":"server is shutting down"}
10
+ {"time":"2025-12-06T16:25:15.385483621Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
11
+ {"time":"2025-12-06T16:25:15.385524268Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-17679-17750-427570504/socket","Net":"unix"}}
12
+ {"time":"2025-12-06T16:25:18.108710089Z","level":"ERROR","msg":"processOutgoingData: flush error","error":"write unix /tmp/wandb-17679-17750-427570504/socket->@: use of closed network connection","id":"1(@)"}
13
+ {"time":"2025-12-06T16:25:18.255839251Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
14
+ {"time":"2025-12-06T16:25:18.255850795Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
15
+ {"time":"2025-12-06T16:25:18.255859826Z","level":"INFO","msg":"server is closed"}
wandb/run-20251206_160514-c7ladxh7/logs/debug-internal.log ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2025-12-06T16:05:14.674167516Z","level":"INFO","msg":"stream: starting","core version":"0.21.4"}
2
+ {"time":"2025-12-06T16:05:14.937920435Z","level":"INFO","msg":"stream: created new stream","id":"c7ladxh7"}
3
+ {"time":"2025-12-06T16:05:14.937950636Z","level":"INFO","msg":"stream: started","id":"c7ladxh7"}
4
+ {"time":"2025-12-06T16:05:14.937992748Z","level":"INFO","msg":"writer: started","stream_id":"c7ladxh7"}
5
+ {"time":"2025-12-06T16:05:14.938002175Z","level":"INFO","msg":"handler: started","stream_id":"c7ladxh7"}
6
+ {"time":"2025-12-06T16:05:14.938035855Z","level":"INFO","msg":"sender: started","stream_id":"c7ladxh7"}
7
+ {"time":"2025-12-06T16:25:15.385384708Z","level":"INFO","msg":"stream: closing","id":"c7ladxh7"}
8
+ {"time":"2025-12-06T16:25:18.158973747Z","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
9
+ {"time":"2025-12-06T16:25:18.254188501Z","level":"INFO","msg":"handler: closed","stream_id":"c7ladxh7"}
10
+ {"time":"2025-12-06T16:25:18.254234232Z","level":"INFO","msg":"sender: closed","stream_id":"c7ladxh7"}
11
+ {"time":"2025-12-06T16:25:18.254240105Z","level":"INFO","msg":"stream: closed","id":"c7ladxh7"}
wandb/run-20251206_160514-c7ladxh7/logs/debug.log ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_setup.py:_flush():81] Current SDK version is 0.21.4
2
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_setup.py:_flush():81] Configure stats pid to 17679
3
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_setup.py:_flush():81] Loading settings from /root/.config/wandb/settings
4
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_setup.py:_flush():81] Loading settings from /workspace/wandb/settings
5
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_setup.py:_flush():81] Loading settings from environment variables
6
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_init.py:setup_run_log_directory():686] Logging user logs to outputs/train/act_mobile_arm_bottle_02/wandb/run-20251206_160514-c7ladxh7/logs/debug.log
7
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_init.py:setup_run_log_directory():687] Logging internal logs to outputs/train/act_mobile_arm_bottle_02/wandb/run-20251206_160514-c7ladxh7/logs/debug-internal.log
8
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_init.py:init():813] calling init triggers
9
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_init.py:init():818] wandb.init called with sweep_config: {}
10
+ config: {'dataset': {'repo_id': 'Biamterdex/amd_hackathon_mobile_arm-02', 'root': None, 'episodes': None, 'image_transforms': {'enable': False, 'max_num_transforms': 3, 'random_order': False, 'tfs': {'brightness': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'brightness': [0.8, 1.2]}}, 'contrast': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'contrast': [0.8, 1.2]}}, 'saturation': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'saturation': [0.5, 1.5]}}, 'hue': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'hue': [-0.05, 0.05]}}, 'sharpness': {'weight': 1.0, 'type': 'SharpnessJitter', 'kwargs': {'sharpness': [0.5, 1.5]}}, 'affine': {'weight': 1.0, 'type': 'RandomAffine', 'kwargs': {'degrees': [-5.0, 5.0], 'translate': [0.05, 0.05]}}}}, 'revision': None, 'use_imagenet_stats': True, 'video_backend': 'torchcodec', 'streaming': False}, 'env': None, 'policy': {'type': 'act', 'n_obs_steps': 1, 'input_features': {}, 'output_features': {}, 'device': 'cuda', 'use_amp': False, 'push_to_hub': False, 'repo_id': None, 'private': None, 'tags': None, 'license': None, 'pretrained_path': None, 'chunk_size': 100, 'n_action_steps': 100, 'normalization_mapping': {'VISUAL': <NormalizationMode.MEAN_STD: 'MEAN_STD'>, 'STATE': <NormalizationMode.MEAN_STD: 'MEAN_STD'>, 'ACTION': <NormalizationMode.MEAN_STD: 'MEAN_STD'>}, 'vision_backbone': 'resnet18', 'pretrained_backbone_weights': 'ResNet18_Weights.IMAGENET1K_V1', 'replace_final_stride_with_dilation': False, 'pre_norm': False, 'dim_model': 512, 'n_heads': 8, 'dim_feedforward': 3200, 'feedforward_activation': 'relu', 'n_encoder_layers': 4, 'n_decoder_layers': 1, 'use_vae': True, 'latent_dim': 32, 'n_vae_encoder_layers': 4, 'temporal_ensemble_coeff': None, 'dropout': 0.1, 'kl_weight': 10.0, 'optimizer_lr': 1e-05, 'optimizer_weight_decay': 0.0001, 'optimizer_lr_backbone': 1e-05}, 'output_dir': 'outputs/train/act_mobile_arm_bottle_02', 'job_name': 'act_mobile_arm_bottle_02', 'resume': False, 'seed': 1000, 'num_workers': 4, 'batch_size': 64, 'steps': 1000, 'eval_freq': 20000, 'log_freq': 200, 'save_checkpoint': True, 'save_freq': 20000, 'use_policy_training_preset': True, 'optimizer': {'type': 'adamw', 'lr': 1e-05, 'weight_decay': 0.0001, 'grad_clip_norm': 10.0, 'betas': [0.9, 0.999], 'eps': 1e-08}, 'scheduler': None, 'eval': {'n_episodes': 50, 'batch_size': 50, 'use_async_envs': False}, 'wandb': {'enable': True, 'disable_artifact': False, 'project': 'lerobot', 'entity': None, 'notes': None, 'run_id': None, 'mode': None}, 'checkpoint_path': None, 'rename_map': {}, '_wandb': {}}
11
+ 2025-12-06 16:05:14,465 INFO MainThread:17679 [wandb_init.py:init():854] starting backend
12
+ 2025-12-06 16:05:14,670 INFO MainThread:17679 [wandb_init.py:init():857] sending inform_init request
13
+ 2025-12-06 16:05:14,672 INFO MainThread:17679 [wandb_init.py:init():865] backend started and connected
14
+ 2025-12-06 16:05:14,674 INFO MainThread:17679 [wandb_init.py:init():936] updated telemetry
15
+ 2025-12-06 16:05:14,674 INFO MainThread:17679 [wandb_init.py:init():960] communicating run to backend with 90.0 second timeout
16
+ 2025-12-06 16:05:15,330 INFO MainThread:17679 [wandb_init.py:init():1011] starting run threads in backend
17
+ 2025-12-06 16:05:15,401 INFO MainThread:17679 [wandb_run.py:_console_start():2506] atexit reg
18
+ 2025-12-06 16:05:15,401 INFO MainThread:17679 [wandb_run.py:_redirect():2354] redirect: wrap_raw
19
+ 2025-12-06 16:05:15,401 INFO MainThread:17679 [wandb_run.py:_redirect():2423] Wrapping output streams.
20
+ 2025-12-06 16:05:15,401 INFO MainThread:17679 [wandb_run.py:_redirect():2446] Redirects installed.
21
+ 2025-12-06 16:05:15,403 INFO MainThread:17679 [wandb_init.py:init():1049] run started, returning control to user process
22
+ 2025-12-06 16:25:15,385 INFO wandb-AsyncioManager-main:17679 [service_client.py:_forward_responses():84] Reached EOF.
23
+ 2025-12-06 16:25:15,385 INFO wandb-AsyncioManager-main:17679 [mailbox.py:close():137] Closing mailbox, abandoning 2 handles.
wandb/run-20251206_160514-c7ladxh7/run-c7ladxh7.wandb ADDED
Binary file (75.4 kB). View file