|
- GPU 0 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 1 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 2 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 3 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- |
|
Initializing training for seed 242 |
|
- GPU 0 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 1 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 2 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 3 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 0 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 1 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 2 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 3 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 0 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 1 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 2 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 3 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 0 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 1 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 2 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 3 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 0 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 0 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 1 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 1 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 2 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 2 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 3 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 3 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- Training config for seed 242: |
|
{'callbacks': {'rollout_lh': {'tasks': {'_target_': 'calvin_env.envs.tasks.Tasks', 'tasks': {'rotate_red_block_right': , 'rotate_red_block_left': , 'rotate_blue_block_right': , 'rotate_blue_block_left': , 'rotate_pink_block_right': , 'rotate_pink_block_left': , 'push_red_block_right': , 'push_red_block_left': , 'push_blue_block_right': , 'push_blue_block_left': , 'push_pink_block_right': , 'push_pink_block_left': , 'move_slider_left': , 'move_slider_right': , 'open_drawer': , 'close_drawer': , 'lift_red_block_table': , 'lift_red_block_slider': , 'lift_red_block_drawer': , 'lift_blue_block_table': , 'lift_blue_block_slider': , 'lift_blue_block_drawer': , 'lift_pink_block_table': , 'lift_pink_block_slider': , 'lift_pink_block_drawer': , 'place_in_slider': , 'place_in_drawer': , 'stack_block': , 'unstack_block': , 'turn_on_lightbulb': , 'turn_off_lightbulb': , 'turn_on_led': , 'turn_off_led': , 'push_into_drawer': }}, 'val_annotations': {'rotate_red_block_right': , 'rotate_red_block_left': , 'rotate_blue_block_right': , 'rotate_blue_block_left': , 'rotate_pink_block_right': , 'rotate_pink_block_left': , 'push_red_block_right': , 'push_red_block_left': , 'push_blue_block_right': , 'push_blue_block_left': , 'push_pink_block_right': , 'push_pink_block_left': , 'move_slider_left': , 'move_slider_right': , 'open_drawer': , 'close_drawer': , 'lift_red_block_table': , 'lift_blue_block_table': , 'lift_pink_block_table': , 'lift_red_block_slider': , 'lift_blue_block_slider': , 'lift_pink_block_slider': , 'lift_red_block_drawer': , 'lift_blue_block_drawer': , 'lift_pink_block_drawer': , 'place_in_slider': , 'place_in_drawer': , 'push_into_drawer': , 'stack_block': , 'unstack_block': , 'turn_on_lightbulb': , 'turn_off_lightbulb': , 'turn_on_led': , 'turn_off_led': }, '_target_': 'flower.rollout.rollout_long_horizon.RolloutLongHorizon', '_recursive_': False, 'env_cfg': {'_target_': 'flower.wrappers.hulc_wrapper.HulcWrapper'}, 'skip_epochs': '${rollout_lh_skip_epochs}', 'rollout_freq': 5, 'num_videos': 0, 'num_sequences': 1000, 'replan_freq': 30, 'ep_len': 360, 'empty_cache': False, 'log_video_to_file': False, 'save_dir': './videos', 'lang_folder': '${lang_folder}', 'debug': False}, 'ema': {'_target_': 'flower.callbacks.ema.EMA', 'decay': 0.999, 'start_step': 0, 'save_ema_weights_in_callback_state': True, 'evaluate_ema_weights_instead': True, 'power': 0.6666666666666666, 'inv_gamma': 1.0, 'min_value': 0.0, 'max_value': 0.9999}, 'checkpoint': {'_target_': 'pytorch_lightning.callbacks.ModelCheckpoint', 'save_top_k': 1, 'verbose': True, 'monitor': 'eval_lh/avg_seq_len', 'mode': 'max', 'dirpath': 'saved_models', 'filename': '{epoch:02d}_{eval_lh/avg_seq_len:.2f}', 'every_n_epochs': '${callbacks.rollout_lh.rollout_freq}'}}, 'datamodule': {'transforms': {'train': {'rgb_static': , 'rgb_gripper': }, 'val': {'rgb_static': , 'rgb_gripper': }}, '_target_': 'flower.datasets.hulc_data_module.HulcDataModule', '_recursive_': False, 'root_data_dir': '${root_data_dir}', 'action_space': 7, 'action_max': , 'action_min': , 'shuffle_val': False, 'observation_space': {'rgb_obs': , 'depth_obs': , 'state_obs': , 'actions': , 'language': }, 'proprioception_dims': {'n_state_obs': 8, 'keep_indices': , 'robot_orientation_idx': , 'normalize': True, 'normalize_robot_orientation': True}, 'datasets': {'lang_dataset': {'_target_': 'flower.datasets.disk_dataset.ExtendedDiskDataset', 'key': 'lang', 'save_format': 'npz', 'batch_size': '${batch_size}', 'min_window_size': '${act_seq_len}', 'max_window_size': '${act_seq_len}', 'proprio_state': '${datamodule.proprioception_dims}', 'obs_space': '${datamodule.observation_space}', 'skip_frames': 1, 'pad': False, 'lang_folder': '${lang_folder}', 'aux_lang_loss_window': 8, 'num_workers': '${num_workers}', 'action_seq_len': '${act_seq_len}', 'obs_seq_len': '${obs_seq_len}', 'future_range': 1, 'use_extracted_rel_actions': '${use_extracted_rel_actions}'}}}, 'model': {'_target_': 'flower.models.flower.FLOWERVLA', '_recursive_': False, 'vlm_path': 'microsoft/Florence-2-large', 'freeze_florence': False, 'freeze_vision_tower': False, 'vlm_prompt_style': 'default', 'token_dropout': 0.1, 'multistep': '${multistep}', 'num_sampling_steps': 4, 'lowdim_obs_dim': 7, 'action_dim': 7, 'act_window_size': 10, 'load_pretrained': True, 'pretrained_model_path': '/home/hk-project-sustainebot/ft4740/code/flower_vla_policy/logs/runs/2025-02-05/10-17-02/360000_model_weights.pt', 'use_second_view': True, 'second_view_key': 'image_wrist', 'action_type_adaln': True, 'use_causal_attention': True, 'use_cross_attn': True, 'use_adaln_cond': False, 'use_readout_token': False, 'use_proprio': False, 'return_act_chunk': False, 'sampling_type': 'uniform', 'dit_dim': 1024, 'n_heads': 16, 'n_layers': 18, 'attn_pdrop': 0.1, 'resid_pdrop': 0.1, 'mlp_pdrop': 0.1, 'use_rope': True, 'use_nope': False, 'query_seq_len': 100, 'rope_theta': 32.0, 'optimizer_type': 'adamw', 'optimizer': {'_target_': 'torch.optim.AdamW', 'transformer_weight_decay': 0.05, 'learning_rate': 2e-05, 'betas': }, 'lr_scheduler': {'lr_scheduler': {'init_lr': 2e-05, 'init_lr_scale': 0.1, 'final_lr_scale': 0.1, 'total_steps': 50000, 'phase_ratio': '(0.05, 0.1, 0.85)', 'lr': 5e-05}}}, 'root_data_dir': '/hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D', 'lang_folder': 'lang_clip_resnet50', 'log_dir': './logs', 'slurm': False, 'seed': 242, 'device': 'cuda', 'batch_size': 8, 'devices': 4, 'act_dim': 7, 'proprio_dims': 7, 'obs_dim': 512, 'goal_dim': 512, 'obs_seq_len': 1, 'act_seq_len': 10, 'multistep': 10, 'p_last_state': 0, 'gen_img_res': 112, 'max_epochs': 35, 'rollout_lh_skip_epochs': 9, 'num_workers': 12, 'benchmark_name': 'calvin_abc', 'use_extracted_rel_actions': True, 'trainer': {'devices': '${devices}', 'precision': 'bf16-mixed', 'max_epochs': '${max_epochs}', 'sync_batchnorm': True, 'accelerator': 'gpu', 'strategy': 'ddp', 'limit_train_batches': 1000, 'limit_val_batches': 4}, 'logger': {'_target_': 'pytorch_lightning.loggers.WandbLogger', 'save_dir': '.', 'name': '2025-02-20/17-55-09_seed242', 'group': '2025-02-20', 'log_model': False, 'project': '${benchmark_name}', 'entity': 'bennoq', 'id': '2025-02-20_17-55-09_seed242'}} |
|
- Git commit: 0cc472e37ca44ca89bf334e692a2dee801bb8e91 |
|
- * CUDA: |
|
- GPU: |
|
- NVIDIA A100-SXM4-40GB |
|
- NVIDIA A100-SXM4-40GB |
|
- NVIDIA A100-SXM4-40GB |
|
- NVIDIA A100-SXM4-40GB |
|
- available: True |
|
- version: 12.1 |
|
* Packages: |
|
- numpy: 1.26.4 |
|
- pyTorch_debug: False |
|
- pyTorch_version: 2.2.2+cu121 |
|
- pytorch-lightning: 2.0.8 |
|
- tqdm: 4.67.1 |
|
- GPU 0 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 1 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 2 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 3 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- loading dataset at /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/training |
|
- loading dataset at /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/training |
|
- loading dataset at /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/training |
|
- finished loading dataset |
|
- finished loading dataset |
|
- finished loading dataset |
|
- loading dataset at /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/training |
|
- finished loading dataset |
|
- Extracted files loaded from /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/training/extracted |
|
- Extracted files loaded from /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/training/extracted |
|
- Extracted files loaded from /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/training/extracted |
|
- Extracted files loaded from /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/training/extracted |
|
- loading dataset at /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/validation |
|
- finished loading dataset |
|
- loading dataset at /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/validation |
|
- loading dataset at /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/validation |
|
- finished loading dataset |
|
- loading dataset at /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/validation |
|
- finished loading dataset |
|
- finished loading dataset |
|
- Extracted files loaded from /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/validation/extracted |
|
- Extracted files loaded from /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/validation/extracted |
|
- Extracted files loaded from /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/validation/extracted |
|
- Extracted files loaded from /hkfs/work/workspace/scratch/ft4740-play3/task_ABC_D/validation/extracted |
|
- EGL_DEVICE_ID 0 <==> CUDA_DEVICE_ID 0 |
|
- Loading EGL plugin (may segfault on misconfigured systems)... |
|
- Successfully loaded egl plugin |
|
- Connected to server with id: 0 |
|
- Connected to server with id: 0 |
|
- Loading robot |
|
- EGL_DEVICE_ID 1 <==> CUDA_DEVICE_ID 1 |
|
- Loading EGL plugin (may segfault on misconfigured systems)... |
|
- Resetting simulation |
|
- Setting gravity |
|
- Successfully loaded egl plugin |
|
- Connected to server with id: 0 |
|
- Connected to server with id: 0 |
|
- Loading robot |
|
- Resetting simulation |
|
- Setting gravity |
|
- Initialized PlayTableEnv for device cuda:0 |
|
- Start generating evaluation sequences. |
|
- EGL_DEVICE_ID 2 <==> CUDA_DEVICE_ID 2 |
|
- Loading EGL plugin (may segfault on misconfigured systems)... |
|
- Initialized PlayTableEnv for device cuda:1 |
|
- Start generating evaluation sequences. |
|
- EGL_DEVICE_ID 3 <==> CUDA_DEVICE_ID 3 |
|
- Successfully loaded egl plugin |
|
- Connected to server with id: 0 |
|
- Connected to server with id: 0 |
|
- Loading robot |
|
- Resetting simulation |
|
- Setting gravity |
|
- Loading EGL plugin (may segfault on misconfigured systems)... |
|
- Successfully loaded egl plugin |
|
- Connected to server with id: 0 |
|
- Connected to server with id: 0 |
|
- Loading robot |
|
- Resetting simulation |
|
- Setting gravity |
|
- Initialized PlayTableEnv for device cuda:2 |
|
- Start generating evaluation sequences. |
|
- Initialized PlayTableEnv for device cuda:3 |
|
- Start generating evaluation sequences. |
|
- Done generating evaluation sequences. |
|
- Done generating evaluation sequences. |
|
- Done generating evaluation sequences. |
|
- Done generating evaluation sequences. |
|
- Creating EMA weights copy. |
|
- Creating EMA weights copy. |
|
- Creating EMA weights copy. |
|
- Creating EMA weights copy. |
|
- 1 / 5 subtasks: 994 / 1000 sequences, SR: 99.4% |
|
- 2 / 5 subtasks: 940 / 1000 sequences, SR: 94.0% |
|
- 3 / 5 subtasks: 857 / 1000 sequences, SR: 85.7% |
|
- 4 / 5 subtasks: 760 / 1000 sequences, SR: 76.0% |
|
- 5 / 5 subtasks: 661 / 1000 sequences, SR: 66.1% |
|
- Average successful sequence length: 4.2 |
|
- 1 / 5 subtasks: 996 / 1000 sequences, SR: 99.6% |
|
- 2 / 5 subtasks: 945 / 1000 sequences, SR: 94.5% |
|
- 3 / 5 subtasks: 859 / 1000 sequences, SR: 85.9% |
|
- 4 / 5 subtasks: 764 / 1000 sequences, SR: 76.4% |
|
- 5 / 5 subtasks: 681 / 1000 sequences, SR: 68.1% |
|
- Average successful sequence length: 4.2 |
|
- 1 / 5 subtasks: 997 / 1000 sequences, SR: 99.7% |
|
- 2 / 5 subtasks: 962 / 1000 sequences, SR: 96.2% |
|
- 3 / 5 subtasks: 890 / 1000 sequences, SR: 89.0% |
|
- 4 / 5 subtasks: 820 / 1000 sequences, SR: 82.0% |
|
- 5 / 5 subtasks: 736 / 1000 sequences, SR: 73.6% |
|
- Average successful sequence length: 4.4 |
|
- 1 / 5 subtasks: 997 / 1000 sequences, SR: 99.7% |
|
- 2 / 5 subtasks: 965 / 1000 sequences, SR: 96.5% |
|
- 3 / 5 subtasks: 900 / 1000 sequences, SR: 90.0% |
|
- 4 / 5 subtasks: 826 / 1000 sequences, SR: 82.6% |
|
- 5 / 5 subtasks: 750 / 1000 sequences, SR: 75.0% |
|
- Average successful sequence length: 4.4 |
|
- 1 / 5 subtasks: 997 / 1000 sequences, SR: 99.7% |
|
- 2 / 5 subtasks: 972 / 1000 sequences, SR: 97.2% |
|
- 3 / 5 subtasks: 921 / 1000 sequences, SR: 92.1% |
|
- 4 / 5 subtasks: 862 / 1000 sequences, SR: 86.2% |
|
- 5 / 5 subtasks: 785 / 1000 sequences, SR: 78.5% |
|
- Average successful sequence length: 4.5 |
|
- 1 / 5 subtasks: 993 / 1000 sequences, SR: 99.3% |
|
- 2 / 5 subtasks: 959 / 1000 sequences, SR: 95.9% |
|
- 3 / 5 subtasks: 905 / 1000 sequences, SR: 90.5% |
|
- 4 / 5 subtasks: 848 / 1000 sequences, SR: 84.8% |
|
- 5 / 5 subtasks: 775 / 1000 sequences, SR: 77.5% |
|
- Average successful sequence length: 4.5 |
|
- GPU 0 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 1 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 2 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 3 Memory: Allocated: 3.73GB, Reserved: 3.83GB |
|
- GPU 0 Memory: Allocated: 3.73GB, Reserved: 3.81GB |
|
- GPU 1 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 2 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 3 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 0 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 1 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 2 Memory: Allocated: 3.73GB, Reserved: 3.81GB |
|
- GPU 3 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 0 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 1 Memory: Allocated: 3.73GB, Reserved: 3.83GB |
|
- GPU 2 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
- GPU 3 Memory: Allocated: 0.00GB, Reserved: 0.00GB |
|
|