kabilanKB commited on
Commit
17bdf4b
·
verified ·
1 Parent(s): 019f596

Upload folder using huggingface_hub

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitattributes +3 -0
  2. act_so101_test/wandb/debug-internal.log +7 -0
  3. act_so101_test/wandb/debug.log +56 -0
  4. act_so101_test/wandb/run-20251230_221749-q1o5aaw1/files/config.yaml +225 -0
  5. act_so101_test/wandb/run-20251230_221749-q1o5aaw1/files/output.log +53 -0
  6. act_so101_test/wandb/run-20251230_221749-q1o5aaw1/files/requirements.txt +237 -0
  7. act_so101_test/wandb/run-20251230_221749-q1o5aaw1/files/wandb-metadata.json +49 -0
  8. act_so101_test/wandb/run-20251230_221749-q1o5aaw1/files/wandb-summary.json +1 -0
  9. act_so101_test/wandb/run-20251230_221749-q1o5aaw1/logs/debug-core.log +12 -0
  10. act_so101_test/wandb/run-20251230_221749-q1o5aaw1/logs/debug-internal.log +7 -0
  11. act_so101_test/wandb/run-20251230_221749-q1o5aaw1/logs/debug.log +56 -0
  12. act_so101_test/wandb/run-20251230_221749-q1o5aaw1/run-q1o5aaw1.wandb +0 -0
  13. lerobot-act_so101_test/checkpoints/020000/pretrained_model/config.json +70 -0
  14. lerobot-act_so101_test/checkpoints/020000/pretrained_model/model.safetensors +3 -0
  15. lerobot-act_so101_test/checkpoints/020000/pretrained_model/policy_postprocessor.json +32 -0
  16. lerobot-act_so101_test/checkpoints/020000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
  17. lerobot-act_so101_test/checkpoints/020000/pretrained_model/policy_preprocessor.json +64 -0
  18. lerobot-act_so101_test/checkpoints/020000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
  19. lerobot-act_so101_test/checkpoints/020000/pretrained_model/train_config.json +193 -0
  20. lerobot-act_so101_test/checkpoints/020000/training_state/optimizer_param_groups.json +191 -0
  21. lerobot-act_so101_test/checkpoints/020000/training_state/optimizer_state.safetensors +3 -0
  22. lerobot-act_so101_test/checkpoints/020000/training_state/rng_state.safetensors +3 -0
  23. lerobot-act_so101_test/checkpoints/020000/training_state/training_step.json +3 -0
  24. lerobot-act_so101_test/checkpoints/035000/pretrained_model/config.json +70 -0
  25. lerobot-act_so101_test/checkpoints/035000/pretrained_model/model.safetensors +3 -0
  26. lerobot-act_so101_test/checkpoints/035000/pretrained_model/policy_postprocessor.json +32 -0
  27. lerobot-act_so101_test/checkpoints/035000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
  28. lerobot-act_so101_test/checkpoints/035000/pretrained_model/policy_preprocessor.json +64 -0
  29. lerobot-act_so101_test/checkpoints/035000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
  30. lerobot-act_so101_test/checkpoints/035000/pretrained_model/train_config.json +193 -0
  31. lerobot-act_so101_test/checkpoints/035000/training_state/optimizer_param_groups.json +191 -0
  32. lerobot-act_so101_test/checkpoints/035000/training_state/optimizer_state.safetensors +3 -0
  33. lerobot-act_so101_test/checkpoints/035000/training_state/rng_state.safetensors +3 -0
  34. lerobot-act_so101_test/checkpoints/035000/training_state/training_step.json +3 -0
  35. lerobot-act_so101_test/wandb/debug-internal.log +11 -0
  36. lerobot-act_so101_test/wandb/debug.log +23 -0
  37. lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/files/config.yaml +227 -0
  38. lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/files/output.log +257 -0
  39. lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/files/requirements.txt +257 -0
  40. lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/files/wandb-metadata.json +50 -0
  41. lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/files/wandb-summary.json +1 -0
  42. lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/logs/debug-core.log +15 -0
  43. lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/logs/debug-internal.log +11 -0
  44. lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/logs/debug.log +23 -0
  45. lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/run-8ce8thqh.wandb +3 -0
  46. lerobot_diffusion_model/checkpoints/020000/pretrained_model/config.json +82 -0
  47. lerobot_diffusion_model/checkpoints/020000/pretrained_model/model.safetensors +3 -0
  48. lerobot_diffusion_model/checkpoints/020000/pretrained_model/policy_postprocessor.json +32 -0
  49. lerobot_diffusion_model/checkpoints/020000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
  50. lerobot_diffusion_model/checkpoints/020000/pretrained_model/policy_preprocessor.json +56 -0
