wandb/debug.log

2025-04-13 08:36:40,398 INFO    MainThread:68651 [wandb_setup.py:_flush():67] Current SDK version is 0.19.9
2025-04-13 08:36:40,398 INFO    MainThread:68651 [wandb_setup.py:_flush():67] Configure stats pid to 68651
2025-04-13 08:36:40,398 INFO    MainThread:68651 [wandb_setup.py:_flush():67] Loading settings from /root/.config/wandb/settings
2025-04-13 08:36:40,398 INFO    MainThread:68651 [wandb_setup.py:_flush():67] Loading settings from /root/lerobot/wandb/settings
2025-04-13 08:36:40,398 INFO    MainThread:68651 [wandb_setup.py:_flush():67] Loading settings from environment variables
2025-04-13 08:36:40,398 INFO    MainThread:68651 [wandb_init.py:setup_run_log_directory():662] Logging user logs to /scratch/train/pi0_extended_dataset_with_multids/wandb/run-20250413_083640-extended_dataset_with_multids/logs/debug.log
2025-04-13 08:36:40,398 INFO    MainThread:68651 [wandb_init.py:setup_run_log_directory():663] Logging internal logs to /scratch/train/pi0_extended_dataset_with_multids/wandb/run-20250413_083640-extended_dataset_with_multids/logs/debug-internal.log
2025-04-13 08:36:40,398 INFO    MainThread:68651 [wandb_init.py:init():781] calling init triggers
2025-04-13 08:36:40,398 INFO    MainThread:68651 [wandb_init.py:init():786] wandb.init called with sweep_config: {}
config: {'dataset': {'repo_id': '[maelic/hackathon7,maelic/hackathon8,maelic/hackathon21,maelic/hackathon10,maelic/hackathon_pepper_3,maelic/hackathon_pepper_2,maelic/hackathon4]', 'root': None, 'episodes': None, 'image_transforms': {'enable': False, 'max_num_transforms': 3, 'random_order': False, 'tfs': {'brightness': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'brightness': [0.8, 1.2]}}, 'contrast': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'contrast': [0.8, 1.2]}}, 'saturation': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'saturation': [0.5, 1.5]}}, 'hue': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'hue': [-0.05, 0.05]}}, 'sharpness': {'weight': 1.0, 'type': 'SharpnessJitter', 'kwargs': {'sharpness': [0.5, 1.5]}}}}, 'revision': None, 'use_imagenet_stats': True, 'video_backend': 'torchcodec'}, 'env': None, 'policy': {'type': 'pi0', 'n_obs_steps': 1, 'normalization_mapping': {'VISUAL': <NormalizationMode.IDENTITY: 'IDENTITY'>, 'STATE': <NormalizationMode.MEAN_STD: 'MEAN_STD'>, 'ACTION': <NormalizationMode.MEAN_STD: 'MEAN_STD'>}, 'input_features': {}, 'output_features': {}, 'device': 'cuda', 'use_amp': False, 'chunk_size': 50, 'n_action_steps': 50, 'max_state_dim': 32, 'max_action_dim': 32, 'resize_imgs_with_padding': [224, 224], 'empty_cameras': 0, 'adapt_to_pi_aloha': False, 'use_delta_joint_actions_aloha': False, 'tokenizer_max_length': 48, 'proj_width': 1024, 'num_steps': 10, 'use_cache': True, 'attention_implementation': 'eager', 'freeze_vision_encoder': True, 'train_expert_only': False, 'train_state_proj': True, 'optimizer_lr': 2.5e-05, 'optimizer_betas': [0.9, 0.95], 'optimizer_eps': 1e-08, 'optimizer_weight_decay': 1e-10, 'scheduler_warmup_steps': 1000, 'scheduler_decay_steps': 30000, 'scheduler_decay_lr': 2.5e-06}, 'output_dir': '/scratch/train/pi0_extended_dataset_with_multids', 'job_name': 'pi0', 'resume': False, 'seed': 1000, 'num_workers': 4, 'batch_size': 20, 'steps': 100000, 'eval_freq': 1000, 'log_freq': 200, 'save_checkpoint': True, 'save_freq': 1000, 'use_policy_training_preset': True, 'optimizer': {'type': 'adamw', 'lr': 2.5e-05, 'weight_decay': 1e-10, 'grad_clip_norm': 10.0, 'betas': [0.9, 0.95], 'eps': 1e-08}, 'scheduler': {'type': 'cosine_decay_with_warmup', 'num_warmup_steps': 1000, 'num_decay_steps': 30000, 'peak_lr': 2.5e-05, 'decay_lr': 2.5e-06}, 'eval': {'n_episodes': 50, 'batch_size': 50, 'use_async_envs': False}, 'wandb': {'enable': True, 'disable_artifact': True, 'project': 'pi0_mistral_hackathon', 'entity': None, 'notes': None, 'run_id': 'extended_dataset_with_multids', 'mode': None}, '_wandb': {}}
2025-04-13 08:36:40,398 INFO    MainThread:68651 [wandb_init.py:init():809] starting backend
2025-04-13 08:36:40,398 INFO    MainThread:68651 [wandb_init.py:init():813] sending inform_init request
2025-04-13 08:36:40,401 INFO    MainThread:68651 [backend.py:_multiprocessing_setup():101] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
2025-04-13 08:36:40,401 INFO    MainThread:68651 [wandb_init.py:init():823] backend started and connected
2025-04-13 08:36:40,402 INFO    MainThread:68651 [wandb_init.py:init():915] updated telemetry
2025-04-13 08:36:40,405 INFO    MainThread:68651 [wandb_init.py:init():939] communicating run to backend with 90.0 second timeout
2025-04-13 08:36:40,807 INFO    MainThread:68651 [wandb_init.py:init():1014] starting run threads in backend
2025-04-13 08:36:40,871 INFO    MainThread:68651 [wandb_run.py:_console_start():2454] atexit reg
2025-04-13 08:36:40,871 INFO    MainThread:68651 [wandb_run.py:_redirect():2306] redirect: wrap_raw
2025-04-13 08:36:40,872 INFO    MainThread:68651 [wandb_run.py:_redirect():2371] Wrapping output streams.
2025-04-13 08:36:40,872 INFO    MainThread:68651 [wandb_run.py:_redirect():2394] Redirects installed.
2025-04-13 08:36:40,873 INFO    MainThread:68651 [wandb_init.py:init():1056] run started, returning control to user process