File size: 5,389 Bytes
ebed49b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
2025-08-06 09:53:53,189 INFO MainThread:4150232 [wandb_setup.py:_flush():80] Current SDK version is 0.21.0
2025-08-06 09:53:53,189 INFO MainThread:4150232 [wandb_setup.py:_flush():80] Configure stats pid to 4150232
2025-08-06 09:53:53,189 INFO MainThread:4150232 [wandb_setup.py:_flush():80] Loading settings from /home/aye8078/.config/wandb/settings
2025-08-06 09:53:53,189 INFO MainThread:4150232 [wandb_setup.py:_flush():80] Loading settings from /gpfs/home/aye8078/Documents/Github/tac_vla/lerobot/wandb/settings
2025-08-06 09:53:53,189 INFO MainThread:4150232 [wandb_setup.py:_flush():80] Loading settings from environment variables
2025-08-06 09:53:53,189 INFO MainThread:4150232 [wandb_init.py:setup_run_log_directory():703] Logging user logs to outputs/train/pick-blue-cube-single-arm_diff/wandb/run-20250806_095353-4m3ikyrf/logs/debug.log
2025-08-06 09:53:53,189 INFO MainThread:4150232 [wandb_init.py:setup_run_log_directory():704] Logging internal logs to outputs/train/pick-blue-cube-single-arm_diff/wandb/run-20250806_095353-4m3ikyrf/logs/debug-internal.log
2025-08-06 09:53:53,189 INFO MainThread:4150232 [wandb_init.py:init():830] calling init triggers
2025-08-06 09:53:53,189 INFO MainThread:4150232 [wandb_init.py:init():835] wandb.init called with sweep_config: {}
config: {'dataset': {'repo_id': 'a1o/pick-blue-cube_sarm', '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': 'diffusion', 'n_obs_steps': 2, 'normalization_mapping': {'VISUAL': <NormalizationMode.MEAN_STD: 'MEAN_STD'>, 'STATE': <NormalizationMode.MIN_MAX: 'MIN_MAX'>, 'ACTION': <NormalizationMode.MIN_MAX: 'MIN_MAX'>}, 'input_features': {}, 'output_features': {}, 'device': 'cuda', 'use_amp': False, 'push_to_hub': True, 'repo_id': 'a1o/diff_pick-blue-cube_sarm_policy', 'private': None, 'tags': None, 'license': None, 'horizon': 16, 'n_action_steps': 8, 'drop_n_last_frames': 7, 'vision_backbone': 'resnet18', 'crop_shape': [84, 84], 'crop_is_random': True, 'pretrained_backbone_weights': None, 'use_group_norm': True, 'spatial_softmax_num_keypoints': 32, 'use_separate_rgb_encoder_per_camera': False, 'down_dims': [512, 1024, 2048], 'kernel_size': 5, 'n_groups': 8, 'diffusion_step_embed_dim': 128, 'use_film_scale_modulation': True, 'noise_scheduler_type': 'DDPM', 'num_train_timesteps': 100, 'beta_schedule': 'squaredcos_cap_v2', 'beta_start': 0.0001, 'beta_end': 0.02, 'prediction_type': 'epsilon', 'clip_sample': True, 'clip_sample_range': 1.0, 'num_inference_steps': None, 'do_mask_loss_for_padding': False, 'optimizer_lr': 0.0001, 'optimizer_betas': [0.95, 0.999], 'optimizer_eps': 1e-08, 'optimizer_weight_decay': 1e-06, 'scheduler_name': 'cosine', 'scheduler_warmup_steps': 500}, 'output_dir': 'outputs/train/pick-blue-cube-single-arm_diff', 'job_name': 'diff_pick-blue-cube-single-arm', 'resume': False, 'seed': 1000, 'num_workers': 4, 'batch_size': 8, 'steps': 100000, 'eval_freq': 20000, 'log_freq': 200, 'save_checkpoint': True, 'save_freq': 20000, 'use_policy_training_preset': True, 'optimizer': {'type': 'adam', 'lr': 0.0001, 'weight_decay': 1e-06, 'grad_clip_norm': 10.0, 'betas': [0.95, 0.999], 'eps': 1e-08}, 'scheduler': {'type': 'diffuser', 'num_warmup_steps': 500, 'name': 'cosine'}, '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}, '_wandb': {}}
2025-08-06 09:53:53,190 INFO MainThread:4150232 [wandb_init.py:init():871] starting backend
2025-08-06 09:53:53,911 INFO MainThread:4150232 [wandb_init.py:init():874] sending inform_init request
2025-08-06 09:53:53,965 INFO MainThread:4150232 [wandb_init.py:init():882] backend started and connected
2025-08-06 09:53:53,966 INFO MainThread:4150232 [wandb_init.py:init():953] updated telemetry
2025-08-06 09:53:53,966 INFO MainThread:4150232 [wandb_init.py:init():977] communicating run to backend with 90.0 second timeout
2025-08-06 09:53:54,373 INFO MainThread:4150232 [wandb_init.py:init():1029] starting run threads in backend
2025-08-06 09:53:55,933 INFO MainThread:4150232 [wandb_run.py:_console_start():2458] atexit reg
2025-08-06 09:53:55,933 INFO MainThread:4150232 [wandb_run.py:_redirect():2306] redirect: wrap_raw
2025-08-06 09:53:55,933 INFO MainThread:4150232 [wandb_run.py:_redirect():2375] Wrapping output streams.
2025-08-06 09:53:55,933 INFO MainThread:4150232 [wandb_run.py:_redirect():2398] Redirects installed.
2025-08-06 09:53:55,940 INFO MainThread:4150232 [wandb_init.py:init():1075] run started, returning control to user process
2025-08-06 18:01:25,533 INFO MsgRouterThr:4150232 [mailbox.py:close():129] [no run ID] Closing mailbox, abandoning 1 handles.
|