.gitattributes CHANGED
@@ -33,3 +33,6 @@ 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
+ lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/run-8ce8thqh.wandb filter=lfs diff=lfs merge=lfs -text
37
+ lerobot_diffusion_model/wandb/run-20251231_105755-pkxhwzf2/run-pkxhwzf2.wandb filter=lfs diff=lfs merge=lfs -text
38
+ lerobotact_so101_test/wandb/run-20251230_222124-umsnhs3c/run-umsnhs3c.wandb filter=lfs diff=lfs merge=lfs -text
act_so101_test/wandb/debug-internal.log ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {"time":"2025-12-30T22:17:50.001599125+05:30","level":"INFO","msg":"stream: starting","core version":"0.21.4"}
2
+ {"time":"2025-12-30T22:17:50.459815178+05:30","level":"INFO","msg":"stream: created new stream","id":"q1o5aaw1"}
3
+ {"time":"2025-12-30T22:17:50.459832404+05:30","level":"INFO","msg":"stream: started","id":"q1o5aaw1"}
4
+ {"time":"2025-12-30T22:17:50.459873772+05:30","level":"INFO","msg":"writer: started","stream_id":"q1o5aaw1"}
5
+ {"time":"2025-12-30T22:17:50.459884981+05:30","level":"INFO","msg":"handler: started","stream_id":"q1o5aaw1"}
6
+ {"time":"2025-12-30T22:17:50.459918165+05:30","level":"INFO","msg":"sender: started","stream_id":"q1o5aaw1"}
7
+ {"time":"2025-12-30T22:18:28.195147193+05:30","level":"INFO","msg":"stream: closing","id":"q1o5aaw1"}
act_so101_test/wandb/debug.log ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_setup.py:_flush():81] Current SDK version is 0.21.4
2
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_setup.py:_flush():81] Configure stats pid to 2365002
3
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_setup.py:_flush():81] Loading settings from /home/zeux/.config/wandb/settings
4
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_setup.py:_flush():81] Loading settings from /home/zeux/wandb/settings
5
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_setup.py:_flush():81] Loading settings from environment variables
6
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_init.py:setup_run_log_directory():686] Logging user logs to outputs/train/act_so101_test/wandb/run-20251230_221749-q1o5aaw1/logs/debug.log
7
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_init.py:setup_run_log_directory():687] Logging internal logs to outputs/train/act_so101_test/wandb/run-20251230_221749-q1o5aaw1/logs/debug-internal.log
8
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_init.py:init():813] calling init triggers
9
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_init.py:init():818] wandb.init called with sweep_config: {}
10
+ config: {'dataset': {'repo_id': 'lerobot-act', 'root': '/home/zeux/lerobot_shadow/lerobotact-dataset', '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': 'pyav', 'streaming': False}, 'env': None, 'policy': {'type': 'act', 'n_obs_steps': 1, 'input_features': {}, 'output_features': {}, 'device': 'cuda', 'use_amp': True, 'push_to_hub': True, 'repo_id': 'local/act-so101-model', '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_so101_test', 'job_name': 'act', 'resume': False, 'seed': 1000, 'num_workers': 1, 'batch_size': 8, 'steps': 20000, '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-30 22:17:49,794 INFO MainThread:2365002 [wandb_init.py:init():854] starting backend
12
+ 2025-12-30 22:17:49,996 INFO MainThread:2365002 [wandb_init.py:init():857] sending inform_init request
13
+ 2025-12-30 22:17:49,999 INFO MainThread:2365002 [wandb_init.py:init():865] backend started and connected
14
+ 2025-12-30 22:17:50,000 INFO MainThread:2365002 [wandb_init.py:init():936] updated telemetry
15
+ 2025-12-30 22:17:50,000 INFO MainThread:2365002 [wandb_init.py:init():960] communicating run to backend with 90.0 second timeout
16
+ 2025-12-30 22:17:50,852 INFO MainThread:2365002 [wandb_init.py:init():1011] starting run threads in backend
17
+ 2025-12-30 22:17:50,911 INFO MainThread:2365002 [wandb_run.py:_console_start():2506] atexit reg
18
+ 2025-12-30 22:17:50,911 INFO MainThread:2365002 [wandb_run.py:_redirect():2354] redirect: wrap_raw
19
+ 2025-12-30 22:17:50,911 INFO MainThread:2365002 [wandb_run.py:_redirect():2423] Wrapping output streams.
20
+ 2025-12-30 22:17:50,911 INFO MainThread:2365002 [wandb_run.py:_redirect():2446] Redirects installed.
21
+ 2025-12-30 22:17:50,911 INFO MainThread:2365002 [wandb_init.py:init():1049] run started, returning control to user process
22
+ 2025-12-30 22:18:28,195 INFO wandb-AsyncioManager-main:2365002 [service_client.py:_forward_responses():84] Reached EOF.
23
+ 2025-12-30 22:18:28,195 INFO wandb-AsyncioManager-main:2365002 [mailbox.py:close():137] Closing mailbox, abandoning 1 handles.
24
+ 2025-12-30 22:18:28,592 ERROR wandb-AsyncioManager-main:2365002 [asyncio_manager.py:fn_wrap_exceptions():183] Uncaught exception in run_soon callback.
25
+ Traceback (most recent call last):
26
+ File "/home/zeux/.local/lib/python3.10/site-packages/wandb/sdk/lib/asyncio_manager.py", line 181, in fn_wrap_exceptions
27
+ await fn()
28
+ File "/home/zeux/.local/lib/python3.10/site-packages/wandb/sdk/lib/service/service_client.py", line 65, in _send_server_request
29
+ await self._writer.drain()
30
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/asyncio/streams.py", line 371, in drain
31
+ await self._protocol._drain_helper()
32
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/asyncio/streams.py", line 167, in _drain_helper
33
+ raise ConnectionResetError('Connection lost')
34
+ ConnectionResetError: Connection lost
35
+ 2025-12-30 22:18:28,601 ERROR wandb-AsyncioManager-main:2365002 [asyncio_manager.py:fn_wrap_exceptions():183] Uncaught exception in run_soon callback.
36
+ Traceback (most recent call last):
37
+ File "/home/zeux/.local/lib/python3.10/site-packages/wandb/sdk/lib/asyncio_manager.py", line 181, in fn_wrap_exceptions
38
+ await fn()
39
+ File "/home/zeux/.local/lib/python3.10/site-packages/wandb/sdk/lib/service/service_client.py", line 65, in _send_server_request
40
+ await self._writer.drain()
41
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/asyncio/streams.py", line 371, in drain
42
+ await self._protocol._drain_helper()
43
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/asyncio/streams.py", line 167, in _drain_helper
44
+ raise ConnectionResetError('Connection lost')
45
+ ConnectionResetError: Connection lost
46
+ 2025-12-30 22:18:28,601 ERROR wandb-AsyncioManager-main:2365002 [asyncio_manager.py:fn_wrap_exceptions():183] Uncaught exception in run_soon callback.
47
+ Traceback (most recent call last):
48
+ File "/home/zeux/.local/lib/python3.10/site-packages/wandb/sdk/lib/asyncio_manager.py", line 181, in fn_wrap_exceptions
49
+ await fn()
50
+ File "/home/zeux/.local/lib/python3.10/site-packages/wandb/sdk/lib/service/service_client.py", line 65, in _send_server_request
51
+ await self._writer.drain()
52
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/asyncio/streams.py", line 371, in drain
53
+ await self._protocol._drain_helper()
54
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/asyncio/streams.py", line 167, in _drain_helper
55
+ raise ConnectionResetError('Connection lost')
56
+ ConnectionResetError: Connection lost
act_so101_test/wandb/run-20251230_221749-q1o5aaw1/files/config.yaml ADDED
@@ -0,0 +1,225 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _wandb:
2
+ value:
3
+ cli_version: 0.21.4
4
+ e:
5
+ jldifxu3qy62ax4zkqe4sl8faxnhsjg9:
6
+ args:
7
+ - --dataset.repo_id=lerobot-act
8
+ - --dataset.root=/home/zeux/lerobot_shadow/lerobotact-dataset
9
+ - --dataset.video_backend=pyav
10
+ - --policy.type=act
11
+ - --policy.repo_id=local/act-so101-model
12
+ - --output_dir=outputs/train/act_so101_test
13
+ - --policy.device=cuda
14
+ - --policy.use_amp=true
15
+ - --batch_size=8
16
+ - --num_workers=1
17
+ - --steps=20000
18
+ - --wandb.enable=true
19
+ codePath: miniconda3/envs/lerobot/bin/lerobot-train
20
+ codePathLocal: miniconda3/envs/lerobot/bin/lerobot-train
21
+ cpu_count: 24
22
+ cpu_count_logical: 24
23
+ cudaVersion: "13.0"
24
+ disk:
25
+ /:
26
+ total: "200449576960"
27
+ used: "34402668544"
28
+ executable: /home/zeux/miniconda3/envs/lerobot/bin/python3.10
29
+ gpu: NVIDIA RTX PRO 6000 Blackwell Workstation Edition
30
+ gpu_count: 1
31
+ gpu_nvidia:
32
+ - architecture: Blackwell
33
+ cudaCores: 24064
34
+ memoryTotal: "102641958912"
35
+ name: NVIDIA RTX PRO 6000 Blackwell Workstation Edition
36
+ uuid: GPU-8c66f3c4-00c8-00a6-e5ff-633c5558ec8e
37
+ host: zeux
38
+ memory:
39
+ total: "134489886720"
40
+ os: Linux-6.8.0-90-generic-x86_64-with-glibc2.35
41
+ program: /home/zeux/miniconda3/envs/lerobot/bin/lerobot-train
42
+ python: CPython 3.10.19
43
+ root: outputs/train/act_so101_test
44
+ startedAt: "2025-12-30T16:47:49.793724Z"
45
+ writerId: jldifxu3qy62ax4zkqe4sl8faxnhsjg9
46
+ m: []
47
+ python_version: 3.10.19
48
+ t:
49
+ "1":
50
+ - 1
51
+ - 11
52
+ - 41
53
+ - 49
54
+ - 51
55
+ - 71
56
+ - 83
57
+ "2":
58
+ - 1
59
+ - 11
60
+ - 41
61
+ - 49
62
+ - 51
63
+ - 71
64
+ - 83
65
+ "3":
66
+ - 13
67
+ - 15
68
+ - 16
69
+ "4": 3.10.19
70
+ "5": 0.21.4
71
+ "6": 4.57.3
72
+ "10":
73
+ - 21
74
+ "12": 0.21.4
75
+ "13": linux-x86_64
76
+ batch_size:
77
+ value: 8
78
+ checkpoint_path:
79
+ value: null
80
+ dataset:
81
+ value:
82
+ episodes: null
83
+ image_transforms:
84
+ enable: false
85
+ max_num_transforms: 3
86
+ random_order: false
87
+ tfs:
88
+ affine:
89
+ kwargs:
90
+ degrees:
91
+ - -5
92
+ - 5
93
+ translate:
94
+ - 0.05
95
+ - 0.05
96
+ type: RandomAffine
97
+ weight: 1
98
+ brightness:
99
+ kwargs:
100
+ brightness:
101
+ - 0.8
102
+ - 1.2
103
+ type: ColorJitter
104
+ weight: 1
105
+ contrast:
106
+ kwargs:
107
+ contrast:
108
+ - 0.8
109
+ - 1.2
110
+ type: ColorJitter
111
+ weight: 1
112
+ hue:
113
+ kwargs:
114
+ hue:
115
+ - -0.05
116
+ - 0.05
117
+ type: ColorJitter
118
+ weight: 1
119
+ saturation:
120
+ kwargs:
121
+ saturation:
122
+ - 0.5
123
+ - 1.5
124
+ type: ColorJitter
125
+ weight: 1
126
+ sharpness:
127
+ kwargs:
128
+ sharpness:
129
+ - 0.5
130
+ - 1.5
131
+ type: SharpnessJitter
132
+ weight: 1
133
+ repo_id: lerobot-act
134
+ revision: null
135
+ root: /home/zeux/lerobot_shadow/lerobotact-dataset
136
+ streaming: false
137
+ use_imagenet_stats: true
138
+ video_backend: pyav
139
+ env:
140
+ value: null
141
+ eval:
142
+ value:
143
+ batch_size: 50
144
+ n_episodes: 50
145
+ use_async_envs: false
146
+ eval_freq:
147
+ value: 20000
148
+ job_name:
149
+ value: act
150
+ log_freq:
151
+ value: 200
152
+ num_workers:
153
+ value: 1
154
+ optimizer:
155
+ value:
156
+ betas:
157
+ - 0.9
158
+ - 0.999
159
+ eps: 1e-08
160
+ grad_clip_norm: 10
161
+ lr: 1e-05
162
+ type: adamw
163
+ weight_decay: 0.0001
164
+ output_dir:
165
+ value: outputs/train/act_so101_test
166
+ policy:
167
+ value:
168
+ chunk_size: 100
169
+ device: cuda
170
+ dim_feedforward: 3200
171
+ dim_model: 512
172
+ dropout: 0.1
173
+ feedforward_activation: relu
174
+ kl_weight: 10
175
+ latent_dim: 32
176
+ license: null
177
+ n_action_steps: 100
178
+ n_decoder_layers: 1
179
+ n_encoder_layers: 4
180
+ n_heads: 8
181
+ n_obs_steps: 1
182
+ n_vae_encoder_layers: 4
183
+ normalization_mapping:
184
+ ACTION: MEAN_STD
185
+ STATE: MEAN_STD
186
+ VISUAL: MEAN_STD
187
+ optimizer_lr: 1e-05
188
+ optimizer_lr_backbone: 1e-05
189
+ optimizer_weight_decay: 0.0001
190
+ pre_norm: false
191
+ pretrained_backbone_weights: ResNet18_Weights.IMAGENET1K_V1
192
+ pretrained_path: null
193
+ private: null
194
+ push_to_hub: true
195
+ replace_final_stride_with_dilation: false
196
+ repo_id: local/act-so101-model
197
+ tags: null
198
+ temporal_ensemble_coeff: null
199
+ type: act
200
+ use_amp: true
201
+ use_vae: true
202
+ vision_backbone: resnet18
203
+ resume:
204
+ value: false
205
+ save_checkpoint:
206
+ value: true
207
+ save_freq:
208
+ value: 20000
209
+ scheduler:
210
+ value: null
211
+ seed:
212
+ value: 1000
213
+ steps:
214
+ value: 20000
215
+ use_policy_training_preset:
216
+ value: true
217
+ wandb:
218
+ value:
219
+ disable_artifact: false
220
+ enable: true
221
+ entity: null
222
+ mode: null
223
+ notes: null
224
+ project: lerobot
225
+ run_id: null
act_so101_test/wandb/run-20251230_221749-q1o5aaw1/files/output.log ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ INFO 2025-12-30 22:17:50 db_utils.py:102 Logs will be synced with wandb.
2
+ INFO 2025-12-30 22:17:50 db_utils.py:103 Track this run --> https://wandb.ai/nwaves-ai-nwaves/lerobot/runs/q1o5aaw1
3
+ INFO 2025-12-30 22:17:50 ot_train.py:183 Creating dataset
4
+ INFO 2025-12-30 22:17:50 ot_train.py:202 Creating policy
5
+ INFO 2025-12-30 22:17:51 ot_train.py:247 Creating optimizer and scheduler
6
+ INFO 2025-12-30 22:17:51 ot_train.py:259 Output dir: outputs/train/act_so101_test
7
+ INFO 2025-12-30 22:17:51 ot_train.py:264 cfg.steps=20000 (20K)
8
+ INFO 2025-12-30 22:17:51 ot_train.py:265 dataset.num_frames=17774 (18K)
9
+ INFO 2025-12-30 22:17:51 ot_train.py:266 dataset.num_episodes=20
10
+ INFO 2025-12-30 22:17:51 ot_train.py:269 Effective batch size: 8 x 1 = 8
11
+ INFO 2025-12-30 22:17:51 ot_train.py:270 num_learnable_params=51597190 (52M)
12
+ INFO 2025-12-30 22:17:51 ot_train.py:271 num_total_params=51597190 (52M)
13
+ INFO 2025-12-30 22:17:51 ot_train.py:327 Start offline training on a fixed dataset
14
+ Traceback (most recent call last):
15
+ File "/home/zeux/miniconda3/envs/lerobot/bin/lerobot-train", line 7, in <module>
16
+ sys.exit(main())
17
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/scripts/lerobot_train.py", line 449, in main
18
+ train()
19
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/configs/parser.py", line 233, in wrapper_inner
20
+ response = fn(cfg, *args, **kwargs)
21
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/scripts/lerobot_train.py", line 335, in train
22
+ train_tracker, output_dict = update_policy(
23
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/scripts/lerobot_train.py", line 95, in update_policy
24
+ accelerator.backward(loss)
25
+ File "/home/zeux/.local/lib/python3.10/site-packages/accelerate/accelerator.py", line 2852, in backward
26
+ loss.backward(**kwargs)
27
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/torch/_tensor.py", line 625, in backward
28
+ torch.autograd.backward(
29
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/torch/autograd/__init__.py", line 354, in backward
30
+ _engine_run_backward(
31
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/torch/autograd/graph.py", line 841, in _engine_run_backward
32
+ return Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass
33
+ KeyboardInterrupt
34
+ Traceback (most recent call last):
35
+ File "/home/zeux/miniconda3/envs/lerobot/bin/lerobot-train", line 7, in <module>
36
+ sys.exit(main())
37
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/scripts/lerobot_train.py", line 449, in main
38
+ train()
39
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/configs/parser.py", line 233, in wrapper_inner
40
+ response = fn(cfg, *args, **kwargs)
41
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/scripts/lerobot_train.py", line 335, in train
42
+ train_tracker, output_dict = update_policy(
43
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/scripts/lerobot_train.py", line 95, in update_policy
44
+ accelerator.backward(loss)
45
+ File "/home/zeux/.local/lib/python3.10/site-packages/accelerate/accelerator.py", line 2852, in backward
46
+ loss.backward(**kwargs)
47
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/torch/_tensor.py", line 625, in backward
48
+ torch.autograd.backward(
49
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/torch/autograd/__init__.py", line 354, in backward
50
+ _engine_run_backward(
51
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/torch/autograd/graph.py", line 841, in _engine_run_backward
52
+ return Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass
53
+ KeyboardInterrupt
act_so101_test/wandb/run-20251230_221749-q1o5aaw1/files/requirements.txt ADDED
@@ -0,0 +1,237 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ nvidia-cusolver==12.0.3.29
2
+ nvidia-curand==10.4.0.35
3
+ anyio==4.12.0
4
+ yarl==1.22.0
5
+ comfyui-workflow-templates-media-other==0.3.15
6
+ tomlkit==0.13.3
7
+ cloudpickle==3.1.2
8
+ typing_extensions==4.15.0
9
+ sentry-sdk==2.48.0
10
+ semver==3.0.4
11
+ mpmath==1.3.0
12
+ draccus==0.10.0
13
+ multidict==6.7.0
14
+ nvidia-cufile-cu12==1.11.1.6
15
+ tokenizers==0.22.1
16
+ GitPython==3.1.45
17
+ datasets==4.1.1
18
+ PySide6_Essentials==6.10.1
19
+ comfyui-workflow-templates-media-image==0.3.17
20
+ Pygments==2.19.2
21
+ aiohappyeyeballs==2.6.1
22
+ protobuf==6.33.2
23
+ PySide6==6.10.1
24
+ comfyui-workflow-templates-core==0.3.12
25
+ PySide2==5.15.2.1
26
+ pyserial==3.5
27
+ accelerate==1.12.0
28
+ pfzy==0.3.4
29
+ scipy==1.15.3
30
+ frozenlist==1.8.0
31
+ nvidia-nvshmem-cu13==3.3.24
32
+ transformers==4.57.3
33
+ typing-inspection==0.4.2
34
+ einops==0.8.1
35
+ shellingham==1.5.4
36
+ Farama-Notifications==0.0.4
37
+ psutil==7.1.3
38
+ rerun-sdk==0.26.2
39
+ PyYAML==6.0.3
40
+ nvidia-cufft==12.0.0.15
41
+ cookiecutter==2.6.0
42
+ rich==14.2.0
43
+ SQLAlchemy==2.0.44
44
+ pydantic_core==2.41.5
45
+ nvidia-nvjitlink==13.0.39
46
+ questionary==2.1.1
47
+ requests==2.32.5
48
+ nvidia-cuda-nvrtc==13.0.48
49
+ torchcodec==0.5
50
+ PyOpenGL-accelerate==3.1.10
51
+ comfyui_frontend_package==1.33.10
52
+ networkx==3.4.2
53
+ nvidia-nvtx==13.0.39
54
+ setuptools==80.9.0
55
+ trampoline==0.1.2
56
+ pandas==2.3.3
57
+ httpx==0.28.1
58
+ av==15.1.0
59
+ comfy-cli==1.5.3
60
+ PyOpenGL==3.1.10
61
+ tomli==2.3.0
62
+ safetensors==0.7.0
63
+ regex==2025.11.3
64
+ wcwidth==0.2.14
65
+ pyyaml-include==1.4.1
66
+ gymnasium==1.2.3
67
+ kornia==0.8.2
68
+ packaging==25.0
69
+ nvidia-nccl-cu13==2.27.7
70
+ pydantic==2.12.5
71
+ mixpanel==5.0.0
72
+ markdown-it-py==4.0.0
73
+ multiprocess==0.70.16
74
+ usd-exchange==2.1.0
75
+ h11==0.16.0
76
+ mergedeep==1.3.4
77
+ shiboken6==6.10.1
78
+ text-unidecode==1.3
79
+ deepdiff==8.6.1
80
+ toml==0.10.2
81
+ aiosignal==1.4.0
82
+ imageio-ffmpeg==0.6.0
83
+ python-dateutil==2.9.0.post0
84
+ uv==0.9.16
85
+ jsonlines==4.0.0
86
+ xxhash==3.6.0
87
+ numpy==2.2.6
88
+ nvidia-cuda-runtime==13.0.48
89
+ arrow==1.4.0
90
+ spandrel==0.4.1
91
+ nvidia-cuda-cupti==13.0.48
92
+ python-dotenv==1.2.1
93
+ async-timeout==5.0.1
94
+ urllib3==2.6.2
95
+ fsspec==2025.9.0
96
+ nvidia-cusparse==12.6.2.49
97
+ binaryornot==0.4.4
98
+ nvidia-cufile==1.15.0.42
99
+ mdurl==0.1.2
100
+ pydantic-settings==2.12.0
101
+ gitdb==4.0.12
102
+ kornia_rs==0.1.10
103
+ attrs==25.4.0
104
+ mypy_extensions==1.1.0
105
+ tqdm==4.67.1
106
+ alembic==1.17.2
107
+ comfyui-embedded-docs==0.3.1
108
+ python-slugify==8.0.4
109
+ cmake==4.1.3
110
+ smmap==5.0.2
111
+ comfyui-workflow-templates-media-api==0.3.14
112
+ torchsde==0.2.6
113
+ pathspec==0.12.1
114
+ sentencepiece==0.2.1
115
+ charset-normalizer==3.4.4
116
+ nvidia-cusparselt-cu13==0.8.0
117
+ python-xlib==0.33
118
+ ruff==0.14.8
119
+ platformdirs==4.5.1
120
+ inquirerpy==0.3.4
121
+ evdev==1.9.2
122
+ huggingface-hub==0.35.3
123
+ nvidia-cublas==13.0.0.19
124
+ prompt_toolkit==3.0.52
125
+ ImageIO==2.37.2
126
+ termcolor==3.3.0
127
+ wandb==0.21.4
128
+ annotated-types==0.7.0
129
+ comfyui-workflow-templates-media-video==0.3.12
130
+ typer==0.20.0
131
+ dill==0.4.0
132
+ propcache==0.4.1
133
+ typing-inspect==0.9.0
134
+ greenlet==3.3.0
135
+ hf-xet==1.2.0
136
+ shiboken2==5.15.2.1
137
+ tzdata==2025.2
138
+ aiohttp==3.13.2
139
+ websocket-client==1.9.0
140
+ nvidia-cudnn-cu13==9.13.0.50
141
+ pyarrow==22.0.0
142
+ orderly-set==5.5.0
143
+ pynput==1.8.1
144
+ diffusers==0.35.2
145
+ PySide6_Addons==6.10.1
146
+ asgiref==3.11.0
147
+ httpcore==1.0.9
148
+ exceptiongroup==1.3.1
149
+ filelock==3.20.0
150
+ comfyui_workflow_templates==0.7.25
151
+ opencv-python-headless==4.12.0.88
152
+ nvidia-curand-cu12==10.3.7.77
153
+ nvidia-cudnn-cu12==9.5.1.17
154
+ nvidia-nvtx-cu12==12.6.77
155
+ cloudpickle==3.1.2
156
+ pytz==2025.2
157
+ nvidia-cusparselt-cu12==0.6.3
158
+ nvidia-cusparselt-cu12==0.7.1
159
+ sentry-sdk==2.48.0
160
+ importlib_metadata==8.7.1
161
+ draccus==0.10.0
162
+ nvidia-cufile-cu12==1.11.1.6
163
+ datasets==4.1.1
164
+ nvidia-cusolver-cu12==11.7.1.2
165
+ protobuf==6.33.2
166
+ pyserial==3.5
167
+ accelerate==1.12.0
168
+ pfzy==0.3.4
169
+ setuptools==80.9.0
170
+ nvidia-cuda-cupti-cu12==12.6.80
171
+ Farama-Notifications==0.0.4
172
+ rerun-sdk==0.26.2
173
+ PyYAML==6.0.3
174
+ torchvision==0.24.1+cu130
175
+ requests==2.32.5
176
+ torchcodec==0.5
177
+ idna==3.11
178
+ zipp==3.23.0
179
+ pandas==2.3.3
180
+ av==15.1.0
181
+ nvidia-cusparse-cu12==12.5.4.2
182
+ pyyaml-include==1.4.1
183
+ gymnasium==1.2.3
184
+ nvidia-cuda-runtime-cu12==12.6.77
185
+ triton==3.6.0+git9844da95
186
+ triton==3.5.1
187
+ multiprocess==0.70.16
188
+ lerobot==0.4.2
189
+ Jinja2==3.1.6
190
+ click==8.3.1
191
+ nvidia-nccl-cu12==2.28.9
192
+ nvidia-nccl-cu12==2.26.2
193
+ mergedeep==1.3.4
194
+ deepdiff==8.6.1
195
+ toml==0.10.2
196
+ sympy==1.14.0
197
+ hf_transfer==0.1.9
198
+ imageio-ffmpeg==0.6.0
199
+ python-dateutil==2.9.0.post0
200
+ jsonlines==4.0.0
201
+ xxhash==3.6.0
202
+ nvidia-cuda-nvrtc-cu12==12.6.77
203
+ urllib3==2.6.2
204
+ fsspec==2025.9.0
205
+ pillow==12.0.0
206
+ pytorch-triton==3.1.0+cf34004b8a
207
+ cuda-pathfinder==1.2.2
208
+ mypy_extensions==1.1.0
209
+ cmake==4.1.3
210
+ certifi==2025.11.12
211
+ num2words==0.5.14
212
+ python-xlib==0.33
213
+ nvidia-nvjitlink-cu12==12.6.85
214
+ platformdirs==4.5.1
215
+ torch==2.9.1+cu130
216
+ inquirerpy==0.3.4
217
+ evdev==1.9.2
218
+ huggingface-hub==0.35.3
219
+ nvidia-nvshmem-cu12==3.4.5
220
+ ImageIO==2.37.2
221
+ termcolor==3.2.0
222
+ six==1.17.0
223
+ wandb==0.21.4
224
+ MarkupSafe==3.0.3
225
+ wheel==0.45.1
226
+ dill==0.4.0
227
+ typing-inspect==0.9.0
228
+ cuda-bindings==12.9.4
229
+ pip==25.3
230
+ pyarrow==22.0.0
231
+ orderly-set==5.5.0
232
+ pynput==1.8.1
233
+ diffusers==0.35.2
234
+ nvidia-cublas-cu12==12.6.4.1
235
+ docopt==0.6.2
236
+ nvidia-cufft-cu12==11.3.0.4
237
+ opencv-python-headless==4.12.0.88
act_so101_test/wandb/run-20251230_221749-q1o5aaw1/files/wandb-metadata.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-6.8.0-90-generic-x86_64-with-glibc2.35",
3
+ "python": "CPython 3.10.19",
4
+ "startedAt": "2025-12-30T16:47:49.793724Z",
5
+ "args": [
6
+ "--dataset.repo_id=lerobot-act",
7
+ "--dataset.root=/home/zeux/lerobot_shadow/lerobotact-dataset",
8
+ "--dataset.video_backend=pyav",
9
+ "--policy.type=act",
10
+ "--policy.repo_id=local/act-so101-model",
11
+ "--output_dir=outputs/train/act_so101_test",
12
+ "--policy.device=cuda",
13
+ "--policy.use_amp=true",
14
+ "--batch_size=8",
15
+ "--num_workers=1",
16
+ "--steps=20000",
17
+ "--wandb.enable=true"
18
+ ],
19
+ "program": "/home/zeux/miniconda3/envs/lerobot/bin/lerobot-train",
20
+ "codePath": "miniconda3/envs/lerobot/bin/lerobot-train",
21
+ "codePathLocal": "miniconda3/envs/lerobot/bin/lerobot-train",
22
+ "root": "outputs/train/act_so101_test",
23
+ "host": "zeux",
24
+ "executable": "/home/zeux/miniconda3/envs/lerobot/bin/python3.10",
25
+ "cpu_count": 24,
26
+ "cpu_count_logical": 24,
27
+ "gpu": "NVIDIA RTX PRO 6000 Blackwell Workstation Edition",
28
+ "gpu_count": 1,
29
+ "disk": {
30
+ "/": {
31
+ "total": "200449576960",
32
+ "used": "34402668544"
33
+ }
34
+ },
35
+ "memory": {
36
+ "total": "134489886720"
37
+ },
38
+ "gpu_nvidia": [
39
+ {
40
+ "name": "NVIDIA RTX PRO 6000 Blackwell Workstation Edition",
41
+ "memoryTotal": "102641958912",
42
+ "cudaCores": 24064,
43
+ "architecture": "Blackwell",
44
+ "uuid": "GPU-8c66f3c4-00c8-00a6-e5ff-633c5558ec8e"
45
+ }
46
+ ],
47
+ "cudaVersion": "13.0",
48
+ "writerId": "jldifxu3qy62ax4zkqe4sl8faxnhsjg9"
49
+ }
act_so101_test/wandb/run-20251230_221749-q1o5aaw1/files/wandb-summary.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"_wandb":{"runtime":37},"_runtime":37}
act_so101_test/wandb/run-20251230_221749-q1o5aaw1/logs/debug-core.log ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2025-12-30T22:17:49.803961932+05:30","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpp_mdrrv8/port-2365002.txt","pid":2365002,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
2
+ {"time":"2025-12-30T22:17:49.804217382+05:30","level":"INFO","msg":"server: will exit if parent process dies","ppid":2365002}
3
+ {"time":"2025-12-30T22:17:49.804200654+05:30","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-2365002-2365059-351286409/socket","Net":"unix"}}
4
+ {"time":"2025-12-30T22:17:49.996879889+05:30","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
5
+ {"time":"2025-12-30T22:17:50.001521224+05:30","level":"INFO","msg":"handleInformInit: received","streamId":"q1o5aaw1","id":"1(@)"}
6
+ {"time":"2025-12-30T22:17:50.459834724+05:30","level":"INFO","msg":"handleInformInit: stream started","streamId":"q1o5aaw1","id":"1(@)"}
7
+ {"time":"2025-12-30T22:18:28.195121336+05:30","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
8
+ {"time":"2025-12-30T22:18:28.195172479+05:30","level":"INFO","msg":"connection: closing","id":"1(@)"}
9
+ {"time":"2025-12-30T22:18:28.195193073+05:30","level":"INFO","msg":"server is shutting down"}
10
+ {"time":"2025-12-30T22:18:28.195214713+05:30","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
11
+ {"time":"2025-12-30T22:18:28.195230222+05:30","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-2365002-2365059-351286409/socket","Net":"unix"}}
12
+ {"time":"2025-12-30T22:18:28.922840586+05:30","level":"INFO","msg":"server: parent process exited, terminating service process"}
act_so101_test/wandb/run-20251230_221749-q1o5aaw1/logs/debug-internal.log ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {"time":"2025-12-30T22:17:50.001599125+05:30","level":"INFO","msg":"stream: starting","core version":"0.21.4"}
2
+ {"time":"2025-12-30T22:17:50.459815178+05:30","level":"INFO","msg":"stream: created new stream","id":"q1o5aaw1"}
3
+ {"time":"2025-12-30T22:17:50.459832404+05:30","level":"INFO","msg":"stream: started","id":"q1o5aaw1"}
4
+ {"time":"2025-12-30T22:17:50.459873772+05:30","level":"INFO","msg":"writer: started","stream_id":"q1o5aaw1"}
5
+ {"time":"2025-12-30T22:17:50.459884981+05:30","level":"INFO","msg":"handler: started","stream_id":"q1o5aaw1"}
6
+ {"time":"2025-12-30T22:17:50.459918165+05:30","level":"INFO","msg":"sender: started","stream_id":"q1o5aaw1"}
7
+ {"time":"2025-12-30T22:18:28.195147193+05:30","level":"INFO","msg":"stream: closing","id":"q1o5aaw1"}
act_so101_test/wandb/run-20251230_221749-q1o5aaw1/logs/debug.log ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_setup.py:_flush():81] Current SDK version is 0.21.4
2
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_setup.py:_flush():81] Configure stats pid to 2365002
3
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_setup.py:_flush():81] Loading settings from /home/zeux/.config/wandb/settings
4
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_setup.py:_flush():81] Loading settings from /home/zeux/wandb/settings
5
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_setup.py:_flush():81] Loading settings from environment variables
6
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_init.py:setup_run_log_directory():686] Logging user logs to outputs/train/act_so101_test/wandb/run-20251230_221749-q1o5aaw1/logs/debug.log
7
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_init.py:setup_run_log_directory():687] Logging internal logs to outputs/train/act_so101_test/wandb/run-20251230_221749-q1o5aaw1/logs/debug-internal.log
8
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_init.py:init():813] calling init triggers
9
+ 2025-12-30 22:17:49,794 INFO MainThread:2365002 [wandb_init.py:init():818] wandb.init called with sweep_config: {}
10
+ config: {'dataset': {'repo_id': 'lerobot-act', 'root': '/home/zeux/lerobot_shadow/lerobotact-dataset', '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': 'pyav', 'streaming': False}, 'env': None, 'policy': {'type': 'act', 'n_obs_steps': 1, 'input_features': {}, 'output_features': {}, 'device': 'cuda', 'use_amp': True, 'push_to_hub': True, 'repo_id': 'local/act-so101-model', '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_so101_test', 'job_name': 'act', 'resume': False, 'seed': 1000, 'num_workers': 1, 'batch_size': 8, 'steps': 20000, '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-30 22:17:49,794 INFO MainThread:2365002 [wandb_init.py:init():854] starting backend
12
+ 2025-12-30 22:17:49,996 INFO MainThread:2365002 [wandb_init.py:init():857] sending inform_init request
13
+ 2025-12-30 22:17:49,999 INFO MainThread:2365002 [wandb_init.py:init():865] backend started and connected
14
+ 2025-12-30 22:17:50,000 INFO MainThread:2365002 [wandb_init.py:init():936] updated telemetry
15
+ 2025-12-30 22:17:50,000 INFO MainThread:2365002 [wandb_init.py:init():960] communicating run to backend with 90.0 second timeout
16
+ 2025-12-30 22:17:50,852 INFO MainThread:2365002 [wandb_init.py:init():1011] starting run threads in backend
17
+ 2025-12-30 22:17:50,911 INFO MainThread:2365002 [wandb_run.py:_console_start():2506] atexit reg
18
+ 2025-12-30 22:17:50,911 INFO MainThread:2365002 [wandb_run.py:_redirect():2354] redirect: wrap_raw
19
+ 2025-12-30 22:17:50,911 INFO MainThread:2365002 [wandb_run.py:_redirect():2423] Wrapping output streams.
20
+ 2025-12-30 22:17:50,911 INFO MainThread:2365002 [wandb_run.py:_redirect():2446] Redirects installed.
21
+ 2025-12-30 22:17:50,911 INFO MainThread:2365002 [wandb_init.py:init():1049] run started, returning control to user process
22
+ 2025-12-30 22:18:28,195 INFO wandb-AsyncioManager-main:2365002 [service_client.py:_forward_responses():84] Reached EOF.
23
+ 2025-12-30 22:18:28,195 INFO wandb-AsyncioManager-main:2365002 [mailbox.py:close():137] Closing mailbox, abandoning 1 handles.
24
+ 2025-12-30 22:18:28,592 ERROR wandb-AsyncioManager-main:2365002 [asyncio_manager.py:fn_wrap_exceptions():183] Uncaught exception in run_soon callback.
25
+ Traceback (most recent call last):
26
+ File "/home/zeux/.local/lib/python3.10/site-packages/wandb/sdk/lib/asyncio_manager.py", line 181, in fn_wrap_exceptions
27
+ await fn()
28
+ File "/home/zeux/.local/lib/python3.10/site-packages/wandb/sdk/lib/service/service_client.py", line 65, in _send_server_request
29
+ await self._writer.drain()
30
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/asyncio/streams.py", line 371, in drain
31
+ await self._protocol._drain_helper()
32
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/asyncio/streams.py", line 167, in _drain_helper
33
+ raise ConnectionResetError('Connection lost')
34
+ ConnectionResetError: Connection lost
35
+ 2025-12-30 22:18:28,601 ERROR wandb-AsyncioManager-main:2365002 [asyncio_manager.py:fn_wrap_exceptions():183] Uncaught exception in run_soon callback.
36
+ Traceback (most recent call last):
37
+ File "/home/zeux/.local/lib/python3.10/site-packages/wandb/sdk/lib/asyncio_manager.py", line 181, in fn_wrap_exceptions
38
+ await fn()
39
+ File "/home/zeux/.local/lib/python3.10/site-packages/wandb/sdk/lib/service/service_client.py", line 65, in _send_server_request
40
+ await self._writer.drain()
41
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/asyncio/streams.py", line 371, in drain
42
+ await self._protocol._drain_helper()
43
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/asyncio/streams.py", line 167, in _drain_helper
44
+ raise ConnectionResetError('Connection lost')
45
+ ConnectionResetError: Connection lost
46
+ 2025-12-30 22:18:28,601 ERROR wandb-AsyncioManager-main:2365002 [asyncio_manager.py:fn_wrap_exceptions():183] Uncaught exception in run_soon callback.
47
+ Traceback (most recent call last):
48
+ File "/home/zeux/.local/lib/python3.10/site-packages/wandb/sdk/lib/asyncio_manager.py", line 181, in fn_wrap_exceptions
49
+ await fn()
50
+ File "/home/zeux/.local/lib/python3.10/site-packages/wandb/sdk/lib/service/service_client.py", line 65, in _send_server_request
51
+ await self._writer.drain()
52
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/asyncio/streams.py", line 371, in drain
53
+ await self._protocol._drain_helper()
54
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/asyncio/streams.py", line 167, in _drain_helper
55
+ raise ConnectionResetError('Connection lost')
56
+ ConnectionResetError: Connection lost
act_so101_test/wandb/run-20251230_221749-q1o5aaw1/run-q1o5aaw1.wandb ADDED
File without changes
lerobot-act_so101_test/checkpoints/020000/pretrained_model/config.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "act",
3
+ "n_obs_steps": 1,
4
+ "input_features": {
5
+ "observation.state": {
6
+ "type": "STATE",
7
+ "shape": [
8
+ 6
9
+ ]
10
+ },
11
+ "observation.images.top": {
12
+ "type": "VISUAL",
13
+ "shape": [
14
+ 3,
15
+ 480,
16
+ 640
17
+ ]
18
+ },
19
+ "observation.images.side": {
20
+ "type": "VISUAL",
21
+ "shape": [
22
+ 3,
23
+ 480,
24
+ 640
25
+ ]
26
+ }
27
+ },
28
+ "output_features": {
29
+ "action": {
30
+ "type": "ACTION",
31
+ "shape": [
32
+ 6
33
+ ]
34
+ }
35
+ },
36
+ "device": "cuda",
37
+ "use_amp": true,
38
+ "push_to_hub": true,
39
+ "repo_id": "local/lerobot-act-so101-model",
40
+ "private": null,
41
+ "tags": null,
42
+ "license": null,
43
+ "pretrained_path": null,
44
+ "chunk_size": 100,
45
+ "n_action_steps": 100,
46
+ "normalization_mapping": {
47
+ "VISUAL": "MEAN_STD",
48
+ "STATE": "MEAN_STD",
49
+ "ACTION": "MEAN_STD"
50
+ },
51
+ "vision_backbone": "resnet18",
52
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
53
+ "replace_final_stride_with_dilation": false,
54
+ "pre_norm": false,
55
+ "dim_model": 512,
56
+ "n_heads": 8,
57
+ "dim_feedforward": 3200,
58
+ "feedforward_activation": "relu",
59
+ "n_encoder_layers": 4,
60
+ "n_decoder_layers": 1,
61
+ "use_vae": true,
62
+ "latent_dim": 32,
63
+ "n_vae_encoder_layers": 4,
64
+ "temporal_ensemble_coeff": null,
65
+ "dropout": 0.1,
66
+ "kl_weight": 10.0,
67
+ "optimizer_lr": 1e-05,
68
+ "optimizer_weight_decay": 0.0001,
69
+ "optimizer_lr_backbone": 1e-05
70
+ }
lerobot-act_so101_test/checkpoints/020000/pretrained_model/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0173494afd88d26d228b0454fafb43979977942d75c9a380f4211e9f5c641322
3
+ size 206699736
lerobot-act_so101_test/checkpoints/020000/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
+ 6
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
+ }
lerobot-act_so101_test/checkpoints/020000/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:18a892bb501c57ca790c3bd1425b3f9c22828f516040443efdee18efb5894a8b
3
+ size 7520
lerobot-act_so101_test/checkpoints/020000/pretrained_model/policy_preprocessor.json ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ 6
30
+ ]
31
+ },
32
+ "observation.images.top": {
33
+ "type": "VISUAL",
34
+ "shape": [
35
+ 3,
36
+ 480,
37
+ 640
38
+ ]
39
+ },
40
+ "observation.images.side": {
41
+ "type": "VISUAL",
42
+ "shape": [
43
+ 3,
44
+ 480,
45
+ 640
46
+ ]
47
+ },
48
+ "action": {
49
+ "type": "ACTION",
50
+ "shape": [
51
+ 6
52
+ ]
53
+ }
54
+ },
55
+ "norm_map": {
56
+ "VISUAL": "MEAN_STD",
57
+ "STATE": "MEAN_STD",
58
+ "ACTION": "MEAN_STD"
59
+ }
60
+ },
61
+ "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
62
+ }
63
+ ]
64
+ }
lerobot-act_so101_test/checkpoints/020000/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:18a892bb501c57ca790c3bd1425b3f9c22828f516040443efdee18efb5894a8b
3
+ size 7520
lerobot-act_so101_test/checkpoints/020000/pretrained_model/train_config.json ADDED
@@ -0,0 +1,193 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": {
3
+ "repo_id": "lerobot-act",
4
+ "root": "/home/zeux/lerobot-act-dataset",
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": "pyav",
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
+ 6
91
+ ]
92
+ },
93
+ "observation.images.top": {
94
+ "type": "VISUAL",
95
+ "shape": [
96
+ 3,
97
+ 480,
98
+ 640
99
+ ]
100
+ },
101
+ "observation.images.side": {
102
+ "type": "VISUAL",
103
+ "shape": [
104
+ 3,
105
+ 480,
106
+ 640
107
+ ]
108
+ }
109
+ },
110
+ "output_features": {
111
+ "action": {
112
+ "type": "ACTION",
113
+ "shape": [
114
+ 6
115
+ ]
116
+ }
117
+ },
118
+ "device": "cuda",
119
+ "use_amp": true,
120
+ "push_to_hub": true,
121
+ "repo_id": "local/lerobot-act-so101-model",
122
+ "private": null,
123
+ "tags": null,
124
+ "license": null,
125
+ "pretrained_path": null,
126
+ "chunk_size": 100,
127
+ "n_action_steps": 100,
128
+ "normalization_mapping": {
129
+ "VISUAL": "MEAN_STD",
130
+ "STATE": "MEAN_STD",
131
+ "ACTION": "MEAN_STD"
132
+ },
133
+ "vision_backbone": "resnet18",
134
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
135
+ "replace_final_stride_with_dilation": false,
136
+ "pre_norm": false,
137
+ "dim_model": 512,
138
+ "n_heads": 8,
139
+ "dim_feedforward": 3200,
140
+ "feedforward_activation": "relu",
141
+ "n_encoder_layers": 4,
142
+ "n_decoder_layers": 1,
143
+ "use_vae": true,
144
+ "latent_dim": 32,
145
+ "n_vae_encoder_layers": 4,
146
+ "temporal_ensemble_coeff": null,
147
+ "dropout": 0.1,
148
+ "kl_weight": 10.0,
149
+ "optimizer_lr": 1e-05,
150
+ "optimizer_weight_decay": 0.0001,
151
+ "optimizer_lr_backbone": 1e-05
152
+ },
153
+ "output_dir": "outputs/train/lerobot-act_so101_test",
154
+ "job_name": "act",
155
+ "resume": false,
156
+ "seed": 1000,
157
+ "num_workers": 8,
158
+ "batch_size": 64,
159
+ "steps": 35000,
160
+ "eval_freq": 20000,
161
+ "log_freq": 200,
162
+ "save_checkpoint": true,
163
+ "save_freq": 20000,
164
+ "use_policy_training_preset": true,
165
+ "optimizer": {
166
+ "type": "adamw",
167
+ "lr": 1e-05,
168
+ "weight_decay": 0.0001,
169
+ "grad_clip_norm": 10.0,
170
+ "betas": [
171
+ 0.9,
172
+ 0.999
173
+ ],
174
+ "eps": 1e-08
175
+ },
176
+ "scheduler": null,
177
+ "eval": {
178
+ "n_episodes": 50,
179
+ "batch_size": 50,
180
+ "use_async_envs": false
181
+ },
182
+ "wandb": {
183
+ "enable": true,
184
+ "disable_artifact": false,
185
+ "project": "lerobot",
186
+ "entity": null,
187
+ "notes": null,
188
+ "run_id": "8ce8thqh",
189
+ "mode": null
190
+ },
191
+ "checkpoint_path": null,
192
+ "rename_map": {}
193
+ }
lerobot-act_so101_test/checkpoints/020000/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
+ ]
lerobot-act_so101_test/checkpoints/020000/training_state/optimizer_state.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c25e18d5aab89b1b6c01e2844f144f3e39f4b2df1b3584073e5c0618bd517f2f
3
+ size 412817652
lerobot-act_so101_test/checkpoints/020000/training_state/rng_state.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08684fe5df9e737d72881edf41caa4456fc8fa51b57ee7ef8389f6e7b1b0af67
3
+ size 15708
lerobot-act_so101_test/checkpoints/020000/training_state/training_step.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "step": 20000
3
+ }
lerobot-act_so101_test/checkpoints/035000/pretrained_model/config.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "act",
3
+ "n_obs_steps": 1,
4
+ "input_features": {
5
+ "observation.state": {
6
+ "type": "STATE",
7
+ "shape": [
8
+ 6
9
+ ]
10
+ },
11
+ "observation.images.top": {
12
+ "type": "VISUAL",
13
+ "shape": [
14
+ 3,
15
+ 480,
16
+ 640
17
+ ]
18
+ },
19
+ "observation.images.side": {
20
+ "type": "VISUAL",
21
+ "shape": [
22
+ 3,
23
+ 480,
24
+ 640
25
+ ]
26
+ }
27
+ },
28
+ "output_features": {
29
+ "action": {
30
+ "type": "ACTION",
31
+ "shape": [
32
+ 6
33
+ ]
34
+ }
35
+ },
36
+ "device": "cuda",
37
+ "use_amp": true,
38
+ "push_to_hub": true,
39
+ "repo_id": "local/lerobot-act-so101-model",
40
+ "private": null,
41
+ "tags": null,
42
+ "license": null,
43
+ "pretrained_path": null,
44
+ "chunk_size": 100,
45
+ "n_action_steps": 100,
46
+ "normalization_mapping": {
47
+ "VISUAL": "MEAN_STD",
48
+ "STATE": "MEAN_STD",
49
+ "ACTION": "MEAN_STD"
50
+ },
51
+ "vision_backbone": "resnet18",
52
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
53
+ "replace_final_stride_with_dilation": false,
54
+ "pre_norm": false,
55
+ "dim_model": 512,
56
+ "n_heads": 8,
57
+ "dim_feedforward": 3200,
58
+ "feedforward_activation": "relu",
59
+ "n_encoder_layers": 4,
60
+ "n_decoder_layers": 1,
61
+ "use_vae": true,
62
+ "latent_dim": 32,
63
+ "n_vae_encoder_layers": 4,
64
+ "temporal_ensemble_coeff": null,
65
+ "dropout": 0.1,
66
+ "kl_weight": 10.0,
67
+ "optimizer_lr": 1e-05,
68
+ "optimizer_weight_decay": 0.0001,
69
+ "optimizer_lr_backbone": 1e-05
70
+ }
lerobot-act_so101_test/checkpoints/035000/pretrained_model/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:05305bec1095fe152694317984d4623935dcf9a4b8c0bbea3fa5244c02af351a
3
+ size 206699736
lerobot-act_so101_test/checkpoints/035000/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
+ 6
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
+ }
lerobot-act_so101_test/checkpoints/035000/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:18a892bb501c57ca790c3bd1425b3f9c22828f516040443efdee18efb5894a8b
3
+ size 7520
lerobot-act_so101_test/checkpoints/035000/pretrained_model/policy_preprocessor.json ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ 6
30
+ ]
31
+ },
32
+ "observation.images.top": {
33
+ "type": "VISUAL",
34
+ "shape": [
35
+ 3,
36
+ 480,
37
+ 640
38
+ ]
39
+ },
40
+ "observation.images.side": {
41
+ "type": "VISUAL",
42
+ "shape": [
43
+ 3,
44
+ 480,
45
+ 640
46
+ ]
47
+ },
48
+ "action": {
49
+ "type": "ACTION",
50
+ "shape": [
51
+ 6
52
+ ]
53
+ }
54
+ },
55
+ "norm_map": {
56
+ "VISUAL": "MEAN_STD",
57
+ "STATE": "MEAN_STD",
58
+ "ACTION": "MEAN_STD"
59
+ }
60
+ },
61
+ "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
62
+ }
63
+ ]
64
+ }
lerobot-act_so101_test/checkpoints/035000/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:18a892bb501c57ca790c3bd1425b3f9c22828f516040443efdee18efb5894a8b
3
+ size 7520
lerobot-act_so101_test/checkpoints/035000/pretrained_model/train_config.json ADDED
@@ -0,0 +1,193 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": {
3
+ "repo_id": "lerobot-act",
4
+ "root": "/home/zeux/lerobot-act-dataset",
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": "pyav",
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
+ 6
91
+ ]
92
+ },
93
+ "observation.images.top": {
94
+ "type": "VISUAL",
95
+ "shape": [
96
+ 3,
97
+ 480,
98
+ 640
99
+ ]
100
+ },
101
+ "observation.images.side": {
102
+ "type": "VISUAL",
103
+ "shape": [
104
+ 3,
105
+ 480,
106
+ 640
107
+ ]
108
+ }
109
+ },
110
+ "output_features": {
111
+ "action": {
112
+ "type": "ACTION",
113
+ "shape": [
114
+ 6
115
+ ]
116
+ }
117
+ },
118
+ "device": "cuda",
119
+ "use_amp": true,
120
+ "push_to_hub": true,
121
+ "repo_id": "local/lerobot-act-so101-model",
122
+ "private": null,
123
+ "tags": null,
124
+ "license": null,
125
+ "pretrained_path": null,
126
+ "chunk_size": 100,
127
+ "n_action_steps": 100,
128
+ "normalization_mapping": {
129
+ "VISUAL": "MEAN_STD",
130
+ "STATE": "MEAN_STD",
131
+ "ACTION": "MEAN_STD"
132
+ },
133
+ "vision_backbone": "resnet18",
134
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
135
+ "replace_final_stride_with_dilation": false,
136
+ "pre_norm": false,
137
+ "dim_model": 512,
138
+ "n_heads": 8,
139
+ "dim_feedforward": 3200,
140
+ "feedforward_activation": "relu",
141
+ "n_encoder_layers": 4,
142
+ "n_decoder_layers": 1,
143
+ "use_vae": true,
144
+ "latent_dim": 32,
145
+ "n_vae_encoder_layers": 4,
146
+ "temporal_ensemble_coeff": null,
147
+ "dropout": 0.1,
148
+ "kl_weight": 10.0,
149
+ "optimizer_lr": 1e-05,
150
+ "optimizer_weight_decay": 0.0001,
151
+ "optimizer_lr_backbone": 1e-05
152
+ },
153
+ "output_dir": "outputs/train/lerobot-act_so101_test",
154
+ "job_name": "act",
155
+ "resume": false,
156
+ "seed": 1000,
157
+ "num_workers": 8,
158
+ "batch_size": 64,
159
+ "steps": 35000,
160
+ "eval_freq": 20000,
161
+ "log_freq": 200,
162
+ "save_checkpoint": true,
163
+ "save_freq": 20000,
164
+ "use_policy_training_preset": true,
165
+ "optimizer": {
166
+ "type": "adamw",
167
+ "lr": 1e-05,
168
+ "weight_decay": 0.0001,
169
+ "grad_clip_norm": 10.0,
170
+ "betas": [
171
+ 0.9,
172
+ 0.999
173
+ ],
174
+ "eps": 1e-08
175
+ },
176
+ "scheduler": null,
177
+ "eval": {
178
+ "n_episodes": 50,
179
+ "batch_size": 50,
180
+ "use_async_envs": false
181
+ },
182
+ "wandb": {
183
+ "enable": true,
184
+ "disable_artifact": false,
185
+ "project": "lerobot",
186
+ "entity": null,
187
+ "notes": null,
188
+ "run_id": "8ce8thqh",
189
+ "mode": null
190
+ },
191
+ "checkpoint_path": null,
192
+ "rename_map": {}
193
+ }
lerobot-act_so101_test/checkpoints/035000/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
+ ]
lerobot-act_so101_test/checkpoints/035000/training_state/optimizer_state.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7b07539463c39fa59f0c98b000c889050575e06d91b79471d458d836a36e7e71
3
+ size 412817652
lerobot-act_so101_test/checkpoints/035000/training_state/rng_state.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb2fd6abc98e0392a9d887f83cba768640011f91bf678d7c01a644c40ebae61a
3
+ size 15708
lerobot-act_so101_test/checkpoints/035000/training_state/training_step.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "step": 35000
3
+ }
lerobot-act_so101_test/wandb/debug-internal.log ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2025-12-31T13:18:54.069787725+05:30","level":"INFO","msg":"stream: starting","core version":"0.21.4"}
2
+ {"time":"2025-12-31T13:18:54.476490438+05:30","level":"INFO","msg":"stream: created new stream","id":"8ce8thqh"}
3
+ {"time":"2025-12-31T13:18:54.476511345+05:30","level":"INFO","msg":"stream: started","id":"8ce8thqh"}
4
+ {"time":"2025-12-31T13:18:54.476515165+05:30","level":"INFO","msg":"writer: started","stream_id":"8ce8thqh"}
5
+ {"time":"2025-12-31T13:18:54.476566118+05:30","level":"INFO","msg":"handler: started","stream_id":"8ce8thqh"}
6
+ {"time":"2025-12-31T13:18:54.476575461+05:30","level":"INFO","msg":"sender: started","stream_id":"8ce8thqh"}
7
+ {"time":"2025-12-31T16:59:00.732388154+05:30","level":"INFO","msg":"stream: closing","id":"8ce8thqh"}
8
+ {"time":"2025-12-31T16:59:23.719001417+05:30","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
9
+ {"time":"2025-12-31T16:59:24.097369588+05:30","level":"INFO","msg":"handler: closed","stream_id":"8ce8thqh"}
10
+ {"time":"2025-12-31T16:59:24.097416229+05:30","level":"INFO","msg":"sender: closed","stream_id":"8ce8thqh"}
11
+ {"time":"2025-12-31T16:59:24.097422049+05:30","level":"INFO","msg":"stream: closed","id":"8ce8thqh"}
lerobot-act_so101_test/wandb/debug.log ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_setup.py:_flush():81] Current SDK version is 0.21.4
2
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_setup.py:_flush():81] Configure stats pid to 1804411
3
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_setup.py:_flush():81] Loading settings from /home/zeux/.config/wandb/settings
4
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_setup.py:_flush():81] Loading settings from /home/zeux/wandb/settings
5
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_setup.py:_flush():81] Loading settings from environment variables
6
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_init.py:setup_run_log_directory():686] Logging user logs to outputs/train/lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/logs/debug.log
7
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_init.py:setup_run_log_directory():687] Logging internal logs to outputs/train/lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/logs/debug-internal.log
8
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_init.py:init():813] calling init triggers
9
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_init.py:init():818] wandb.init called with sweep_config: {}
10
+ config: {'dataset': {'repo_id': 'lerobot-act', 'root': '/home/zeux/lerobot-act-dataset', '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': 'pyav', 'streaming': False}, 'env': None, 'policy': {'type': 'act', 'n_obs_steps': 1, 'input_features': {}, 'output_features': {}, 'device': 'cuda', 'use_amp': True, 'push_to_hub': True, 'repo_id': 'local/lerobot-act-so101-model', '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/lerobot-act_so101_test', 'job_name': 'act', 'resume': False, 'seed': 1000, 'num_workers': 8, 'batch_size': 64, 'steps': 35000, '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-31 13:18:53,862 INFO MainThread:1804411 [wandb_init.py:init():854] starting backend
12
+ 2025-12-31 13:18:54,065 INFO MainThread:1804411 [wandb_init.py:init():857] sending inform_init request
13
+ 2025-12-31 13:18:54,067 INFO MainThread:1804411 [wandb_init.py:init():865] backend started and connected
14
+ 2025-12-31 13:18:54,068 INFO MainThread:1804411 [wandb_init.py:init():936] updated telemetry
15
+ 2025-12-31 13:18:54,069 INFO MainThread:1804411 [wandb_init.py:init():960] communicating run to backend with 90.0 second timeout
16
+ 2025-12-31 13:18:54,935 INFO MainThread:1804411 [wandb_init.py:init():1011] starting run threads in backend
17
+ 2025-12-31 13:18:54,995 INFO MainThread:1804411 [wandb_run.py:_console_start():2506] atexit reg
18
+ 2025-12-31 13:18:54,995 INFO MainThread:1804411 [wandb_run.py:_redirect():2354] redirect: wrap_raw
19
+ 2025-12-31 13:18:54,995 INFO MainThread:1804411 [wandb_run.py:_redirect():2423] Wrapping output streams.
20
+ 2025-12-31 13:18:54,995 INFO MainThread:1804411 [wandb_run.py:_redirect():2446] Redirects installed.
21
+ 2025-12-31 13:18:54,996 INFO MainThread:1804411 [wandb_init.py:init():1049] run started, returning control to user process
22
+ 2025-12-31 16:59:00,732 INFO wandb-AsyncioManager-main:1804411 [service_client.py:_forward_responses():84] Reached EOF.
23
+ 2025-12-31 16:59:00,732 INFO wandb-AsyncioManager-main:1804411 [mailbox.py:close():137] Closing mailbox, abandoning 2 handles.
lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/files/config.yaml ADDED
@@ -0,0 +1,227 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _wandb:
2
+ value:
3
+ cli_version: 0.21.4
4
+ e:
5
+ nxltchl4gaduvozis0gz9xveosym3k7a:
6
+ args:
7
+ - --dataset.repo_id=lerobot-act
8
+ - --dataset.root=/home/zeux/lerobot-act-dataset
9
+ - --dataset.video_backend=pyav
10
+ - --policy.type=act
11
+ - --policy.repo_id=local/lerobot-act-so101-model
12
+ - --output_dir=outputs/train/lerobot-act_so101_test
13
+ - --policy.device=cuda
14
+ - --policy.use_amp=true
15
+ - --batch_size=64
16
+ - --num_workers=8
17
+ - --steps=35000
18
+ - --wandb.enable=true
19
+ - --policy.push_to_hub=true
20
+ codePath: miniconda3/envs/lerobot/bin/lerobot-train
21
+ codePathLocal: miniconda3/envs/lerobot/bin/lerobot-train
22
+ cpu_count: 24
23
+ cpu_count_logical: 24
24
+ cudaVersion: "13.0"
25
+ disk:
26
+ /:
27
+ total: "200449576960"
28
+ used: "34473017344"
29
+ executable: /home/zeux/miniconda3/envs/lerobot/bin/python3.10
30
+ gpu: NVIDIA RTX PRO 6000 Blackwell Workstation Edition
31
+ gpu_count: 1
32
+ gpu_nvidia:
33
+ - architecture: Blackwell
34
+ cudaCores: 24064
35
+ memoryTotal: "102641958912"
36
+ name: NVIDIA RTX PRO 6000 Blackwell Workstation Edition
37
+ uuid: GPU-8c66f3c4-00c8-00a6-e5ff-633c5558ec8e
38
+ host: zeux
39
+ memory:
40
+ total: "134489886720"
41
+ os: Linux-6.8.0-90-generic-x86_64-with-glibc2.35
42
+ program: /home/zeux/miniconda3/envs/lerobot/bin/lerobot-train
43
+ python: CPython 3.10.19
44
+ root: outputs/train/lerobot-act_so101_test
45
+ startedAt: "2025-12-31T07:48:53.840056Z"
46
+ writerId: nxltchl4gaduvozis0gz9xveosym3k7a
47
+ m: []
48
+ python_version: 3.10.19
49
+ t:
50
+ "1":
51
+ - 1
52
+ - 11
53
+ - 41
54
+ - 49
55
+ - 51
56
+ - 71
57
+ - 83
58
+ "2":
59
+ - 1
60
+ - 11
61
+ - 41
62
+ - 49
63
+ - 51
64
+ - 71
65
+ - 83
66
+ "3":
67
+ - 13
68
+ - 15
69
+ - 16
70
+ - 61
71
+ "4": 3.10.19
72
+ "5": 0.21.4
73
+ "6": 4.57.3
74
+ "10":
75
+ - 21
76
+ "12": 0.21.4
77
+ "13": linux-x86_64
78
+ batch_size:
79
+ value: 64
80
+ checkpoint_path:
81
+ value: null
82
+ dataset:
83
+ value:
84
+ episodes: null
85
+ image_transforms:
86
+ enable: false
87
+ max_num_transforms: 3
88
+ random_order: false
89
+ tfs:
90
+ affine:
91
+ kwargs:
92
+ degrees:
93
+ - -5
94
+ - 5
95
+ translate:
96
+ - 0.05
97
+ - 0.05
98
+ type: RandomAffine
99
+ weight: 1
100
+ brightness:
101
+ kwargs:
102
+ brightness:
103
+ - 0.8
104
+ - 1.2
105
+ type: ColorJitter
106
+ weight: 1
107
+ contrast:
108
+ kwargs:
109
+ contrast:
110
+ - 0.8
111
+ - 1.2
112
+ type: ColorJitter
113
+ weight: 1
114
+ hue:
115
+ kwargs:
116
+ hue:
117
+ - -0.05
118
+ - 0.05
119
+ type: ColorJitter
120
+ weight: 1
121
+ saturation:
122
+ kwargs:
123
+ saturation:
124
+ - 0.5
125
+ - 1.5
126
+ type: ColorJitter
127
+ weight: 1
128
+ sharpness:
129
+ kwargs:
130
+ sharpness:
131
+ - 0.5
132
+ - 1.5
133
+ type: SharpnessJitter
134
+ weight: 1
135
+ repo_id: lerobot-act
136
+ revision: null
137
+ root: /home/zeux/lerobot-act-dataset
138
+ streaming: false
139
+ use_imagenet_stats: true
140
+ video_backend: pyav
141
+ env:
142
+ value: null
143
+ eval:
144
+ value:
145
+ batch_size: 50
146
+ n_episodes: 50
147
+ use_async_envs: false
148
+ eval_freq:
149
+ value: 20000
150
+ job_name:
151
+ value: act
152
+ log_freq:
153
+ value: 200
154
+ num_workers:
155
+ value: 8
156
+ optimizer:
157
+ value:
158
+ betas:
159
+ - 0.9
160
+ - 0.999
161
+ eps: 1e-08
162
+ grad_clip_norm: 10
163
+ lr: 1e-05
164
+ type: adamw
165
+ weight_decay: 0.0001
166
+ output_dir:
167
+ value: outputs/train/lerobot-act_so101_test
168
+ policy:
169
+ value:
170
+ chunk_size: 100
171
+ device: cuda
172
+ dim_feedforward: 3200
173
+ dim_model: 512
174
+ dropout: 0.1
175
+ feedforward_activation: relu
176
+ kl_weight: 10
177
+ latent_dim: 32
178
+ license: null
179
+ n_action_steps: 100
180
+ n_decoder_layers: 1
181
+ n_encoder_layers: 4
182
+ n_heads: 8
183
+ n_obs_steps: 1
184
+ n_vae_encoder_layers: 4
185
+ normalization_mapping:
186
+ ACTION: MEAN_STD
187
+ STATE: MEAN_STD
188
+ VISUAL: MEAN_STD
189
+ optimizer_lr: 1e-05
190
+ optimizer_lr_backbone: 1e-05
191
+ optimizer_weight_decay: 0.0001
192
+ pre_norm: false
193
+ pretrained_backbone_weights: ResNet18_Weights.IMAGENET1K_V1
194
+ pretrained_path: null
195
+ private: null
196
+ push_to_hub: true
197
+ replace_final_stride_with_dilation: false
198
+ repo_id: local/lerobot-act-so101-model
199
+ tags: null
200
+ temporal_ensemble_coeff: null
201
+ type: act
202
+ use_amp: true
203
+ use_vae: true
204
+ vision_backbone: resnet18
205
+ resume:
206
+ value: false
207
+ save_checkpoint:
208
+ value: true
209
+ save_freq:
210
+ value: 20000
211
+ scheduler:
212
+ value: null
213
+ seed:
214
+ value: 1000
215
+ steps:
216
+ value: 35000
217
+ use_policy_training_preset:
218
+ value: true
219
+ wandb:
220
+ value:
221
+ disable_artifact: false
222
+ enable: true
223
+ entity: null
224
+ mode: null
225
+ notes: null
226
+ project: lerobot
227
+ run_id: null
lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/files/output.log ADDED
@@ -0,0 +1,257 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ INFO 2025-12-31 13:18:54 db_utils.py:102 Logs will be synced with wandb.
2
+ INFO 2025-12-31 13:18:54 db_utils.py:103 Track this run --> https://wandb.ai/nwaves-ai-nwaves/lerobot/runs/8ce8thqh
3
+ INFO 2025-12-31 13:18:54 ot_train.py:183 Creating dataset
4
+ INFO 2025-12-31 13:18:55 ot_train.py:202 Creating policy
5
+ INFO 2025-12-31 13:18:55 ot_train.py:247 Creating optimizer and scheduler
6
+ INFO 2025-12-31 13:18:55 ot_train.py:259 Output dir: outputs/train/lerobot-act_so101_test
7
+ INFO 2025-12-31 13:18:55 ot_train.py:264 cfg.steps=35000 (35K)
8
+ INFO 2025-12-31 13:18:55 ot_train.py:265 dataset.num_frames=35618 (36K)
9
+ INFO 2025-12-31 13:18:55 ot_train.py:266 dataset.num_episodes=40
10
+ INFO 2025-12-31 13:18:55 ot_train.py:269 Effective batch size: 64 x 1 = 64
11
+ INFO 2025-12-31 13:18:55 ot_train.py:270 num_learnable_params=51597190 (52M)
12
+ INFO 2025-12-31 13:18:55 ot_train.py:271 num_total_params=51597190 (52M)
13
+ INFO 2025-12-31 13:18:55 ot_train.py:327 Start offline training on a fixed dataset
14
+ INFO 2025-12-31 13:20:16 ot_train.py:354 step:200 smpl:13K ep:14 epch:0.36 loss:5.354 grdn:79.461 lr:1.0e-05 updt_s:0.366 data_s:0.037
15
+ INFO 2025-12-31 13:21:29 ot_train.py:354 step:400 smpl:26K ep:29 epch:0.72 loss:2.042 grdn:35.399 lr:1.0e-05 updt_s:0.351 data_s:0.015
16
+ INFO 2025-12-31 13:22:48 ot_train.py:354 step:600 smpl:38K ep:43 epch:1.08 loss:1.576 grdn:32.378 lr:1.0e-05 updt_s:0.362 data_s:0.031
17
+ INFO 2025-12-31 13:24:02 ot_train.py:354 step:800 smpl:51K ep:57 epch:1.44 loss:1.250 grdn:28.865 lr:1.0e-05 updt_s:0.355 data_s:0.015
18
+ INFO 2025-12-31 13:25:16 ot_train.py:354 step:1K smpl:64K ep:72 epch:1.80 loss:1.008 grdn:26.938 lr:1.0e-05 updt_s:0.356 data_s:0.015
19
+ INFO 2025-12-31 13:26:33 ot_train.py:354 step:1K smpl:77K ep:86 epch:2.16 loss:0.811 grdn:24.199 lr:1.0e-05 updt_s:0.354 data_s:0.031
20
+ INFO 2025-12-31 13:27:48 ot_train.py:354 step:1K smpl:90K ep:101 epch:2.52 loss:0.661 grdn:22.866 lr:1.0e-05 updt_s:0.356 data_s:0.015
21
+ INFO 2025-12-31 13:29:02 ot_train.py:354 step:2K smpl:102K ep:115 epch:2.87 loss:0.540 grdn:20.452 lr:1.0e-05 updt_s:0.356 data_s:0.015
22
+ INFO 2025-12-31 13:30:19 ot_train.py:354 step:2K smpl:115K ep:129 epch:3.23 loss:0.450 grdn:18.978 lr:1.0e-05 updt_s:0.354 data_s:0.032
23
+ INFO 2025-12-31 13:31:34 ot_train.py:354 step:2K smpl:128K ep:144 epch:3.59 loss:0.380 grdn:17.760 lr:1.0e-05 updt_s:0.356 data_s:0.015
24
+ INFO 2025-12-31 13:32:48 ot_train.py:354 step:2K smpl:141K ep:158 epch:3.95 loss:0.331 grdn:17.224 lr:1.0e-05 updt_s:0.356 data_s:0.015
25
+ INFO 2025-12-31 13:34:05 ot_train.py:354 step:2K smpl:154K ep:172 epch:4.31 loss:0.297 grdn:16.458 lr:1.0e-05 updt_s:0.354 data_s:0.028
26
+ INFO 2025-12-31 13:35:19 ot_train.py:354 step:3K smpl:166K ep:187 epch:4.67 loss:0.268 grdn:16.033 lr:1.0e-05 updt_s:0.356 data_s:0.015
27
+ INFO 2025-12-31 13:36:36 ot_train.py:354 step:3K smpl:179K ep:201 epch:5.03 loss:0.248 grdn:15.266 lr:1.0e-05 updt_s:0.355 data_s:0.032
28
+ INFO 2025-12-31 13:37:51 ot_train.py:354 step:3K smpl:192K ep:216 epch:5.39 loss:0.232 grdn:15.217 lr:1.0e-05 updt_s:0.356 data_s:0.017
29
+ INFO 2025-12-31 13:39:06 ot_train.py:354 step:3K smpl:205K ep:230 epch:5.75 loss:0.216 grdn:14.124 lr:1.0e-05 updt_s:0.356 data_s:0.016
30
+ INFO 2025-12-31 13:40:23 ot_train.py:354 step:3K smpl:218K ep:244 epch:6.11 loss:0.207 grdn:13.907 lr:1.0e-05 updt_s:0.354 data_s:0.034
31
+ INFO 2025-12-31 13:41:38 ot_train.py:354 step:4K smpl:230K ep:259 epch:6.47 loss:0.191 grdn:13.139 lr:1.0e-05 updt_s:0.357 data_s:0.015
32
+ INFO 2025-12-31 13:42:52 ot_train.py:354 step:4K smpl:243K ep:273 epch:6.83 loss:0.187 grdn:13.581 lr:1.0e-05 updt_s:0.355 data_s:0.015
33
+ INFO 2025-12-31 13:44:10 ot_train.py:354 step:4K smpl:256K ep:287 epch:7.19 loss:0.178 grdn:12.612 lr:1.0e-05 updt_s:0.354 data_s:0.034
34
+ INFO 2025-12-31 13:45:25 ot_train.py:354 step:4K smpl:269K ep:302 epch:7.55 loss:0.170 grdn:12.367 lr:1.0e-05 updt_s:0.357 data_s:0.017
35
+ INFO 2025-12-31 13:46:39 ot_train.py:354 step:4K smpl:282K ep:316 epch:7.91 loss:0.163 grdn:12.463 lr:1.0e-05 updt_s:0.355 data_s:0.015
36
+ INFO 2025-12-31 13:47:57 ot_train.py:354 step:5K smpl:294K ep:331 epch:8.27 loss:0.158 grdn:12.824 lr:1.0e-05 updt_s:0.355 data_s:0.033
37
+ INFO 2025-12-31 13:49:11 ot_train.py:354 step:5K smpl:307K ep:345 epch:8.62 loss:0.154 grdn:11.798 lr:1.0e-05 updt_s:0.357 data_s:0.016
38
+ INFO 2025-12-31 13:50:25 ot_train.py:354 step:5K smpl:320K ep:359 epch:8.98 loss:0.148 grdn:12.203 lr:1.0e-05 updt_s:0.355 data_s:0.015
39
+ INFO 2025-12-31 13:51:43 ot_train.py:354 step:5K smpl:333K ep:374 epch:9.34 loss:0.143 grdn:11.829 lr:1.0e-05 updt_s:0.355 data_s:0.032
40
+ INFO 2025-12-31 13:52:57 ot_train.py:354 step:5K smpl:346K ep:388 epch:9.70 loss:0.141 grdn:12.040 lr:1.0e-05 updt_s:0.356 data_s:0.015
41
+ INFO 2025-12-31 13:54:14 ot_train.py:354 step:6K smpl:358K ep:402 epch:10.06 loss:0.133 grdn:11.093 lr:1.0e-05 updt_s:0.354 data_s:0.032
42
+ INFO 2025-12-31 13:55:29 ot_train.py:354 step:6K smpl:371K ep:417 epch:10.42 loss:0.130 grdn:10.938 lr:1.0e-05 updt_s:0.356 data_s:0.015
43
+ INFO 2025-12-31 13:56:43 ot_train.py:354 step:6K smpl:384K ep:431 epch:10.78 loss:0.130 grdn:11.160 lr:1.0e-05 updt_s:0.356 data_s:0.015
44
+ INFO 2025-12-31 13:58:00 ot_train.py:354 step:6K smpl:397K ep:446 epch:11.14 loss:0.126 grdn:10.801 lr:1.0e-05 updt_s:0.354 data_s:0.030
45
+ INFO 2025-12-31 13:59:14 ot_train.py:354 step:6K smpl:410K ep:460 epch:11.50 loss:0.121 grdn:10.905 lr:1.0e-05 updt_s:0.356 data_s:0.015
46
+ INFO 2025-12-31 14:00:29 ot_train.py:354 step:7K smpl:422K ep:474 epch:11.86 loss:0.120 grdn:10.922 lr:1.0e-05 updt_s:0.356 data_s:0.015
47
+ INFO 2025-12-31 14:01:46 ot_train.py:354 step:7K smpl:435K ep:489 epch:12.22 loss:0.116 grdn:9.954 lr:1.0e-05 updt_s:0.354 data_s:0.032
48
+ INFO 2025-12-31 14:03:00 ot_train.py:354 step:7K smpl:448K ep:503 epch:12.58 loss:0.113 grdn:9.952 lr:1.0e-05 updt_s:0.356 data_s:0.015
49
+ INFO 2025-12-31 14:04:15 ot_train.py:354 step:7K smpl:461K ep:517 epch:12.94 loss:0.111 grdn:10.116 lr:1.0e-05 updt_s:0.356 data_s:0.015
50
+ INFO 2025-12-31 14:05:32 ot_train.py:354 step:7K smpl:474K ep:532 epch:13.30 loss:0.110 grdn:10.374 lr:1.0e-05 updt_s:0.354 data_s:0.031
51
+ INFO 2025-12-31 14:06:46 ot_train.py:354 step:8K smpl:486K ep:546 epch:13.66 loss:0.107 grdn:10.180 lr:1.0e-05 updt_s:0.356 data_s:0.015
52
+ INFO 2025-12-31 14:08:03 ot_train.py:354 step:8K smpl:499K ep:561 epch:14.02 loss:0.106 grdn:10.077 lr:1.0e-05 updt_s:0.355 data_s:0.032
53
+ INFO 2025-12-31 14:09:18 ot_train.py:354 step:8K smpl:512K ep:575 epch:14.37 loss:0.102 grdn:9.894 lr:1.0e-05 updt_s:0.356 data_s:0.015
54
+ INFO 2025-12-31 14:10:32 ot_train.py:354 step:8K smpl:525K ep:589 epch:14.73 loss:0.101 grdn:10.146 lr:1.0e-05 updt_s:0.356 data_s:0.015
55
+ INFO 2025-12-31 14:11:49 ot_train.py:354 step:8K smpl:538K ep:604 epch:15.09 loss:0.100 grdn:9.853 lr:1.0e-05 updt_s:0.354 data_s:0.032
56
+ INFO 2025-12-31 14:13:04 ot_train.py:354 step:9K smpl:550K ep:618 epch:15.45 loss:0.097 grdn:9.097 lr:1.0e-05 updt_s:0.357 data_s:0.015
57
+ INFO 2025-12-31 14:14:18 ot_train.py:354 step:9K smpl:563K ep:632 epch:15.81 loss:0.098 grdn:9.567 lr:1.0e-05 updt_s:0.356 data_s:0.015
58
+ INFO 2025-12-31 14:15:35 ot_train.py:354 step:9K smpl:576K ep:647 epch:16.17 loss:0.094 grdn:9.654 lr:1.0e-05 updt_s:0.354 data_s:0.032
59
+ INFO 2025-12-31 14:16:50 ot_train.py:354 step:9K smpl:589K ep:661 epch:16.53 loss:0.095 grdn:9.516 lr:1.0e-05 updt_s:0.357 data_s:0.015
60
+ INFO 2025-12-31 14:18:04 ot_train.py:354 step:9K smpl:602K ep:676 epch:16.89 loss:0.092 grdn:9.076 lr:1.0e-05 updt_s:0.355 data_s:0.015
61
+ INFO 2025-12-31 14:19:21 ot_train.py:354 step:10K smpl:614K ep:690 epch:17.25 loss:0.091 grdn:8.884 lr:1.0e-05 updt_s:0.354 data_s:0.031
62
+ INFO 2025-12-31 14:20:36 ot_train.py:354 step:10K smpl:627K ep:704 epch:17.61 loss:0.090 grdn:8.930 lr:1.0e-05 updt_s:0.356 data_s:0.015
63
+ INFO 2025-12-31 14:21:50 ot_train.py:354 step:10K smpl:640K ep:719 epch:17.97 loss:0.086 grdn:8.885 lr:1.0e-05 updt_s:0.356 data_s:0.015
64
+ INFO 2025-12-31 14:23:07 ot_train.py:354 step:10K smpl:653K ep:733 epch:18.33 loss:0.088 grdn:9.240 lr:1.0e-05 updt_s:0.355 data_s:0.033
65
+ INFO 2025-12-31 14:24:22 ot_train.py:354 step:10K smpl:666K ep:747 epch:18.69 loss:0.085 grdn:8.758 lr:1.0e-05 updt_s:0.356 data_s:0.015
66
+ INFO 2025-12-31 14:25:39 ot_train.py:354 step:11K smpl:678K ep:762 epch:19.05 loss:0.085 grdn:8.849 lr:1.0e-05 updt_s:0.355 data_s:0.030
67
+ INFO 2025-12-31 14:26:53 ot_train.py:354 step:11K smpl:691K ep:776 epch:19.41 loss:0.083 grdn:8.831 lr:1.0e-05 updt_s:0.356 data_s:0.015
68
+ INFO 2025-12-31 14:28:07 ot_train.py:354 step:11K smpl:704K ep:791 epch:19.77 loss:0.082 grdn:8.277 lr:1.0e-05 updt_s:0.356 data_s:0.015
69
+ INFO 2025-12-31 14:29:25 ot_train.py:354 step:11K smpl:717K ep:805 epch:20.12 loss:0.082 grdn:8.938 lr:1.0e-05 updt_s:0.355 data_s:0.031
70
+ INFO 2025-12-31 14:30:39 ot_train.py:354 step:11K smpl:730K ep:819 epch:20.48 loss:0.081 grdn:8.285 lr:1.0e-05 updt_s:0.356 data_s:0.015
71
+ INFO 2025-12-31 14:31:53 ot_train.py:354 step:12K smpl:742K ep:834 epch:20.84 loss:0.079 grdn:8.153 lr:1.0e-05 updt_s:0.356 data_s:0.015
72
+ INFO 2025-12-31 14:33:10 ot_train.py:354 step:12K smpl:755K ep:848 epch:21.20 loss:0.079 grdn:8.469 lr:1.0e-05 updt_s:0.355 data_s:0.030
73
+ INFO 2025-12-31 14:34:25 ot_train.py:354 step:12K smpl:768K ep:862 epch:21.56 loss:0.077 grdn:9.393 lr:1.0e-05 updt_s:0.356 data_s:0.015
74
+ INFO 2025-12-31 14:35:39 ot_train.py:354 step:12K smpl:781K ep:877 epch:21.92 loss:0.077 grdn:8.871 lr:1.0e-05 updt_s:0.356 data_s:0.015
75
+ INFO 2025-12-31 14:36:56 ot_train.py:354 step:12K smpl:794K ep:891 epch:22.28 loss:0.076 grdn:7.807 lr:1.0e-05 updt_s:0.354 data_s:0.031
76
+ INFO 2025-12-31 14:38:10 ot_train.py:354 step:13K smpl:806K ep:906 epch:22.64 loss:0.075 grdn:7.986 lr:1.0e-05 updt_s:0.357 data_s:0.015
77
+ INFO 2025-12-31 14:39:24 ot_train.py:354 step:13K smpl:819K ep:920 epch:23.00 loss:0.076 grdn:8.523 lr:1.0e-05 updt_s:0.356 data_s:0.015
78
+ INFO 2025-12-31 14:40:41 ot_train.py:354 step:13K smpl:832K ep:934 epch:23.36 loss:0.074 grdn:7.770 lr:1.0e-05 updt_s:0.355 data_s:0.029
79
+ INFO 2025-12-31 14:41:56 ot_train.py:354 step:13K smpl:845K ep:949 epch:23.72 loss:0.075 grdn:8.904 lr:1.0e-05 updt_s:0.357 data_s:0.015
80
+ INFO 2025-12-31 14:43:13 ot_train.py:354 step:13K smpl:858K ep:963 epch:24.08 loss:0.072 grdn:8.099 lr:1.0e-05 updt_s:0.355 data_s:0.030
81
+ INFO 2025-12-31 14:44:27 ot_train.py:354 step:14K smpl:870K ep:977 epch:24.44 loss:0.072 grdn:7.648 lr:1.0e-05 updt_s:0.356 data_s:0.015
82
+ INFO 2025-12-31 14:45:41 ot_train.py:354 step:14K smpl:883K ep:992 epch:24.80 loss:0.073 grdn:8.119 lr:1.0e-05 updt_s:0.357 data_s:0.015
83
+ INFO 2025-12-31 14:46:59 ot_train.py:354 step:14K smpl:896K ep:1K epch:25.16 loss:0.072 grdn:7.931 lr:1.0e-05 updt_s:0.354 data_s:0.031
84
+ INFO 2025-12-31 14:48:13 ot_train.py:354 step:14K smpl:909K ep:1K epch:25.52 loss:0.069 grdn:7.301 lr:1.0e-05 updt_s:0.356 data_s:0.015
85
+ INFO 2025-12-31 14:49:27 ot_train.py:354 step:14K smpl:922K ep:1K epch:25.87 loss:0.069 grdn:7.827 lr:1.0e-05 updt_s:0.356 data_s:0.015
86
+ INFO 2025-12-31 14:50:44 ot_train.py:354 step:15K smpl:934K ep:1K epch:26.23 loss:0.070 grdn:8.804 lr:1.0e-05 updt_s:0.354 data_s:0.030
87
+ INFO 2025-12-31 14:51:58 ot_train.py:354 step:15K smpl:947K ep:1K epch:26.59 loss:0.068 grdn:7.208 lr:1.0e-05 updt_s:0.356 data_s:0.015
88
+ INFO 2025-12-31 14:53:13 ot_train.py:354 step:15K smpl:960K ep:1K epch:26.95 loss:0.066 grdn:6.920 lr:1.0e-05 updt_s:0.356 data_s:0.015
89
+ INFO 2025-12-31 14:54:30 ot_train.py:354 step:15K smpl:973K ep:1K epch:27.31 loss:0.067 grdn:7.735 lr:1.0e-05 updt_s:0.355 data_s:0.031
90
+ INFO 2025-12-31 14:55:44 ot_train.py:354 step:15K smpl:986K ep:1K epch:27.67 loss:0.066 grdn:7.494 lr:1.0e-05 updt_s:0.356 data_s:0.015
91
+ INFO 2025-12-31 14:57:01 ot_train.py:354 step:16K smpl:998K ep:1K epch:28.03 loss:0.067 grdn:7.772 lr:1.0e-05 updt_s:0.354 data_s:0.030
92
+ INFO 2025-12-31 14:58:16 ot_train.py:354 step:16K smpl:1M ep:1K epch:28.39 loss:0.066 grdn:7.257 lr:1.0e-05 updt_s:0.356 data_s:0.015
93
+ INFO 2025-12-31 14:59:30 ot_train.py:354 step:16K smpl:1M ep:1K epch:28.75 loss:0.065 grdn:7.126 lr:1.0e-05 updt_s:0.356 data_s:0.015
94
+ INFO 2025-12-31 15:00:47 ot_train.py:354 step:16K smpl:1M ep:1K epch:29.11 loss:0.066 grdn:7.749 lr:1.0e-05 updt_s:0.354 data_s:0.032
95
+ INFO 2025-12-31 15:02:01 ot_train.py:354 step:16K smpl:1M ep:1K epch:29.47 loss:0.064 grdn:7.243 lr:1.0e-05 updt_s:0.356 data_s:0.015
96
+ INFO 2025-12-31 15:03:16 ot_train.py:354 step:17K smpl:1M ep:1K epch:29.83 loss:0.064 grdn:7.316 lr:1.0e-05 updt_s:0.356 data_s:0.015
97
+ INFO 2025-12-31 15:04:33 ot_train.py:354 step:17K smpl:1M ep:1K epch:30.19 loss:0.062 grdn:6.284 lr:1.0e-05 updt_s:0.354 data_s:0.032
98
+ INFO 2025-12-31 15:05:47 ot_train.py:354 step:17K smpl:1M ep:1K epch:30.55 loss:0.062 grdn:7.610 lr:1.0e-05 updt_s:0.356 data_s:0.015
99
+ INFO 2025-12-31 15:07:02 ot_train.py:354 step:17K smpl:1M ep:1K epch:30.91 loss:0.064 grdn:7.469 lr:1.0e-05 updt_s:0.356 data_s:0.015
100
+ INFO 2025-12-31 15:08:19 ot_train.py:354 step:17K smpl:1M ep:1K epch:31.27 loss:0.061 grdn:7.114 lr:1.0e-05 updt_s:0.355 data_s:0.032
101
+ INFO 2025-12-31 15:09:33 ot_train.py:354 step:18K smpl:1M ep:1K epch:31.62 loss:0.062 grdn:6.946 lr:1.0e-05 updt_s:0.356 data_s:0.015
102
+ INFO 2025-12-31 15:10:48 ot_train.py:354 step:18K smpl:1M ep:1K epch:31.98 loss:0.063 grdn:8.028 lr:1.0e-05 updt_s:0.356 data_s:0.015
103
+ INFO 2025-12-31 15:12:04 ot_train.py:354 step:18K smpl:1M ep:1K epch:32.34 loss:0.062 grdn:7.480 lr:1.0e-05 updt_s:0.354 data_s:0.030
104
+ INFO 2025-12-31 15:13:19 ot_train.py:354 step:18K smpl:1M ep:1K epch:32.70 loss:0.061 grdn:7.474 lr:1.0e-05 updt_s:0.356 data_s:0.015
105
+ INFO 2025-12-31 15:14:36 ot_train.py:354 step:18K smpl:1M ep:1K epch:33.06 loss:0.060 grdn:6.592 lr:1.0e-05 updt_s:0.355 data_s:0.032
106
+ INFO 2025-12-31 15:15:50 ot_train.py:354 step:19K smpl:1M ep:1K epch:33.42 loss:0.059 grdn:6.739 lr:1.0e-05 updt_s:0.356 data_s:0.015
107
+ INFO 2025-12-31 15:17:05 ot_train.py:354 step:19K smpl:1M ep:1K epch:33.78 loss:0.061 grdn:7.565 lr:1.0e-05 updt_s:0.355 data_s:0.015
108
+ INFO 2025-12-31 15:18:22 ot_train.py:354 step:19K smpl:1M ep:1K epch:34.14 loss:0.058 grdn:6.927 lr:1.0e-05 updt_s:0.355 data_s:0.031
109
+ INFO 2025-12-31 15:19:36 ot_train.py:354 step:19K smpl:1M ep:1K epch:34.50 loss:0.060 grdn:7.307 lr:1.0e-05 updt_s:0.356 data_s:0.015
110
+ INFO 2025-12-31 15:20:50 ot_train.py:354 step:19K smpl:1M ep:1K epch:34.86 loss:0.058 grdn:6.553 lr:1.0e-05 updt_s:0.355 data_s:0.015
111
+ INFO 2025-12-31 15:22:08 ot_train.py:354 step:20K smpl:1M ep:1K epch:35.22 loss:0.058 grdn:6.391 lr:1.0e-05 updt_s:0.354 data_s:0.032
112
+ INFO 2025-12-31 15:23:22 ot_train.py:354 step:20K smpl:1M ep:1K epch:35.58 loss:0.057 grdn:6.716 lr:1.0e-05 updt_s:0.357 data_s:0.015
113
+ INFO 2025-12-31 15:24:36 ot_train.py:354 step:20K smpl:1M ep:1K epch:35.94 loss:0.058 grdn:7.495 lr:1.0e-05 updt_s:0.355 data_s:0.015
114
+ INFO 2025-12-31 15:24:36 ot_train.py:364 Checkpoint policy after step 20000
115
+ INFO 2025-12-31 15:25:55 ot_train.py:354 step:20K smpl:1M ep:1K epch:36.30 loss:0.058 grdn:7.561 lr:1.0e-05 updt_s:0.355 data_s:0.032
116
+ INFO 2025-12-31 15:27:10 ot_train.py:354 step:20K smpl:1M ep:1K epch:36.66 loss:0.058 grdn:7.133 lr:1.0e-05 updt_s:0.356 data_s:0.015
117
+ INFO 2025-12-31 15:28:24 ot_train.py:354 step:21K smpl:1M ep:1K epch:37.01 loss:0.057 grdn:6.540 lr:1.0e-05 updt_s:0.356 data_s:0.015
118
+ INFO 2025-12-31 15:29:41 ot_train.py:354 step:21K smpl:1M ep:1K epch:37.37 loss:0.056 grdn:6.810 lr:1.0e-05 updt_s:0.355 data_s:0.031
119
+ INFO 2025-12-31 15:30:55 ot_train.py:354 step:21K smpl:1M ep:2K epch:37.73 loss:0.055 grdn:5.939 lr:1.0e-05 updt_s:0.356 data_s:0.015
120
+ INFO 2025-12-31 15:32:13 ot_train.py:354 step:21K smpl:1M ep:2K epch:38.09 loss:0.057 grdn:7.004 lr:1.0e-05 updt_s:0.354 data_s:0.031
121
+ INFO 2025-12-31 15:33:27 ot_train.py:354 step:21K smpl:1M ep:2K epch:38.45 loss:0.054 grdn:5.749 lr:1.0e-05 updt_s:0.356 data_s:0.015
122
+ INFO 2025-12-31 15:34:41 ot_train.py:354 step:22K smpl:1M ep:2K epch:38.81 loss:0.055 grdn:6.297 lr:1.0e-05 updt_s:0.356 data_s:0.015
123
+ INFO 2025-12-31 15:35:58 ot_train.py:354 step:22K smpl:1M ep:2K epch:39.17 loss:0.056 grdn:6.978 lr:1.0e-05 updt_s:0.354 data_s:0.031
124
+ INFO 2025-12-31 15:37:13 ot_train.py:354 step:22K smpl:1M ep:2K epch:39.53 loss:0.054 grdn:6.632 lr:1.0e-05 updt_s:0.356 data_s:0.015
125
+ INFO 2025-12-31 15:38:27 ot_train.py:354 step:22K smpl:1M ep:2K epch:39.89 loss:0.055 grdn:6.527 lr:1.0e-05 updt_s:0.356 data_s:0.015
126
+ INFO 2025-12-31 15:39:44 ot_train.py:354 step:22K smpl:1M ep:2K epch:40.25 loss:0.053 grdn:6.137 lr:1.0e-05 updt_s:0.354 data_s:0.031
127
+ INFO 2025-12-31 15:40:58 ot_train.py:354 step:23K smpl:1M ep:2K epch:40.61 loss:0.054 grdn:6.441 lr:1.0e-05 updt_s:0.357 data_s:0.015
128
+ INFO 2025-12-31 15:42:13 ot_train.py:354 step:23K smpl:1M ep:2K epch:40.97 loss:0.054 grdn:6.459 lr:1.0e-05 updt_s:0.355 data_s:0.015
129
+ INFO 2025-12-31 15:43:30 ot_train.py:354 step:23K smpl:1M ep:2K epch:41.33 loss:0.053 grdn:6.023 lr:1.0e-05 updt_s:0.355 data_s:0.032
130
+ INFO 2025-12-31 15:44:44 ot_train.py:354 step:23K smpl:1M ep:2K epch:41.69 loss:0.053 grdn:6.146 lr:1.0e-05 updt_s:0.356 data_s:0.015
131
+ INFO 2025-12-31 15:46:02 ot_train.py:354 step:23K smpl:1M ep:2K epch:42.05 loss:0.053 grdn:6.212 lr:1.0e-05 updt_s:0.355 data_s:0.031
132
+ INFO 2025-12-31 15:47:16 ot_train.py:354 step:24K smpl:2M ep:2K epch:42.41 loss:0.052 grdn:6.566 lr:1.0e-05 updt_s:0.355 data_s:0.015
133
+ INFO 2025-12-31 15:48:30 ot_train.py:354 step:24K smpl:2M ep:2K epch:42.76 loss:0.052 grdn:6.177 lr:1.0e-05 updt_s:0.356 data_s:0.015
134
+ INFO 2025-12-31 15:49:47 ot_train.py:354 step:24K smpl:2M ep:2K epch:43.12 loss:0.052 grdn:6.439 lr:1.0e-05 updt_s:0.354 data_s:0.031
135
+ INFO 2025-12-31 15:51:01 ot_train.py:354 step:24K smpl:2M ep:2K epch:43.48 loss:0.052 grdn:6.557 lr:1.0e-05 updt_s:0.356 data_s:0.015
136
+ INFO 2025-12-31 15:52:16 ot_train.py:354 step:24K smpl:2M ep:2K epch:43.84 loss:0.051 grdn:5.968 lr:1.0e-05 updt_s:0.357 data_s:0.015
137
+ INFO 2025-12-31 15:53:33 ot_train.py:354 step:25K smpl:2M ep:2K epch:44.20 loss:0.052 grdn:6.389 lr:1.0e-05 updt_s:0.354 data_s:0.031
138
+ INFO 2025-12-31 15:54:47 ot_train.py:354 step:25K smpl:2M ep:2K epch:44.56 loss:0.050 grdn:5.622 lr:1.0e-05 updt_s:0.356 data_s:0.015
139
+ INFO 2025-12-31 15:56:02 ot_train.py:354 step:25K smpl:2M ep:2K epch:44.92 loss:0.051 grdn:6.066 lr:1.0e-05 updt_s:0.356 data_s:0.015
140
+ INFO 2025-12-31 15:57:19 ot_train.py:354 step:25K smpl:2M ep:2K epch:45.28 loss:0.051 grdn:6.150 lr:1.0e-05 updt_s:0.354 data_s:0.031
141
+ INFO 2025-12-31 15:58:33 ot_train.py:354 step:25K smpl:2M ep:2K epch:45.64 loss:0.050 grdn:6.133 lr:1.0e-05 updt_s:0.357 data_s:0.015
142
+ INFO 2025-12-31 15:59:48 ot_train.py:354 step:26K smpl:2M ep:2K epch:46.00 loss:0.050 grdn:5.844 lr:1.0e-05 updt_s:0.355 data_s:0.016
143
+ INFO 2025-12-31 16:01:06 ot_train.py:354 step:26K smpl:2M ep:2K epch:46.36 loss:0.050 grdn:6.100 lr:1.0e-05 updt_s:0.355 data_s:0.034
144
+ INFO 2025-12-31 16:02:20 ot_train.py:354 step:26K smpl:2M ep:2K epch:46.72 loss:0.051 grdn:6.424 lr:1.0e-05 updt_s:0.356 data_s:0.015
145
+ INFO 2025-12-31 16:03:38 ot_train.py:354 step:26K smpl:2M ep:2K epch:47.08 loss:0.049 grdn:5.945 lr:1.0e-05 updt_s:0.355 data_s:0.035
146
+ INFO 2025-12-31 16:04:52 ot_train.py:354 step:26K smpl:2M ep:2K epch:47.44 loss:0.049 grdn:6.103 lr:1.0e-05 updt_s:0.356 data_s:0.016
147
+ INFO 2025-12-31 16:06:07 ot_train.py:354 step:27K smpl:2M ep:2K epch:47.80 loss:0.049 grdn:5.937 lr:1.0e-05 updt_s:0.356 data_s:0.017
148
+ INFO 2025-12-31 16:07:24 ot_train.py:354 step:27K smpl:2M ep:2K epch:48.16 loss:0.049 grdn:5.822 lr:1.0e-05 updt_s:0.354 data_s:0.032
149
+ INFO 2025-12-31 16:08:38 ot_train.py:354 step:27K smpl:2M ep:2K epch:48.51 loss:0.049 grdn:6.268 lr:1.0e-05 updt_s:0.357 data_s:0.015
150
+ INFO 2025-12-31 16:09:53 ot_train.py:354 step:27K smpl:2M ep:2K epch:48.87 loss:0.048 grdn:5.491 lr:1.0e-05 updt_s:0.356 data_s:0.015
151
+ INFO 2025-12-31 16:11:10 ot_train.py:354 step:27K smpl:2M ep:2K epch:49.23 loss:0.049 grdn:5.944 lr:1.0e-05 updt_s:0.354 data_s:0.032
152
+ INFO 2025-12-31 16:12:25 ot_train.py:354 step:28K smpl:2M ep:2K epch:49.59 loss:0.048 grdn:5.436 lr:1.0e-05 updt_s:0.356 data_s:0.016
153
+ INFO 2025-12-31 16:13:39 ot_train.py:354 step:28K smpl:2M ep:2K epch:49.95 loss:0.047 grdn:5.470 lr:1.0e-05 updt_s:0.356 data_s:0.015
154
+ INFO 2025-12-31 16:14:56 ot_train.py:354 step:28K smpl:2M ep:2K epch:50.31 loss:0.048 grdn:6.592 lr:1.0e-05 updt_s:0.355 data_s:0.032
155
+ INFO 2025-12-31 16:16:11 ot_train.py:354 step:28K smpl:2M ep:2K epch:50.67 loss:0.047 grdn:5.629 lr:1.0e-05 updt_s:0.356 data_s:0.015
156
+ INFO 2025-12-31 16:17:25 ot_train.py:354 step:28K smpl:2M ep:2K epch:51.03 loss:0.047 grdn:5.651 lr:1.0e-05 updt_s:0.355 data_s:0.015
157
+ INFO 2025-12-31 16:18:42 ot_train.py:354 step:29K smpl:2M ep:2K epch:51.39 loss:0.048 grdn:5.836 lr:1.0e-05 updt_s:0.355 data_s:0.032
158
+ INFO 2025-12-31 16:19:57 ot_train.py:354 step:29K smpl:2M ep:2K epch:51.75 loss:0.047 grdn:5.715 lr:1.0e-05 updt_s:0.357 data_s:0.015
159
+ INFO 2025-12-31 16:21:14 ot_train.py:354 step:29K smpl:2M ep:2K epch:52.11 loss:0.048 grdn:5.874 lr:1.0e-05 updt_s:0.354 data_s:0.033
160
+ INFO 2025-12-31 16:22:28 ot_train.py:354 step:29K smpl:2M ep:2K epch:52.47 loss:0.046 grdn:5.131 lr:1.0e-05 updt_s:0.356 data_s:0.015
161
+ INFO 2025-12-31 16:23:43 ot_train.py:354 step:29K smpl:2M ep:2K epch:52.83 loss:0.046 grdn:5.731 lr:1.0e-05 updt_s:0.356 data_s:0.015
162
+ INFO 2025-12-31 16:25:00 ot_train.py:354 step:30K smpl:2M ep:2K epch:53.19 loss:0.047 grdn:5.571 lr:1.0e-05 updt_s:0.354 data_s:0.031
163
+ INFO 2025-12-31 16:26:14 ot_train.py:354 step:30K smpl:2M ep:2K epch:53.55 loss:0.047 grdn:5.913 lr:1.0e-05 updt_s:0.357 data_s:0.015
164
+ INFO 2025-12-31 16:27:29 ot_train.py:354 step:30K smpl:2M ep:2K epch:53.91 loss:0.047 grdn:5.821 lr:1.0e-05 updt_s:0.356 data_s:0.015
165
+ INFO 2025-12-31 16:28:46 ot_train.py:354 step:30K smpl:2M ep:2K epch:54.26 loss:0.046 grdn:5.620 lr:1.0e-05 updt_s:0.354 data_s:0.033
166
+ INFO 2025-12-31 16:30:01 ot_train.py:354 step:30K smpl:2M ep:2K epch:54.62 loss:0.045 grdn:5.175 lr:1.0e-05 updt_s:0.356 data_s:0.015
167
+ INFO 2025-12-31 16:31:15 ot_train.py:354 step:31K smpl:2M ep:2K epch:54.98 loss:0.044 grdn:4.907 lr:1.0e-05 updt_s:0.356 data_s:0.015
168
+ INFO 2025-12-31 16:32:32 ot_train.py:354 step:31K smpl:2M ep:2K epch:55.34 loss:0.045 grdn:5.579 lr:1.0e-05 updt_s:0.355 data_s:0.031
169
+ INFO 2025-12-31 16:33:47 ot_train.py:354 step:31K smpl:2M ep:2K epch:55.70 loss:0.045 grdn:5.493 lr:1.0e-05 updt_s:0.356 data_s:0.015
170
+ INFO 2025-12-31 16:35:04 ot_train.py:354 step:31K smpl:2M ep:2K epch:56.06 loss:0.045 grdn:5.320 lr:1.0e-05 updt_s:0.355 data_s:0.031
171
+ INFO 2025-12-31 16:36:18 ot_train.py:354 step:31K smpl:2M ep:2K epch:56.42 loss:0.044 grdn:5.413 lr:1.0e-05 updt_s:0.356 data_s:0.015
172
+ INFO 2025-12-31 16:37:33 ot_train.py:354 step:32K smpl:2M ep:2K epch:56.78 loss:0.044 grdn:5.496 lr:1.0e-05 updt_s:0.357 data_s:0.015
173
+ INFO 2025-12-31 16:38:50 ot_train.py:354 step:32K smpl:2M ep:2K epch:57.14 loss:0.045 grdn:5.232 lr:1.0e-05 updt_s:0.354 data_s:0.033
174
+ INFO 2025-12-31 16:40:04 ot_train.py:354 step:32K smpl:2M ep:2K epch:57.50 loss:0.044 grdn:5.274 lr:1.0e-05 updt_s:0.356 data_s:0.015
175
+ INFO 2025-12-31 16:41:19 ot_train.py:354 step:32K smpl:2M ep:2K epch:57.86 loss:0.045 grdn:5.434 lr:1.0e-05 updt_s:0.356 data_s:0.016
176
+ INFO 2025-12-31 16:42:36 ot_train.py:354 step:32K smpl:2M ep:2K epch:58.22 loss:0.044 grdn:5.135 lr:1.0e-05 updt_s:0.355 data_s:0.030
177
+ INFO 2025-12-31 16:43:50 ot_train.py:354 step:33K smpl:2M ep:2K epch:58.58 loss:0.044 grdn:5.474 lr:1.0e-05 updt_s:0.356 data_s:0.015
178
+ INFO 2025-12-31 16:45:05 ot_train.py:354 step:33K smpl:2M ep:2K epch:58.94 loss:0.044 grdn:5.560 lr:1.0e-05 updt_s:0.356 data_s:0.016
179
+ INFO 2025-12-31 16:46:23 ot_train.py:354 step:33K smpl:2M ep:2K epch:59.30 loss:0.044 grdn:5.377 lr:1.0e-05 updt_s:0.355 data_s:0.035
180
+ INFO 2025-12-31 16:47:37 ot_train.py:354 step:33K smpl:2M ep:2K epch:59.66 loss:0.044 grdn:5.795 lr:1.0e-05 updt_s:0.357 data_s:0.016
181
+ INFO 2025-12-31 16:48:52 ot_train.py:354 step:33K smpl:2M ep:2K epch:60.01 loss:0.044 grdn:5.346 lr:1.0e-05 updt_s:0.355 data_s:0.016
182
+ INFO 2025-12-31 16:50:09 ot_train.py:354 step:34K smpl:2M ep:2K epch:60.37 loss:0.043 grdn:5.306 lr:1.0e-05 updt_s:0.355 data_s:0.034
183
+ INFO 2025-12-31 16:51:24 ot_train.py:354 step:34K smpl:2M ep:2K epch:60.73 loss:0.045 grdn:5.791 lr:1.0e-05 updt_s:0.356 data_s:0.016
184
+ INFO 2025-12-31 16:52:42 ot_train.py:354 step:34K smpl:2M ep:2K epch:61.09 loss:0.042 grdn:5.182 lr:1.0e-05 updt_s:0.355 data_s:0.032
185
+ INFO 2025-12-31 16:53:56 ot_train.py:354 step:34K smpl:2M ep:2K epch:61.45 loss:0.042 grdn:5.181 lr:1.0e-05 updt_s:0.356 data_s:0.015
186
+ INFO 2025-12-31 16:55:10 ot_train.py:354 step:34K smpl:2M ep:2K epch:61.81 loss:0.042 grdn:5.000 lr:1.0e-05 updt_s:0.356 data_s:0.016
187
+ INFO 2025-12-31 16:56:29 ot_train.py:354 step:35K smpl:2M ep:2K epch:62.17 loss:0.042 grdn:5.466 lr:1.0e-05 updt_s:0.354 data_s:0.038
188
+ INFO 2025-12-31 16:57:44 ot_train.py:354 step:35K smpl:2M ep:3K epch:62.53 loss:0.042 grdn:5.392 lr:1.0e-05 updt_s:0.357 data_s:0.016
189
+ INFO 2025-12-31 16:58:58 ot_train.py:354 step:35K smpl:2M ep:3K epch:62.89 loss:0.043 grdn:5.261 lr:1.0e-05 updt_s:0.355 data_s:0.016
190
+ INFO 2025-12-31 16:58:58 ot_train.py:364 Checkpoint policy after step 35000
191
+ INFO 2025-12-31 16:59:00 ot_train.py:435 End of training
192
+ Traceback (most recent call last):
193
+ File "/home/zeux/.local/lib/python3.10/site-packages/huggingface_hub/utils/_http.py", line 407, in hf_raise_for_status
194
+ response.raise_for_status()
195
+ File "/home/zeux/.local/lib/python3.10/site-packages/requests/models.py", line 1026, in raise_for_status
196
+ raise HTTPError(http_error_msg, response=self)
197
+ requests.exceptions.HTTPError: 403 Client Error: Forbidden for url: https://huggingface.co/api/repos/create
198
+
199
+ The above exception was the direct cause of the following exception:
200
+
201
+ Traceback (most recent call last):
202
+ File "/home/zeux/miniconda3/envs/lerobot/bin/lerobot-train", line 7, in <module>
203
+ sys.exit(main())
204
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/scripts/lerobot_train.py", line 449, in main
205
+ train()
206
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/configs/parser.py", line 233, in wrapper_inner
207
+ response = fn(cfg, *args, **kwargs)
208
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/scripts/lerobot_train.py", line 439, in train
209
+ unwrapped_policy.push_model_to_hub(cfg)
210
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/policies/pretrained.py", line 211, in push_model_to_hub
211
+ repo_id = api.create_repo(
212
+ File "/home/zeux/.local/lib/python3.10/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
213
+ return fn(*args, **kwargs)
214
+ File "/home/zeux/.local/lib/python3.10/site-packages/huggingface_hub/hf_api.py", line 3779, in create_repo
215
+ raise err
216
+ File "/home/zeux/.local/lib/python3.10/site-packages/huggingface_hub/hf_api.py", line 3766, in create_repo
217
+ hf_raise_for_status(r)
218
+ File "/home/zeux/.local/lib/python3.10/site-packages/huggingface_hub/utils/_http.py", line 471, in hf_raise_for_status
219
+ raise _format(HfHubHTTPError, message, response) from e
220
+ huggingface_hub.errors.HfHubHTTPError: (Request ID: Root=1-695508fc-3fbeb463197fb585233e2603;49958f1c-88b4-4567-919e-842ecfe126ab)
221
+
222
+ 403 Forbidden: You don't have the rights to create a model under the namespace "local".
223
+ Cannot access content at: https://huggingface.co/api/repos/create.
224
+ Make sure your token has the correct permissions.
225
+ Traceback (most recent call last):
226
+ File "/home/zeux/.local/lib/python3.10/site-packages/huggingface_hub/utils/_http.py", line 407, in hf_raise_for_status
227
+ response.raise_for_status()
228
+ File "/home/zeux/.local/lib/python3.10/site-packages/requests/models.py", line 1026, in raise_for_status
229
+ raise HTTPError(http_error_msg, response=self)
230
+ requests.exceptions.HTTPError: 403 Client Error: Forbidden for url: https://huggingface.co/api/repos/create
231
+
232
+ The above exception was the direct cause of the following exception:
233
+
234
+ Traceback (most recent call last):
235
+ File "/home/zeux/miniconda3/envs/lerobot/bin/lerobot-train", line 7, in <module>
236
+ sys.exit(main())
237
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/scripts/lerobot_train.py", line 449, in main
238
+ train()
239
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/configs/parser.py", line 233, in wrapper_inner
240
+ response = fn(cfg, *args, **kwargs)
241
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/scripts/lerobot_train.py", line 439, in train
242
+ unwrapped_policy.push_model_to_hub(cfg)
243
+ File "/home/zeux/miniconda3/envs/lerobot/lib/python3.10/site-packages/lerobot/policies/pretrained.py", line 211, in push_model_to_hub
244
+ repo_id = api.create_repo(
245
+ File "/home/zeux/.local/lib/python3.10/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
246
+ return fn(*args, **kwargs)
247
+ File "/home/zeux/.local/lib/python3.10/site-packages/huggingface_hub/hf_api.py", line 3779, in create_repo
248
+ raise err
249
+ File "/home/zeux/.local/lib/python3.10/site-packages/huggingface_hub/hf_api.py", line 3766, in create_repo
250
+ hf_raise_for_status(r)
251
+ File "/home/zeux/.local/lib/python3.10/site-packages/huggingface_hub/utils/_http.py", line 471, in hf_raise_for_status
252
+ raise _format(HfHubHTTPError, message, response) from e
253
+ huggingface_hub.errors.HfHubHTTPError: (Request ID: Root=1-695508fc-3fbeb463197fb585233e2603;49958f1c-88b4-4567-919e-842ecfe126ab)
254
+
255
+ 403 Forbidden: You don't have the rights to create a model under the namespace "local".
256
+ Cannot access content at: https://huggingface.co/api/repos/create.
257
+ Make sure your token has the correct permissions.
lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/files/requirements.txt ADDED
@@ -0,0 +1,257 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ nvidia-cusolver==12.0.3.29
2
+ nvidia-curand==10.4.0.35
3
+ anyio==4.12.0
4
+ yarl==1.22.0
5
+ comfyui-workflow-templates-media-other==0.3.15
6
+ tomlkit==0.13.3
7
+ cloudpickle==3.1.2
8
+ typing_extensions==4.15.0
9
+ sentry-sdk==2.48.0
10
+ semver==3.0.4
11
+ mpmath==1.3.0
12
+ draccus==0.10.0
13
+ multidict==6.7.0
14
+ nvidia-cufile-cu12==1.11.1.6
15
+ tokenizers==0.22.1
16
+ GitPython==3.1.45
17
+ datasets==4.1.1
18
+ PySide6_Essentials==6.10.1
19
+ comfyui-workflow-templates-media-image==0.3.17
20
+ Pygments==2.19.2
21
+ aiohappyeyeballs==2.6.1
22
+ protobuf==6.33.2
23
+ PySide6==6.10.1
24
+ comfyui-workflow-templates-core==0.3.12
25
+ PySide2==5.15.2.1
26
+ pyserial==3.5
27
+ accelerate==1.12.0
28
+ pfzy==0.3.4
29
+ scipy==1.15.3
30
+ frozenlist==1.8.0
31
+ nvidia-nvshmem-cu13==3.3.24
32
+ transformers==4.57.3
33
+ typing-inspection==0.4.2
34
+ einops==0.8.1
35
+ shellingham==1.5.4
36
+ Farama-Notifications==0.0.4
37
+ psutil==7.1.3
38
+ rerun-sdk==0.26.2
39
+ PyYAML==6.0.3
40
+ nvidia-cufft==12.0.0.15
41
+ cookiecutter==2.6.0
42
+ rich==14.2.0
43
+ SQLAlchemy==2.0.44
44
+ pydantic_core==2.41.5
45
+ nvidia-nvjitlink==13.0.39
46
+ questionary==2.1.1
47
+ requests==2.32.5
48
+ nvidia-cuda-nvrtc==13.0.48
49
+ torchcodec==0.5
50
+ PyOpenGL-accelerate==3.1.10
51
+ comfyui_frontend_package==1.33.10
52
+ networkx==3.4.2
53
+ nvidia-nvtx==13.0.39
54
+ setuptools==80.9.0
55
+ trampoline==0.1.2
56
+ pandas==2.3.3
57
+ httpx==0.28.1
58
+ av==15.1.0
59
+ comfy-cli==1.5.3
60
+ PyOpenGL==3.1.10
61
+ tomli==2.3.0
62
+ safetensors==0.7.0
63
+ regex==2025.11.3
64
+ wcwidth==0.2.14
65
+ pyyaml-include==1.4.1
66
+ kornia==0.8.2
67
+ packaging==25.0
68
+ nvidia-nccl-cu13==2.27.7
69
+ pydantic==2.12.5
70
+ mixpanel==5.0.0
71
+ markdown-it-py==4.0.0
72
+ multiprocess==0.70.16
73
+ usd-exchange==2.1.0
74
+ h11==0.16.0
75
+ mergedeep==1.3.4
76
+ shiboken6==6.10.1
77
+ text-unidecode==1.3
78
+ deepdiff==8.6.1
79
+ toml==0.10.2
80
+ aiosignal==1.4.0
81
+ imageio-ffmpeg==0.6.0
82
+ python-dateutil==2.9.0.post0
83
+ uv==0.9.16
84
+ jsonlines==4.0.0
85
+ xxhash==3.6.0
86
+ numpy==2.2.6
87
+ nvidia-cuda-runtime==13.0.48
88
+ arrow==1.4.0
89
+ spandrel==0.4.1
90
+ nvidia-cuda-cupti==13.0.48
91
+ python-dotenv==1.2.1
92
+ async-timeout==5.0.1
93
+ urllib3==2.6.2
94
+ fsspec==2025.9.0
95
+ nvidia-cusparse==12.6.2.49
96
+ binaryornot==0.4.4
97
+ nvidia-cufile==1.15.0.42
98
+ mdurl==0.1.2
99
+ pydantic-settings==2.12.0
100
+ gitdb==4.0.12
101
+ kornia_rs==0.1.10
102
+ attrs==25.4.0
103
+ mypy_extensions==1.1.0
104
+ tqdm==4.67.1
105
+ alembic==1.17.2
106
+ comfyui-embedded-docs==0.3.1
107
+ python-slugify==8.0.4
108
+ cmake==4.1.3
109
+ smmap==5.0.2
110
+ comfyui-workflow-templates-media-api==0.3.14
111
+ torchsde==0.2.6
112
+ pathspec==0.12.1
113
+ sentencepiece==0.2.1
114
+ charset-normalizer==3.4.4
115
+ nvidia-cusparselt-cu13==0.8.0
116
+ python-xlib==0.33
117
+ ruff==0.14.8
118
+ platformdirs==4.5.1
119
+ inquirerpy==0.3.4
120
+ evdev==1.9.2
121
+ huggingface-hub==0.35.3
122
+ nvidia-cublas==13.0.0.19
123
+ prompt_toolkit==3.0.52
124
+ ImageIO==2.37.2
125
+ termcolor==3.3.0
126
+ wandb==0.21.4
127
+ annotated-types==0.7.0
128
+ comfyui-workflow-templates-media-video==0.3.12
129
+ typer==0.20.0
130
+ dill==0.4.0
131
+ propcache==0.4.1
132
+ typing-inspect==0.9.0
133
+ greenlet==3.3.0
134
+ hf-xet==1.2.0
135
+ shiboken2==5.15.2.1
136
+ tzdata==2025.2
137
+ aiohttp==3.13.2
138
+ websocket-client==1.9.0
139
+ nvidia-cudnn-cu13==9.13.0.50
140
+ pyarrow==22.0.0
141
+ orderly-set==5.5.0
142
+ pynput==1.8.1
143
+ diffusers==0.35.2
144
+ PySide6_Addons==6.10.1
145
+ asgiref==3.11.0
146
+ httpcore==1.0.9
147
+ exceptiongroup==1.3.1
148
+ filelock==3.20.0
149
+ comfyui_workflow_templates==0.7.25
150
+ opencv-python-headless==4.12.0.88
151
+ nvidia-curand-cu12==10.3.7.77
152
+ nvidia-cudnn-cu12==9.5.1.17
153
+ nvidia-nvtx-cu12==12.6.77
154
+ cloudpickle==3.1.2
155
+ Flask==3.1.2
156
+ pytz==2025.2
157
+ nvidia-cusparselt-cu12==0.6.3
158
+ nvidia-cusparselt-cu12==0.7.1
159
+ sentry-sdk==2.48.0
160
+ importlib_metadata==8.7.1
161
+ draccus==0.10.0
162
+ pycparser==2.23
163
+ nvidia-cufile-cu12==1.11.1.6
164
+ numcodecs==0.13.1
165
+ datasets==4.1.1
166
+ nvidia-cusolver-cu12==11.7.1.2
167
+ protobuf==6.33.2
168
+ pyserial==3.5
169
+ accelerate==1.12.0
170
+ pfzy==0.3.4
171
+ setuptools==80.9.0
172
+ asciitree==0.3.3
173
+ antlr4-python3-runtime==4.9.3
174
+ nvidia-cuda-cupti-cu12==12.6.80
175
+ Farama-Notifications==0.0.4
176
+ rerun-sdk==0.26.2
177
+ PyYAML==6.0.3
178
+ PySocks==1.7.1
179
+ torchvision==0.24.1+cu130
180
+ requests==2.32.5
181
+ torchcodec==0.5
182
+ cffi==2.0.0
183
+ itsdangerous==2.2.0
184
+ idna==3.11
185
+ zipp==3.23.0
186
+ pandas==2.3.3
187
+ av==15.1.0
188
+ nvidia-cusparse-cu12==12.5.4.2
189
+ pymunk==6.11.1
190
+ pyyaml-include==1.4.1
191
+ gymnasium==1.2.3
192
+ nvidia-cuda-runtime-cu12==12.6.77
193
+ h5py==3.15.1
194
+ triton==3.6.0+git9844da95
195
+ triton==3.5.1
196
+ multiprocess==0.70.16
197
+ lerobot==0.4.2
198
+ Jinja2==3.1.6
199
+ click==8.3.1
200
+ nvidia-nccl-cu12==2.28.9
201
+ nvidia-nccl-cu12==2.26.2
202
+ mergedeep==1.3.4
203
+ deepdiff==8.6.1
204
+ toml==0.10.2
205
+ sympy==1.14.0
206
+ hf_transfer==0.1.9
207
+ imageio-ffmpeg==0.6.0
208
+ python-dateutil==2.9.0.post0
209
+ soupsieve==2.8.1
210
+ jsonlines==4.0.0
211
+ xxhash==3.6.0
212
+ pyzmq==27.1.0
213
+ nvidia-cuda-nvrtc-cu12==12.6.77
214
+ omegaconf==2.3.0
215
+ Werkzeug==3.1.4
216
+ urllib3==2.6.2
217
+ fsspec==2025.9.0
218
+ pillow==12.0.0
219
+ pytorch-triton==3.1.0+cf34004b8a
220
+ cuda-pathfinder==1.2.2
221
+ gdown==5.2.0
222
+ beautifulsoup4==4.14.3
223
+ mypy_extensions==1.1.0
224
+ numba==0.63.1
225
+ cmake==4.1.3
226
+ certifi==2025.11.12
227
+ num2words==0.5.14
228
+ fasteners==0.20
229
+ python-xlib==0.33
230
+ nvidia-nvjitlink-cu12==12.6.85
231
+ platformdirs==4.5.1
232
+ torch==2.9.1+cu130
233
+ inquirerpy==0.3.4
234
+ evdev==1.9.2
235
+ huggingface-hub==0.35.3
236
+ nvidia-nvshmem-cu12==3.4.5
237
+ ImageIO==2.37.2
238
+ termcolor==3.2.0
239
+ zarr==2.18.3
240
+ six==1.17.0
241
+ blinker==1.9.0
242
+ wandb==0.21.4
243
+ MarkupSafe==3.0.3
244
+ wheel==0.45.1
245
+ dill==0.4.0
246
+ typing-inspect==0.9.0
247
+ cuda-bindings==12.9.4
248
+ pip==25.3
249
+ pyarrow==22.0.0
250
+ orderly-set==5.5.0
251
+ pynput==1.8.1
252
+ diffusers==0.35.2
253
+ nvidia-cublas-cu12==12.6.4.1
254
+ docopt==0.6.2
255
+ llvmlite==0.46.0
256
+ nvidia-cufft-cu12==11.3.0.4
257
+ opencv-python-headless==4.12.0.88
lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/files/wandb-metadata.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-6.8.0-90-generic-x86_64-with-glibc2.35",
3
+ "python": "CPython 3.10.19",
4
+ "startedAt": "2025-12-31T07:48:53.840056Z",
5
+ "args": [
6
+ "--dataset.repo_id=lerobot-act",
7
+ "--dataset.root=/home/zeux/lerobot-act-dataset",
8
+ "--dataset.video_backend=pyav",
9
+ "--policy.type=act",
10
+ "--policy.repo_id=local/lerobot-act-so101-model",
11
+ "--output_dir=outputs/train/lerobot-act_so101_test",
12
+ "--policy.device=cuda",
13
+ "--policy.use_amp=true",
14
+ "--batch_size=64",
15
+ "--num_workers=8",
16
+ "--steps=35000",
17
+ "--wandb.enable=true",
18
+ "--policy.push_to_hub=true"
19
+ ],
20
+ "program": "/home/zeux/miniconda3/envs/lerobot/bin/lerobot-train",
21
+ "codePath": "miniconda3/envs/lerobot/bin/lerobot-train",
22
+ "codePathLocal": "miniconda3/envs/lerobot/bin/lerobot-train",
23
+ "root": "outputs/train/lerobot-act_so101_test",
24
+ "host": "zeux",
25
+ "executable": "/home/zeux/miniconda3/envs/lerobot/bin/python3.10",
26
+ "cpu_count": 24,
27
+ "cpu_count_logical": 24,
28
+ "gpu": "NVIDIA RTX PRO 6000 Blackwell Workstation Edition",
29
+ "gpu_count": 1,
30
+ "disk": {
31
+ "/": {
32
+ "total": "200449576960",
33
+ "used": "34473017344"
34
+ }
35
+ },
36
+ "memory": {
37
+ "total": "134489886720"
38
+ },
39
+ "gpu_nvidia": [
40
+ {
41
+ "name": "NVIDIA RTX PRO 6000 Blackwell Workstation Edition",
42
+ "memoryTotal": "102641958912",
43
+ "cudaCores": 24064,
44
+ "architecture": "Blackwell",
45
+ "uuid": "GPU-8c66f3c4-00c8-00a6-e5ff-633c5558ec8e"
46
+ }
47
+ ],
48
+ "cudaVersion": "13.0",
49
+ "writerId": "nxltchl4gaduvozis0gz9xveosym3k7a"
50
+ }
lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/files/wandb-summary.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"train/loss":0.04274633360095322,"_wandb":{"runtime":13205},"train/epochs":62.8895502274131,"train/samples":2240000,"_runtime":13205.796850913,"train/kld_loss":9.622704237699509e-05,"train/update_s":0.35532118829112735,"train/grad_norm":5.261109675168991,"train/lr":1.0000000000000021e-05,"train/steps":35000,"_timestamp":1.7671805384628017e+09,"train/episodes":2515.582009096524,"train/l1_loss":0.04637971520423889,"train/dataloading_s":0.016312788063078187,"_step":35000}
lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/logs/debug-core.log ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2025-12-31T13:18:53.875818977+05:30","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpgumkqn5p/port-1804411.txt","pid":1804411,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
2
+ {"time":"2025-12-31T13:18:53.876237277+05:30","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-1804411-1804472-673694980/socket","Net":"unix"}}
3
+ {"time":"2025-12-31T13:18:53.876297602+05:30","level":"INFO","msg":"server: will exit if parent process dies","ppid":1804411}
4
+ {"time":"2025-12-31T13:18:54.06607548+05:30","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
5
+ {"time":"2025-12-31T13:18:54.069707541+05:30","level":"INFO","msg":"handleInformInit: received","streamId":"8ce8thqh","id":"1(@)"}
6
+ {"time":"2025-12-31T13:18:54.476514062+05:30","level":"INFO","msg":"handleInformInit: stream started","streamId":"8ce8thqh","id":"1(@)"}
7
+ {"time":"2025-12-31T16:59:00.732355125+05:30","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
8
+ {"time":"2025-12-31T16:59:00.732432869+05:30","level":"INFO","msg":"connection: closing","id":"1(@)"}
9
+ {"time":"2025-12-31T16:59:00.732457152+05:30","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
10
+ {"time":"2025-12-31T16:59:00.732459909+05:30","level":"INFO","msg":"server is shutting down"}
11
+ {"time":"2025-12-31T16:59:00.732491666+05:30","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-1804411-1804472-673694980/socket","Net":"unix"}}
12
+ {"time":"2025-12-31T16:59:23.668865223+05:30","level":"ERROR","msg":"processOutgoingData: flush error","error":"write unix /tmp/wandb-1804411-1804472-673694980/socket->@: use of closed network connection","id":"1(@)"}
13
+ {"time":"2025-12-31T16:59:24.106575344+05:30","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
14
+ {"time":"2025-12-31T16:59:24.10659737+05:30","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
15
+ {"time":"2025-12-31T16:59:24.10660352+05:30","level":"INFO","msg":"server is closed"}
lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/logs/debug-internal.log ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2025-12-31T13:18:54.069787725+05:30","level":"INFO","msg":"stream: starting","core version":"0.21.4"}
2
+ {"time":"2025-12-31T13:18:54.476490438+05:30","level":"INFO","msg":"stream: created new stream","id":"8ce8thqh"}
3
+ {"time":"2025-12-31T13:18:54.476511345+05:30","level":"INFO","msg":"stream: started","id":"8ce8thqh"}
4
+ {"time":"2025-12-31T13:18:54.476515165+05:30","level":"INFO","msg":"writer: started","stream_id":"8ce8thqh"}
5
+ {"time":"2025-12-31T13:18:54.476566118+05:30","level":"INFO","msg":"handler: started","stream_id":"8ce8thqh"}
6
+ {"time":"2025-12-31T13:18:54.476575461+05:30","level":"INFO","msg":"sender: started","stream_id":"8ce8thqh"}
7
+ {"time":"2025-12-31T16:59:00.732388154+05:30","level":"INFO","msg":"stream: closing","id":"8ce8thqh"}
8
+ {"time":"2025-12-31T16:59:23.719001417+05:30","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
9
+ {"time":"2025-12-31T16:59:24.097369588+05:30","level":"INFO","msg":"handler: closed","stream_id":"8ce8thqh"}
10
+ {"time":"2025-12-31T16:59:24.097416229+05:30","level":"INFO","msg":"sender: closed","stream_id":"8ce8thqh"}
11
+ {"time":"2025-12-31T16:59:24.097422049+05:30","level":"INFO","msg":"stream: closed","id":"8ce8thqh"}
lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/logs/debug.log ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_setup.py:_flush():81] Current SDK version is 0.21.4
2
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_setup.py:_flush():81] Configure stats pid to 1804411
3
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_setup.py:_flush():81] Loading settings from /home/zeux/.config/wandb/settings
4
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_setup.py:_flush():81] Loading settings from /home/zeux/wandb/settings
5
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_setup.py:_flush():81] Loading settings from environment variables
6
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_init.py:setup_run_log_directory():686] Logging user logs to outputs/train/lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/logs/debug.log
7
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_init.py:setup_run_log_directory():687] Logging internal logs to outputs/train/lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/logs/debug-internal.log
8
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_init.py:init():813] calling init triggers
9
+ 2025-12-31 13:18:53,862 INFO MainThread:1804411 [wandb_init.py:init():818] wandb.init called with sweep_config: {}
10
+ config: {'dataset': {'repo_id': 'lerobot-act', 'root': '/home/zeux/lerobot-act-dataset', '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': 'pyav', 'streaming': False}, 'env': None, 'policy': {'type': 'act', 'n_obs_steps': 1, 'input_features': {}, 'output_features': {}, 'device': 'cuda', 'use_amp': True, 'push_to_hub': True, 'repo_id': 'local/lerobot-act-so101-model', '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/lerobot-act_so101_test', 'job_name': 'act', 'resume': False, 'seed': 1000, 'num_workers': 8, 'batch_size': 64, 'steps': 35000, '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-31 13:18:53,862 INFO MainThread:1804411 [wandb_init.py:init():854] starting backend
12
+ 2025-12-31 13:18:54,065 INFO MainThread:1804411 [wandb_init.py:init():857] sending inform_init request
13
+ 2025-12-31 13:18:54,067 INFO MainThread:1804411 [wandb_init.py:init():865] backend started and connected
14
+ 2025-12-31 13:18:54,068 INFO MainThread:1804411 [wandb_init.py:init():936] updated telemetry
15
+ 2025-12-31 13:18:54,069 INFO MainThread:1804411 [wandb_init.py:init():960] communicating run to backend with 90.0 second timeout
16
+ 2025-12-31 13:18:54,935 INFO MainThread:1804411 [wandb_init.py:init():1011] starting run threads in backend
17
+ 2025-12-31 13:18:54,995 INFO MainThread:1804411 [wandb_run.py:_console_start():2506] atexit reg
18
+ 2025-12-31 13:18:54,995 INFO MainThread:1804411 [wandb_run.py:_redirect():2354] redirect: wrap_raw
19
+ 2025-12-31 13:18:54,995 INFO MainThread:1804411 [wandb_run.py:_redirect():2423] Wrapping output streams.
20
+ 2025-12-31 13:18:54,995 INFO MainThread:1804411 [wandb_run.py:_redirect():2446] Redirects installed.
21
+ 2025-12-31 13:18:54,996 INFO MainThread:1804411 [wandb_init.py:init():1049] run started, returning control to user process
22
+ 2025-12-31 16:59:00,732 INFO wandb-AsyncioManager-main:1804411 [service_client.py:_forward_responses():84] Reached EOF.
23
+ 2025-12-31 16:59:00,732 INFO wandb-AsyncioManager-main:1804411 [mailbox.py:close():137] Closing mailbox, abandoning 2 handles.
lerobot-act_so101_test/wandb/run-20251231_131853-8ce8thqh/run-8ce8thqh.wandb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:39d903884c62959749ee1e3cceed6567cfad77a4541b94b21994eb8b12fc9e4a
3
+ size 958256
lerobot_diffusion_model/checkpoints/020000/pretrained_model/config.json ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "diffusion",
3
+ "n_obs_steps": 2,
4
+ "input_features": {
5
+ "observation.state": {
6
+ "type": "STATE",
7
+ "shape": [
8
+ 6
9
+ ]
10
+ },
11
+ "observation.images.top": {
12
+ "type": "VISUAL",
13
+ "shape": [
14
+ 3,
15
+ 480,
16
+ 640
17
+ ]
18
+ }
19
+ },
20
+ "output_features": {
21
+ "action": {
22
+ "type": "ACTION",
23
+ "shape": [
24
+ 6
25
+ ]
26
+ }
27
+ },
28
+ "device": "cuda",
29
+ "use_amp": true,
30
+ "push_to_hub": true,
31
+ "repo_id": "kabilanKB/diffusion-lerobotact",
32
+ "private": null,
33
+ "tags": null,
34
+ "license": null,
35
+ "pretrained_path": null,
36
+ "horizon": 16,
37
+ "n_action_steps": 8,
38
+ "normalization_mapping": {
39
+ "VISUAL": "MEAN_STD",
40
+ "STATE": "MIN_MAX",
41
+ "ACTION": "MIN_MAX"
42
+ },
43
+ "drop_n_last_frames": 7,
44
+ "vision_backbone": "resnet18",
45
+ "crop_shape": [
46
+ 84,
47
+ 84
48
+ ],
49
+ "crop_is_random": true,
50
+ "pretrained_backbone_weights": null,
51
+ "use_group_norm": true,
52
+ "spatial_softmax_num_keypoints": 32,
53
+ "use_separate_rgb_encoder_per_camera": false,
54
+ "down_dims": [
55
+ 512,
56
+ 1024,
57
+ 2048
58
+ ],
59
+ "kernel_size": 5,
60
+ "n_groups": 8,
61
+ "diffusion_step_embed_dim": 128,
62
+ "use_film_scale_modulation": true,
63
+ "noise_scheduler_type": "DDPM",
64
+ "num_train_timesteps": 100,
65
+ "beta_schedule": "squaredcos_cap_v2",
66
+ "beta_start": 0.0001,
67
+ "beta_end": 0.02,
68
+ "prediction_type": "epsilon",
69
+ "clip_sample": true,
70
+ "clip_sample_range": 1.0,
71
+ "num_inference_steps": null,
72
+ "do_mask_loss_for_padding": false,
73
+ "optimizer_lr": 0.0001,
74
+ "optimizer_betas": [
75
+ 0.95,
76
+ 0.999
77
+ ],
78
+ "optimizer_eps": 1e-08,
79
+ "optimizer_weight_decay": 1e-06,
80
+ "scheduler_name": "cosine",
81
+ "scheduler_warmup_steps": 500
82
+ }
lerobot_diffusion_model/checkpoints/020000/pretrained_model/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88e40741113e5628491fda2973d1a1b6a41c92dc9643e41ce5f07d770474f933
3
+ size 1051836312
lerobot_diffusion_model/checkpoints/020000/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
+ 6
13
+ ]
14
+ }
15
+ },
16
+ "norm_map": {
17
+ "VISUAL": "MEAN_STD",
18
+ "STATE": "MIN_MAX",
19
+ "ACTION": "MIN_MAX"
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
+ }
lerobot_diffusion_model/checkpoints/020000/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:a521cd2d101c29b05f1a61273cce28a42771c079c5568672627c9c96c355c7f8
3
+ size 6544
lerobot_diffusion_model/checkpoints/020000/pretrained_model/policy_preprocessor.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ 6
30
+ ]
31
+ },
32
+ "observation.images.top": {
33
+ "type": "VISUAL",
34
+ "shape": [
35
+ 3,
36
+ 480,
37
+ 640
38
+ ]
39
+ },
40
+ "action": {
41
+ "type": "ACTION",
42
+ "shape": [
43
+ 6
44
+ ]
45
+ }
46
+ },
47
+ "norm_map": {
48
+ "VISUAL": "MEAN_STD",
49
+ "STATE": "MIN_MAX",
50
+ "ACTION": "MIN_MAX"
51
+ }
52
+ },
53
+ "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
54
+ }
55
+ ]
56
+ }