File size: 6,778 Bytes
0395609 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | 2026-04-20 03:35:43,449 INFO MainThread:1208621 [wandb_setup.py:_flush():81] Current SDK version is 0.26.0
2026-04-20 03:35:43,449 INFO MainThread:1208621 [wandb_setup.py:_flush():81] Configure stats pid to 1208621
2026-04-20 03:35:43,449 INFO MainThread:1208621 [wandb_setup.py:_flush():81] Loading settings from environment variables
2026-04-20 03:35:43,449 INFO MainThread:1208621 [wandb_init.py:setup_run_log_directory():721] Logging user logs to outputs/train/SmolVLA_Stacking_3_Cubes/wandb/run-20260420_033543-ckca16hg/logs/debug.log
2026-04-20 03:35:43,449 INFO MainThread:1208621 [wandb_init.py:setup_run_log_directory():722] Logging internal logs to outputs/train/SmolVLA_Stacking_3_Cubes/wandb/run-20260420_033543-ckca16hg/logs/debug-internal.log
2026-04-20 03:35:43,449 INFO MainThread:1208621 [wandb_init.py:init():848] calling init triggers
2026-04-20 03:35:43,449 INFO MainThread:1208621 [wandb_init.py:init():853] wandb.init called with sweep_config: {}
config: {'dataset': {'repo_id': 'TANAY779/SmolVLA_Stacking', 'root': None, 'episodes': None, 'image_transforms': {'enable': False, 'max_num_transforms': 3, 'random_order': False, 'tfs': {'brightness': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'brightness': [0.8, 1.2]}}, 'contrast': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'contrast': [0.8, 1.2]}}, 'saturation': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'saturation': [0.5, 1.5]}}, 'hue': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'hue': [-0.05, 0.05]}}, 'sharpness': {'weight': 1.0, 'type': 'SharpnessJitter', 'kwargs': {'sharpness': [0.5, 1.5]}}, 'affine': {'weight': 1.0, 'type': 'RandomAffine', 'kwargs': {'degrees': [-5.0, 5.0], 'translate': [0.05, 0.05]}}}}, 'revision': None, 'use_imagenet_stats': True, 'video_backend': 'torchcodec', 'return_uint8': False, 'streaming': False}, 'env': None, 'policy': {'type': 'smolvla', 'n_obs_steps': 1, 'input_features': {'observation.state': {'type': <FeatureType.STATE: 'STATE'>, 'shape': [6]}, 'observation.images.camera1': {'type': <FeatureType.VISUAL: 'VISUAL'>, 'shape': [3, 256, 256]}, 'observation.images.camera2': {'type': <FeatureType.VISUAL: 'VISUAL'>, 'shape': [3, 256, 256]}, 'observation.images.camera3': {'type': <FeatureType.VISUAL: 'VISUAL'>, 'shape': [3, 256, 256]}, 'observation.images.empty_camera_0': {'type': <FeatureType.VISUAL: 'VISUAL'>, 'shape': [3, 480, 640]}}, 'output_features': {'action': {'type': <FeatureType.ACTION: 'ACTION'>, 'shape': [6]}}, 'device': 'cuda', 'use_amp': False, 'use_peft': False, 'push_to_hub': True, 'repo_id': 'TANAY779/SmolVLA_Stacking_3_Cubes', 'private': None, 'tags': None, 'license': None, 'pretrained_path': 'outputs/train/SmolVLA_Stacking_3_Cubes/checkpoints/012000/pretrained_model', 'chunk_size': 50, 'n_action_steps': 50, 'normalization_mapping': {'VISUAL': <NormalizationMode.IDENTITY: 'IDENTITY'>, 'STATE': <NormalizationMode.MEAN_STD: 'MEAN_STD'>, 'ACTION': <NormalizationMode.MEAN_STD: 'MEAN_STD'>}, 'max_state_dim': 32, 'max_action_dim': 32, 'resize_imgs_with_padding': [512, 512], 'empty_cameras': 1, 'adapt_to_pi_aloha': False, 'use_delta_joint_actions_aloha': False, 'tokenizer_max_length': 48, 'num_steps': 10, 'use_cache': True, 'freeze_vision_encoder': True, 'train_expert_only': True, 'train_state_proj': True, 'optimizer_lr': 0.0001, 'optimizer_betas': [0.9, 0.95], 'optimizer_eps': 1e-08, 'optimizer_weight_decay': 1e-10, 'optimizer_grad_clip_norm': 10.0, 'scheduler_warmup_steps': 1000, 'scheduler_decay_steps': 30000, 'scheduler_decay_lr': 2.5e-06, 'vlm_model_name': 'HuggingFaceTB/SmolVLM2-500M-Video-Instruct', 'load_vlm_weights': True, 'add_image_special_tokens': False, 'attention_mode': 'cross_attn', 'prefix_length': 0, 'pad_language_to': 'max_length', 'num_expert_layers': 0, 'num_vlm_layers': 16, 'self_attn_every_n_layers': 2, 'expert_width_multiplier': 0.75, 'min_period': 0.004, 'max_period': 4.0, 'rtc_config': None, 'compile_model': False, 'compile_mode': 'max-autotune'}, 'output_dir': 'outputs/train/SmolVLA_Stacking_3_Cubes', 'job_name': 'SmolVLA_Stacking_3_Cubes', 'resume': True, 'seed': 1000, 'cudnn_deterministic': False, 'num_workers': 4, 'batch_size': 48, 'prefetch_factor': 4, 'persistent_workers': True, 'steps': 20000, 'eval_freq': 2000, 'log_freq': 100, 'tolerance_s': 0.0001, 'save_checkpoint': True, 'save_freq': 1000, 'use_policy_training_preset': True, 'optimizer': {'type': 'adamw', 'lr': 0.0001, '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': 0.0001, 'decay_lr': 2.5e-06}, 'eval': {'n_episodes': 50, 'batch_size': 50, 'use_async_envs': True}, 'wandb': {'enable': True, 'disable_artifact': False, 'project': 'lerobot', 'entity': None, 'notes': None, 'run_id': 'ckca16hg', 'mode': None, 'add_tags': True}, 'peft': None, 'use_rabc': False, 'rabc_progress_path': None, 'rabc_kappa': 0.01, 'rabc_epsilon': 1e-06, 'rabc_head_mode': 'sparse', 'rename_map': {'observation.images.overhead': 'observation.images.camera1', 'observation.images.wrist': 'observation.images.camera2'}, 'checkpoint_path': 'outputs/train/SmolVLA_Stacking_3_Cubes/checkpoints/012000', '_wandb': {}}
2026-04-20 03:35:43,449 INFO MainThread:1208621 [wandb_init.py:init():896] starting backend
2026-04-20 03:35:43,663 INFO MainThread:1208621 [wandb_init.py:init():911] sending inform_init request
2026-04-20 03:35:43,667 INFO MainThread:1208621 [wandb_init.py:init():919] backend started and connected
2026-04-20 03:35:43,669 INFO MainThread:1208621 [wandb_init.py:init():989] updated telemetry
2026-04-20 03:35:43,674 INFO MainThread:1208621 [wandb_init.py:init():1013] communicating run to backend with 90.0 second timeout
2026-04-20 03:35:45,044 INFO MainThread:1208621 [wandb_init.py:init():1053] run resumed
2026-04-20 03:35:45,046 INFO MainThread:1208621 [wandb_init.py:init():1058] starting run threads in backend
2026-04-20 03:35:45,098 INFO MainThread:1208621 [wandb_run.py:_console_start():2542] atexit reg
2026-04-20 03:35:45,098 INFO MainThread:1208621 [wandb_run.py:_redirect():2391] redirect: wrap_raw
2026-04-20 03:35:45,098 INFO MainThread:1208621 [wandb_run.py:_redirect():2460] Wrapping output streams.
2026-04-20 03:35:45,098 INFO MainThread:1208621 [wandb_run.py:_redirect():2483] Redirects installed.
2026-04-20 03:35:45,101 INFO MainThread:1208621 [wandb_init.py:init():1098] run started, returning control to user process
2026-04-20 08:27:11,376 INFO wandb-AsyncioManager-main:1208621 [service_client.py:_forward_responses():134] Reached EOF.
2026-04-20 08:27:11,376 INFO wandb-AsyncioManager-main:1208621 [mailbox.py:close():155] Closing mailbox, abandoning 2 handles.
|