CHYang25 commited on
Commit
e8f7e2a
·
verified ·
1 Parent(s): 419234d

Upload folder using huggingface_hub

Browse files
Files changed (45) hide show
  1. .gitattributes +4 -0
  2. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/.hydra/config.yaml +163 -0
  3. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/.hydra/hydra.yaml +155 -0
  4. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/.hydra/overrides.yaml +1 -0
  5. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0370-test_success_rate=0.920.ckpt +3 -0
  6. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0555-test_success_rate=0.900.ckpt +3 -0
  7. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0670-test_success_rate=0.900.ckpt +3 -0
  8. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0710-test_success_rate=0.920.ckpt +3 -0
  9. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0765-test_success_rate=0.900.ckpt +3 -0
  10. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/checkpoints/latest.ckpt +3 -0
  11. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/logs.json.txt +3 -0
  12. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/train.log +8 -0
  13. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/debug-internal.log +16 -0
  14. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/debug.log +27 -0
  15. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/files/config.yaml +271 -0
  16. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/files/output.log +3 -0
  17. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/files/wandb-metadata.json +107 -0
  18. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/files/wandb-summary.json +1 -0
  19. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/logs/debug-core.log +14 -0
  20. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/logs/debug-internal.log +16 -0
  21. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/logs/debug.log +27 -0
  22. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/run-y2drrt7x.wandb +3 -0
  23. 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/wandb-resume.json +1 -0
  24. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/.hydra/config.yaml +163 -0
  25. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/.hydra/hydra.yaml +155 -0
  26. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/.hydra/overrides.yaml +1 -0
  27. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0245-test_success_rate=0.900.ckpt +3 -0
  28. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0355-test_success_rate=0.860.ckpt +3 -0
  29. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0710-test_success_rate=0.880.ckpt +3 -0
  30. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0735-test_success_rate=0.900.ckpt +3 -0
  31. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0905-test_success_rate=0.880.ckpt +3 -0
  32. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/checkpoints/latest.ckpt +3 -0
  33. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/logs.json.txt +3 -0
  34. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/train.log +8 -0
  35. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/debug-internal.log +16 -0
  36. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/debug.log +27 -0
  37. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/files/config.yaml +271 -0
  38. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/files/output.log +3 -0
  39. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/files/wandb-metadata.json +107 -0
  40. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/files/wandb-summary.json +1 -0
  41. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/logs/debug-core.log +14 -0
  42. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/logs/debug-internal.log +16 -0
  43. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/logs/debug.log +27 -0
  44. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/run-318q5q4f.wandb +3 -0
  45. 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/wandb-resume.json +1 -0
.gitattributes CHANGED
@@ -116,3 +116,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
116
  2026.02.24/23.25.46_train_llmbc_lowdim_box-close-v2/wandb/run-20260224_232605-x2fbtqtp/run-x2fbtqtp.wandb filter=lfs diff=lfs merge=lfs -text
117
  2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/logs.json.txt filter=lfs diff=lfs merge=lfs -text
118
  2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/run-4cc5hfsj.wandb filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
116
  2026.02.24/23.25.46_train_llmbc_lowdim_box-close-v2/wandb/run-20260224_232605-x2fbtqtp/run-x2fbtqtp.wandb filter=lfs diff=lfs merge=lfs -text
117
  2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/logs.json.txt filter=lfs diff=lfs merge=lfs -text
118
  2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/run-4cc5hfsj.wandb filter=lfs diff=lfs merge=lfs -text
119
+ 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/logs.json.txt filter=lfs diff=lfs merge=lfs -text
120
+ 2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/run-y2drrt7x.wandb filter=lfs diff=lfs merge=lfs -text
121
+ 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/logs.json.txt filter=lfs diff=lfs merge=lfs -text
122
+ 2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/run-318q5q4f.wandb filter=lfs diff=lfs merge=lfs -text
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/.hydra/config.yaml ADDED
@@ -0,0 +1,163 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: train_llmbc_lowdim
2
+ _target_: llmbc.workspace.train_llmbc_lowdim_workspace.TrainLLMBCLowdimWorkspace
3
+ obs_dim: ${task.obs_dim}
4
+ action_dim: ${task.action_dim}
5
+ task_name: ${task.name}
6
+ exp_name: default
7
+ model_name: ${llm.name}
8
+ horizon: 1
9
+ n_obs_steps: 1
10
+ n_action_steps: 1
11
+ n_latency_steps: 0
12
+ past_action_visible: false
13
+ llm_orig_expert_feedback: true
14
+ llm_do_sample: false
15
+ policy:
16
+ _target_: llmbc.policy.llmbc_lowdim_policy.LLMBCLowdimPolicy
17
+ model:
18
+ _target_: llmbc.model.policy.policy_mlp.PolicyMLP
19
+ input_size: ${eval:'${n_obs_steps}*${obs_dim}'}
20
+ hidden_size:
21
+ - 256
22
+ - 256
23
+ output_size: ${eval:'${n_action_steps}*${action_dim}'}
24
+ activation: relu
25
+ n_obs_steps: ${n_obs_steps}
26
+ n_action_steps: ${n_action_steps}
27
+ obs_dim: ${obs_dim}
28
+ action_dim: ${action_dim}
29
+ llm_discriminator:
30
+ _target_: llmbc.discriminator.llm_ce_discriminator.LLMCEDiscriminator
31
+ task_id: ${task_name}
32
+ llm_translator:
33
+ _target_: llmbc.translator.llm_translator.LLMTranslator
34
+ cfg: ${llm}
35
+ obs_dim: ${task.obs_dim}
36
+ action_dim: ${task.action_dim}
37
+ horizon: ${horizon}
38
+ n_obs_steps: ${n_obs_steps}
39
+ n_action_steps: ${n_action_steps}
40
+ loss_bc_weight: 1.0
41
+ loss_llm_weight: 0.001
42
+ horizon: ${horizon}
43
+ n_obs_steps: ${n_obs_steps}
44
+ n_action_steps: ${n_action_steps}
45
+ normalize_llm_loss: true
46
+ dataloader:
47
+ batch_size: 16
48
+ num_workers: 0
49
+ shuffle: true
50
+ pin_memory: false
51
+ persistent_workers: false
52
+ val_dataloader:
53
+ batch_size: 16
54
+ num_workers: 0
55
+ shuffle: true
56
+ pin_memory: false
57
+ persistent_workers: false
58
+ optimizer:
59
+ _target_: torch.optim.AdamW
60
+ lr: 0.01
61
+ betas:
62
+ - 0.95
63
+ - 0.999
64
+ eps: 1.0e-08
65
+ weight_decay: 1.0e-06
66
+ training:
67
+ device: cuda:0
68
+ seed: 45
69
+ debug: false
70
+ resume: false
71
+ lr_scheduler: cosine
72
+ lr_warmup_steps: 10
73
+ num_epochs: 1001
74
+ gradient_accumulate_every: 8
75
+ grad_norm_clip: 0.5
76
+ rollout_every: 5
77
+ checkpoint_every: 5
78
+ val_every: 1
79
+ sample_every: 5
80
+ sample_max_batch: 128
81
+ max_train_steps: null
82
+ max_val_steps: null
83
+ tqdm_interval_sec: 1.0
84
+ logging:
85
+ project: ${task.name}-training-llm-backbone
86
+ resume: true
87
+ mode: online
88
+ name: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
89
+ tags:
90
+ - ${name}
91
+ - ${task_name}
92
+ - ${exp_name}
93
+ id: null
94
+ group: null
95
+ checkpoint:
96
+ topk:
97
+ monitor_key: test_success_rate
98
+ mode: max
99
+ k: 5
100
+ format_str: epoch={epoch:04d}-test_success_rate={test_success_rate:.3f}.ckpt
101
+ save_last_ckpt: true
102
+ save_last_snapshot: false
103
+ multi_run:
104
+ run_dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
105
+ wandb_name_base: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
106
+ task:
107
+ name: box-close-v2
108
+ obs_dim: 9
109
+ action_dim: 4
110
+ env_runner:
111
+ _target_: llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner
112
+ env_name: llf-metaworld-box-close-v2
113
+ n_train: 10
114
+ n_test: 50
115
+ n_envs: 10
116
+ max_steps: 30
117
+ n_obs_steps: ${n_obs_steps}
118
+ n_action_steps: ${n_action_steps}
119
+ instruction_type: b
120
+ feedback_type:
121
+ - hp
122
+ - hn
123
+ - fp
124
+ visual: false
125
+ discount: 0.9
126
+ dataset:
127
+ _target_: llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset
128
+ data_path: datasets/box-close-v2-o4-mini.pt
129
+ data_path2: datasets/box-close-v2-o4-mini.pt
130
+ horizon: ${horizon}
131
+ pad_before: ${eval:'${n_obs_steps}-1'}
132
+ pad_after: ${eval:'${n_action_steps}-1'}
133
+ obs_eef_target: true
134
+ use_manual_normalizer: false
135
+ val_ratio: 0.1
136
+ dummy_normalizer: true
137
+ instructor:
138
+ _target_: llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor
139
+ llm:
140
+ name: Qwen/Qwen2-0.5B-Instruct
141
+ model_name: Qwen2-0.5B-Instruct
142
+ config_target: llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2Config
143
+ causal_lm_target: llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2ForCausalLM
144
+ use_quantization: false
145
+ use_joint_mlp_projector: true
146
+ llm_mode: ete-finetuned
147
+ finetune_mode: orig
148
+ checkpoint: data/outputs/2026.02.20/17.05.08_train_llm_lowdim_box-close-v2/Qwen/Qwen2-0.5B-Instruct-finetuned-box-close-v2/checkpoint-17670
149
+ max_length: 100
150
+ lora_config:
151
+ r: 32
152
+ lora_alpha: 64
153
+ lora_dropout: 0.05
154
+ bias: none
155
+ task_type: CAUSAL_LM
156
+ prompter:
157
+ _target_: llmbc.translator.prompter.qwen2_prompter.Qwen2Prompter
158
+ use_joint_mlp_projector: true
159
+ hydra:
160
+ job:
161
+ override_dirname: ${model_name}
162
+ run:
163
+ dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${model_name}
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/.hydra/hydra.yaml ADDED
@@ -0,0 +1,155 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
4
+ sweep:
5
+ dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
9
+ sweeper:
10
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
11
+ max_batch_size: null
12
+ params: null
13
+ help:
14
+ app_name: ${hydra.job.name}
15
+ header: '${hydra.help.app_name} is powered by Hydra.
16
+
17
+ '
18
+ footer: 'Powered by Hydra (https://hydra.cc)
19
+
20
+ Use --hydra-help to view Hydra specific help
21
+
22
+ '
23
+ template: '${hydra.help.header}
24
+
25
+ == Configuration groups ==
26
+
27
+ Compose your configuration from those groups (group=option)
28
+
29
+
30
+ $APP_CONFIG_GROUPS
31
+
32
+
33
+ == Config ==
34
+
35
+ Override anything in the config (foo.bar=value)
36
+
37
+
38
+ $CONFIG
39
+
40
+
41
+ ${hydra.help.footer}
42
+
43
+ '
44
+ hydra_help:
45
+ template: 'Hydra (${hydra.runtime.version})
46
+
47
+ See https://hydra.cc for more info.
48
+
49
+
50
+ == Flags ==
51
+
52
+ $FLAGS_HELP
53
+
54
+
55
+ == Configuration groups ==
56
+
57
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
58
+ to command line)
59
+
60
+
61
+ $HYDRA_CONFIG_GROUPS
62
+
63
+
64
+ Use ''--cfg hydra'' to Show the Hydra config.
65
+
66
+ '
67
+ hydra_help: ???
68
+ hydra_logging:
69
+ version: 1
70
+ formatters:
71
+ simple:
72
+ format: '[%(asctime)s][HYDRA] %(message)s'
73
+ handlers:
74
+ console:
75
+ class: logging.StreamHandler
76
+ formatter: simple
77
+ stream: ext://sys.stdout
78
+ root:
79
+ level: INFO
80
+ handlers:
81
+ - console
82
+ loggers:
83
+ logging_example:
84
+ level: DEBUG
85
+ disable_existing_loggers: false
86
+ job_logging:
87
+ version: 1
88
+ formatters:
89
+ simple:
90
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
91
+ handlers:
92
+ console:
93
+ class: logging.StreamHandler
94
+ formatter: simple
95
+ stream: ext://sys.stdout
96
+ file:
97
+ class: logging.FileHandler
98
+ formatter: simple
99
+ filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
100
+ root:
101
+ level: INFO
102
+ handlers:
103
+ - console
104
+ - file
105
+ disable_existing_loggers: false
106
+ env: {}
107
+ mode: RUN
108
+ searchpath: []
109
+ callbacks: {}
110
+ output_subdir: .hydra
111
+ overrides:
112
+ hydra:
113
+ - hydra.mode=RUN
114
+ task:
115
+ - training.seed=45
116
+ job:
117
+ name: train
118
+ chdir: null
119
+ override_dirname: training.seed=45
120
+ id: ???
121
+ num: ???
122
+ config_name: llmbc_box-close-v2_o4-mini_qwen2-0.5b-instruct.yaml
123
+ env_set: {}
124
+ env_copy: []
125
+ config:
126
+ override_dirname:
127
+ kv_sep: '='
128
+ item_sep: ','
129
+ exclude_keys: []
130
+ runtime:
131
+ version: 1.2.0
132
+ version_base: '1.2'
133
+ cwd: /work/u1131674/LLM-BC
134
+ config_sources:
135
+ - path: hydra.conf
136
+ schema: pkg
137
+ provider: hydra
138
+ - path: /work/u1131674/LLM-BC/config/llm_backbone
139
+ schema: file
140
+ provider: main
141
+ - path: ''
142
+ schema: structured
143
+ provider: schema
144
+ output_dir: /work/u1131674/LLM-BC/data/outputs/2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2
145
+ choices:
146
+ hydra/env: default
147
+ hydra/callbacks: null
148
+ hydra/job_logging: default
149
+ hydra/hydra_logging: default
150
+ hydra/hydra_help: default
151
+ hydra/help: default
152
+ hydra/sweeper: basic
153
+ hydra/launcher: basic
154
+ hydra/output: default
155
+ verbose: false
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/.hydra/overrides.yaml ADDED
@@ -0,0 +1 @@
 
 
1
+ - training.seed=45
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0370-test_success_rate=0.920.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f6761e6ffb03b3a4d7997913b53d08bcc37c1f1921ace9d8e17a0f7ba9ea8483
3
+ size 864520
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0555-test_success_rate=0.900.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:919e8cfaa68b1f972e19838a979d5d7de66d42a47ba2f1519fe4104feffcb159
3
+ size 864520
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0670-test_success_rate=0.900.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bb83ccac6685881895955746b174c5cdf2f59f5f36674568a5093d446d4743de
3
+ size 864520
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0710-test_success_rate=0.920.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67e2be43ef474ba6f7682a3ac73e906f9cb9764c142b4f6f687365efecd2b782
3
+ size 864520
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0765-test_success_rate=0.900.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f1989af91bb2d381b974369811f6dc53dba9b8dc2612ac4ccfe93d2796c5aea
3
+ size 864520
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/checkpoints/latest.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d758edbd9d6dd3bfc6a2e6a4962d2f271b427aefa0eff74e49ce7a4cda83f27a
3
+ size 864520
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/logs.json.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:56c6bda41415c94c381efe9657b18b785e49a66d004e2a6546645808aefc09f1
3
+ size 51685613
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/train.log ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ [2026-02-27 06:56:41,595][numexpr.utils][INFO] - Note: detected 112 virtual cores but NumExpr set to maximum of 64, check "NUMEXPR_MAX_THREADS" environment variable.
2
+ [2026-02-27 06:56:41,595][numexpr.utils][INFO] - Note: NumExpr detected 112 cores but "NUMEXPR_MAX_THREADS" not set, so enforcing safe limit of 16.
3
+ [2026-02-27 06:56:41,595][numexpr.utils][INFO] - NumExpr defaulting to 16 threads.
4
+ [2026-02-27 06:56:48,325][datasets][INFO] - PyTorch version 2.2.2 available.
5
+ [2026-02-27 06:56:48,326][datasets][INFO] - TensorFlow version 2.15.1 available.
6
+ [2026-02-27 06:56:48,327][datasets][INFO] - JAX version 0.4.30 available.
7
+ [2026-02-27 06:56:57,938][absl][INFO] - MUJOCO_GL=osmesa, attempting to import specified OpenGL backend.
8
+ [2026-02-27 06:56:57,972][absl][INFO] - MuJoCo library version is: 2.3.7
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/debug-internal.log ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-02-27T06:56:58.963004774+08:00","level":"INFO","msg":"using version","core version":"0.18.6"}
2
+ {"time":"2026-02-27T06:56:58.963014266+08:00","level":"INFO","msg":"created symlink","path":"/work/u1131674/LLM-BC/data/outputs/2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/logs/debug-core.log"}
3
+ {"time":"2026-02-27T06:56:59.07867238+08:00","level":"INFO","msg":"created new stream","id":"y2drrt7x"}
4
+ {"time":"2026-02-27T06:56:59.078695479+08:00","level":"INFO","msg":"stream: started","id":"y2drrt7x"}
5
+ {"time":"2026-02-27T06:56:59.078723031+08:00","level":"INFO","msg":"sender: started","stream_id":"y2drrt7x"}
6
+ {"time":"2026-02-27T06:56:59.078712125+08:00","level":"INFO","msg":"writer: Do: started","stream_id":{"value":"y2drrt7x"}}
7
+ {"time":"2026-02-27T06:56:59.078723394+08:00","level":"INFO","msg":"handler: started","stream_id":{"value":"y2drrt7x"}}
8
+ {"time":"2026-02-27T06:56:59.81316194+08:00","level":"INFO","msg":"Starting system monitor"}
9
+ {"time":"2026-02-28T01:59:13.283337585+08:00","level":"INFO","msg":"stream: closing","id":"y2drrt7x"}
10
+ {"time":"2026-02-28T01:59:13.283363707+08:00","level":"INFO","msg":"Stopping system monitor"}
11
+ {"time":"2026-02-28T01:59:13.283834312+08:00","level":"INFO","msg":"Stopped system monitor"}
12
+ {"time":"2026-02-28T01:59:18.731651769+08:00","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
13
+ {"time":"2026-02-28T01:59:19.338161326+08:00","level":"INFO","msg":"handler: closed","stream_id":{"value":"y2drrt7x"}}
14
+ {"time":"2026-02-28T01:59:19.3382042+08:00","level":"INFO","msg":"writer: Close: closed","stream_id":{"value":"y2drrt7x"}}
15
+ {"time":"2026-02-28T01:59:19.33824045+08:00","level":"INFO","msg":"sender: closed","stream_id":"y2drrt7x"}
16
+ {"time":"2026-02-28T01:59:19.338261081+08:00","level":"INFO","msg":"stream: closed","id":"y2drrt7x"}
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/debug.log ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-02-27 06:56:58,958 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Current SDK version is 0.18.6
2
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Configure stats pid to 1285863
3
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Loading settings from /home/u1131674/.config/wandb/settings
4
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Loading settings from /work/u1131674/LLM-BC/wandb/settings
5
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Loading settings from environment variables: {}
6
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Applying setup settings: {'mode': 'online', '_disable_service': None}
7
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Inferring run settings from compute environment: {'program_relpath': 'train.py', 'program_abspath': '/work/u1131674/LLM-BC/train.py', 'program': '/work/u1131674/LLM-BC/./train.py'}
8
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Applying login settings: {}
9
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_init.py:_log_setup():533] Logging user logs to /work/u1131674/LLM-BC/data/outputs/2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/logs/debug.log
10
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_init.py:_log_setup():534] Logging internal logs to /work/u1131674/LLM-BC/data/outputs/2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/logs/debug-internal.log
11
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_init.py:init():619] calling init triggers
12
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_init.py:init():626] wandb.init called with sweep_config: {}
13
+ config: {'name': 'train_llmbc_lowdim', '_target_': 'llmbc.workspace.train_llmbc_lowdim_workspace.TrainLLMBCLowdimWorkspace', 'obs_dim': 9, 'action_dim': 4, 'task_name': 'box-close-v2', 'exp_name': 'default', 'model_name': 'Qwen/Qwen2-0.5B-Instruct', 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1, 'n_latency_steps': 0, 'past_action_visible': False, 'llm_orig_expert_feedback': True, 'llm_do_sample': False, 'policy': {'_target_': 'llmbc.policy.llmbc_lowdim_policy.LLMBCLowdimPolicy', 'model': {'_target_': 'llmbc.model.policy.policy_mlp.PolicyMLP', 'input_size': 9, 'hidden_size': [256, 256], 'output_size': 4, 'activation': 'relu', 'n_obs_steps': 1, 'n_action_steps': 1}, 'obs_dim': 9, 'action_dim': 4, 'llm_discriminator': {'_target_': 'llmbc.discriminator.llm_ce_discriminator.LLMCEDiscriminator', 'task_id': 'box-close-v2', 'llm_translator': {'_target_': 'llmbc.translator.llm_translator.LLMTranslator', 'cfg': {'name': 'Qwen/Qwen2-0.5B-Instruct', 'model_name': 'Qwen2-0.5B-Instruct', 'config_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2Config', 'causal_lm_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2ForCausalLM', 'use_quantization': False, 'use_joint_mlp_projector': True, 'llm_mode': 'ete-finetuned', 'finetune_mode': 'orig', 'checkpoint': 'data/outputs/2026.02.20/17.05.08_train_llm_lowdim_box-close-v2/Qwen/Qwen2-0.5B-Instruct-finetuned-box-close-v2/checkpoint-17670', 'max_length': 100, 'lora_config': {'r': 32, 'lora_alpha': 64, 'lora_dropout': 0.05, 'bias': 'none', 'task_type': 'CAUSAL_LM'}, 'prompter': {'_target_': 'llmbc.translator.prompter.qwen2_prompter.Qwen2Prompter', 'use_joint_mlp_projector': True}, 'hydra': {'job': {'override_dirname': 'Qwen/Qwen2-0.5B-Instruct'}, 'run': {'dir': 'data/outputs/2026.02.27/06.56.39_Qwen/Qwen2-0.5B-Instruct'}}}, 'obs_dim': 9, 'action_dim': 4, 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1}}, 'loss_bc_weight': 1.0, 'loss_llm_weight': 0.001, 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1, 'normalize_llm_loss': True}, 'dataloader': {'batch_size': 16, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'val_dataloader': {'batch_size': 16, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'optimizer': {'_target_': 'torch.optim.AdamW', 'lr': 0.01, 'betas': [0.95, 0.999], 'eps': 1e-08, 'weight_decay': 1e-06}, 'training': {'device': 'cuda:0', 'seed': 45, 'debug': False, 'resume': False, 'lr_scheduler': 'cosine', 'lr_warmup_steps': 10, 'num_epochs': 1001, 'gradient_accumulate_every': 8, 'grad_norm_clip': 0.5, 'rollout_every': 5, 'checkpoint_every': 5, 'val_every': 1, 'sample_every': 5, 'sample_max_batch': 128, 'max_train_steps': None, 'max_val_steps': None, 'tqdm_interval_sec': 1.0}, 'logging': {'project': 'box-close-v2-training-llm-backbone', 'resume': True, 'mode': 'online', 'name': '2026.02.27-06.56.39_train_llmbc_lowdim_box-close-v2', 'tags': ['train_llmbc_lowdim', 'box-close-v2', 'default'], 'id': None, 'group': None}, 'checkpoint': {'topk': {'monitor_key': 'test_success_rate', 'mode': 'max', 'k': 5, 'format_str': 'epoch={epoch:04d}-test_success_rate={test_success_rate:.3f}.ckpt'}, 'save_last_ckpt': True, 'save_last_snapshot': False}, 'multi_run': {'run_dir': 'data/outputs/2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2', 'wandb_name_base': '2026.02.27-06.56.39_train_llmbc_lowdim_box-close-v2'}, 'task': {'name': 'box-close-v2', 'obs_dim': 9, 'action_dim': 4, 'env_runner': {'_target_': 'llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner', 'env_name': 'llf-metaworld-box-close-v2', 'n_train': 10, 'n_test': 50, 'n_envs': 10, 'max_steps': 30, 'n_obs_steps': 1, 'n_action_steps': 1, 'instruction_type': 'b', 'feedback_type': ['hp', 'hn', 'fp'], 'visual': False, 'discount': 0.9}, 'dataset': {'_target_': 'llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset', 'data_path': 'datasets/box-close-v2-o4-mini.pt', 'data_path2': 'datasets/box-close-v2-o4-mini.pt', 'horizon': 1, 'pad_before': 0, 'pad_after': 0, 'obs_eef_target': True, 'use_manual_normalizer': False, 'val_ratio': 0.1, 'dummy_normalizer': True}, 'instructor': {'_target_': 'llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor'}}, 'llm': {'name': 'Qwen/Qwen2-0.5B-Instruct', 'model_name': 'Qwen2-0.5B-Instruct', 'config_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2Config', 'causal_lm_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2ForCausalLM', 'use_quantization': False, 'use_joint_mlp_projector': True, 'llm_mode': 'ete-finetuned', 'finetune_mode': 'orig', 'checkpoint': 'data/outputs/2026.02.20/17.05.08_train_llm_lowdim_box-close-v2/Qwen/Qwen2-0.5B-Instruct-finetuned-box-close-v2/checkpoint-17670', 'max_length': 100, 'lora_config': {'r': 32, 'lora_alpha': 64, 'lora_dropout': 0.05, 'bias': 'none', 'task_type': 'CAUSAL_LM'}, 'prompter': {'_target_': 'llmbc.translator.prompter.qwen2_prompter.Qwen2Prompter', 'use_joint_mlp_projector': True}, 'hydra': {'job': {'override_dirname': 'Qwen/Qwen2-0.5B-Instruct'}, 'run': {'dir': 'data/outputs/2026.02.27/06.56.39_Qwen/Qwen2-0.5B-Instruct'}}}}
14
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_init.py:init():669] starting backend
15
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_init.py:init():673] sending inform_init request
16
+ 2026-02-27 06:56:58,960 INFO MainThread:1285863 [backend.py:_multiprocessing_setup():104] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
17
+ 2026-02-27 06:56:58,961 INFO MainThread:1285863 [wandb_init.py:init():686] backend started and connected
18
+ 2026-02-27 06:56:58,969 INFO MainThread:1285863 [wandb_init.py:init():781] updated telemetry
19
+ 2026-02-27 06:56:59,021 INFO MainThread:1285863 [wandb_init.py:init():814] communicating run to backend with 90.0 second timeout
20
+ 2026-02-27 06:56:59,808 INFO MainThread:1285863 [wandb_init.py:init():867] starting run threads in backend
21
+ 2026-02-27 06:57:00,340 INFO MainThread:1285863 [wandb_run.py:_console_start():2451] atexit reg
22
+ 2026-02-27 06:57:00,340 INFO MainThread:1285863 [wandb_run.py:_redirect():2299] redirect: wrap_raw
23
+ 2026-02-27 06:57:00,340 INFO MainThread:1285863 [wandb_run.py:_redirect():2364] Wrapping output streams.
24
+ 2026-02-27 06:57:00,340 INFO MainThread:1285863 [wandb_run.py:_redirect():2389] Redirects installed.
25
+ 2026-02-27 06:57:00,343 INFO MainThread:1285863 [wandb_init.py:init():911] run started, returning control to user process
26
+ 2026-02-27 06:57:00,343 INFO MainThread:1285863 [wandb_run.py:_config_callback():1389] config_cb None None {'output_dir': '/work/u1131674/LLM-BC/data/outputs/2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2'}
27
+ 2026-02-28 01:59:13,283 WARNING MsgRouterThr:1285863 [router.py:message_loop():75] message_loop has been closed
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/files/config.yaml ADDED
@@ -0,0 +1,271 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _target_:
2
+ value: llmbc.workspace.train_llmbc_lowdim_workspace.TrainLLMBCLowdimWorkspace
3
+ _wandb:
4
+ value:
5
+ cli_version: 0.18.6
6
+ m: []
7
+ python_version: 3.9.25
8
+ t:
9
+ "1":
10
+ - 1
11
+ - 2
12
+ - 3
13
+ - 5
14
+ - 11
15
+ - 12
16
+ - 41
17
+ - 49
18
+ - 50
19
+ - 51
20
+ - 53
21
+ - 55
22
+ - 71
23
+ - 83
24
+ - 95
25
+ - 98
26
+ - 100
27
+ - 105
28
+ "2":
29
+ - 1
30
+ - 2
31
+ - 3
32
+ - 5
33
+ - 11
34
+ - 12
35
+ - 41
36
+ - 49
37
+ - 50
38
+ - 51
39
+ - 53
40
+ - 55
41
+ - 71
42
+ - 83
43
+ - 95
44
+ - 98
45
+ - 100
46
+ - 105
47
+ "3":
48
+ - 13
49
+ - 15
50
+ - 16
51
+ - 23
52
+ - 55
53
+ - 61
54
+ "4": 3.9.25
55
+ "5": 0.18.6
56
+ "6": 4.47.1
57
+ "8":
58
+ - 5
59
+ "12": 0.18.6
60
+ "13": linux-x86_64
61
+ action_dim:
62
+ value: 4
63
+ checkpoint:
64
+ value:
65
+ save_last_ckpt: true
66
+ save_last_snapshot: false
67
+ topk:
68
+ format_str: epoch={epoch:04d}-test_success_rate={test_success_rate:.3f}.ckpt
69
+ k: 5
70
+ mode: max
71
+ monitor_key: test_success_rate
72
+ dataloader:
73
+ value:
74
+ batch_size: 16
75
+ num_workers: 0
76
+ persistent_workers: false
77
+ pin_memory: false
78
+ shuffle: true
79
+ exp_name:
80
+ value: default
81
+ horizon:
82
+ value: 1
83
+ llm:
84
+ value:
85
+ causal_lm_target: llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2ForCausalLM
86
+ checkpoint: data/outputs/2026.02.20/17.05.08_train_llm_lowdim_box-close-v2/Qwen/Qwen2-0.5B-Instruct-finetuned-box-close-v2/checkpoint-17670
87
+ config_target: llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2Config
88
+ finetune_mode: orig
89
+ hydra:
90
+ job:
91
+ override_dirname: Qwen/Qwen2-0.5B-Instruct
92
+ run:
93
+ dir: data/outputs/2026.02.27/06.56.39_Qwen/Qwen2-0.5B-Instruct
94
+ llm_mode: ete-finetuned
95
+ lora_config:
96
+ bias: none
97
+ lora_alpha: 64
98
+ lora_dropout: 0.05
99
+ r: 32
100
+ task_type: CAUSAL_LM
101
+ max_length: 100
102
+ model_name: Qwen2-0.5B-Instruct
103
+ name: Qwen/Qwen2-0.5B-Instruct
104
+ prompter:
105
+ _target_: llmbc.translator.prompter.qwen2_prompter.Qwen2Prompter
106
+ use_joint_mlp_projector: true
107
+ use_joint_mlp_projector: true
108
+ use_quantization: false
109
+ llm_do_sample:
110
+ value: false
111
+ llm_orig_expert_feedback:
112
+ value: true
113
+ logging:
114
+ value:
115
+ group: null
116
+ id: null
117
+ mode: online
118
+ name: 2026.02.27-06.56.39_train_llmbc_lowdim_box-close-v2
119
+ project: box-close-v2-training-llm-backbone
120
+ resume: true
121
+ tags:
122
+ - train_llmbc_lowdim
123
+ - box-close-v2
124
+ - default
125
+ model_name:
126
+ value: Qwen/Qwen2-0.5B-Instruct
127
+ multi_run:
128
+ value:
129
+ run_dir: data/outputs/2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2
130
+ wandb_name_base: 2026.02.27-06.56.39_train_llmbc_lowdim_box-close-v2
131
+ n_action_steps:
132
+ value: 1
133
+ n_latency_steps:
134
+ value: 0
135
+ n_obs_steps:
136
+ value: 1
137
+ name:
138
+ value: train_llmbc_lowdim
139
+ obs_dim:
140
+ value: 9
141
+ optimizer:
142
+ value:
143
+ _target_: torch.optim.AdamW
144
+ betas:
145
+ - 0.95
146
+ - 0.999
147
+ eps: 1e-08
148
+ lr: 0.01
149
+ weight_decay: 1e-06
150
+ output_dir:
151
+ value: /work/u1131674/LLM-BC/data/outputs/2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2
152
+ past_action_visible:
153
+ value: false
154
+ policy:
155
+ value:
156
+ _target_: llmbc.policy.llmbc_lowdim_policy.LLMBCLowdimPolicy
157
+ action_dim: 4
158
+ horizon: 1
159
+ llm_discriminator:
160
+ _target_: llmbc.discriminator.llm_ce_discriminator.LLMCEDiscriminator
161
+ llm_translator:
162
+ _target_: llmbc.translator.llm_translator.LLMTranslator
163
+ action_dim: 4
164
+ cfg:
165
+ causal_lm_target: llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2ForCausalLM
166
+ checkpoint: data/outputs/2026.02.20/17.05.08_train_llm_lowdim_box-close-v2/Qwen/Qwen2-0.5B-Instruct-finetuned-box-close-v2/checkpoint-17670
167
+ config_target: llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2Config
168
+ finetune_mode: orig
169
+ hydra:
170
+ job:
171
+ override_dirname: Qwen/Qwen2-0.5B-Instruct
172
+ run:
173
+ dir: data/outputs/2026.02.27/06.56.39_Qwen/Qwen2-0.5B-Instruct
174
+ llm_mode: ete-finetuned
175
+ lora_config:
176
+ bias: none
177
+ lora_alpha: 64
178
+ lora_dropout: 0.05
179
+ r: 32
180
+ task_type: CAUSAL_LM
181
+ max_length: 100
182
+ model_name: Qwen2-0.5B-Instruct
183
+ name: Qwen/Qwen2-0.5B-Instruct
184
+ prompter:
185
+ _target_: llmbc.translator.prompter.qwen2_prompter.Qwen2Prompter
186
+ use_joint_mlp_projector: true
187
+ use_joint_mlp_projector: true
188
+ use_quantization: false
189
+ horizon: 1
190
+ n_action_steps: 1
191
+ n_obs_steps: 1
192
+ obs_dim: 9
193
+ task_id: box-close-v2
194
+ loss_bc_weight: 1
195
+ loss_llm_weight: 0.001
196
+ model:
197
+ _target_: llmbc.model.policy.policy_mlp.PolicyMLP
198
+ activation: relu
199
+ hidden_size:
200
+ - 256
201
+ - 256
202
+ input_size: 9
203
+ n_action_steps: 1
204
+ n_obs_steps: 1
205
+ output_size: 4
206
+ n_action_steps: 1
207
+ n_obs_steps: 1
208
+ normalize_llm_loss: true
209
+ obs_dim: 9
210
+ task:
211
+ value:
212
+ action_dim: 4
213
+ dataset:
214
+ _target_: llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset
215
+ data_path: datasets/box-close-v2-o4-mini.pt
216
+ data_path2: datasets/box-close-v2-o4-mini.pt
217
+ dummy_normalizer: true
218
+ horizon: 1
219
+ obs_eef_target: true
220
+ pad_after: 0
221
+ pad_before: 0
222
+ use_manual_normalizer: false
223
+ val_ratio: 0.1
224
+ env_runner:
225
+ _target_: llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner
226
+ discount: 0.9
227
+ env_name: llf-metaworld-box-close-v2
228
+ feedback_type:
229
+ - hp
230
+ - hn
231
+ - fp
232
+ instruction_type: b
233
+ max_steps: 30
234
+ n_action_steps: 1
235
+ n_envs: 10
236
+ n_obs_steps: 1
237
+ n_test: 50
238
+ n_train: 10
239
+ visual: false
240
+ instructor:
241
+ _target_: llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor
242
+ name: box-close-v2
243
+ obs_dim: 9
244
+ task_name:
245
+ value: box-close-v2
246
+ training:
247
+ value:
248
+ checkpoint_every: 5
249
+ debug: false
250
+ device: cuda:0
251
+ grad_norm_clip: 0.5
252
+ gradient_accumulate_every: 8
253
+ lr_scheduler: cosine
254
+ lr_warmup_steps: 10
255
+ max_train_steps: null
256
+ max_val_steps: null
257
+ num_epochs: 1001
258
+ resume: false
259
+ rollout_every: 5
260
+ sample_every: 5
261
+ sample_max_batch: 128
262
+ seed: 45
263
+ tqdm_interval_sec: 1
264
+ val_every: 1
265
+ val_dataloader:
266
+ value:
267
+ batch_size: 16
268
+ num_workers: 0
269
+ persistent_workers: false
270
+ pin_memory: false
271
+ shuffle: true
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/files/output.log ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ Eval MetaworldLowdimRunner 1/6: 0%| | 0/30 [00:00<?, ?it/s]/work/u1131674/LLM-BC/llmbc/common/llfbench_util.py:39: UserWarning: Creating a tensor from a list of numpy.ndarrays is extremely slow. Please consider converting the list to a single numpy.ndarray with numpy.array() before converting to a tensor. (Triggered internally at ../torch/csrc/utils/tensor_new.cpp:275.)
2
+ obs = torch.tensor(obs, dtype=torch.float32).unsqueeze(dim=0).to(device)
3
+
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/files/wandb-metadata.json ADDED
@@ -0,0 +1,107 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-4.18.0-513.24.1.el8_9.x86_64-x86_64-with-glibc2.28",
3
+ "python": "3.9.25",
4
+ "startedAt": "2026-02-26T22:56:58.961212Z",
5
+ "args": [
6
+ "--config-path",
7
+ "config/llm_backbone",
8
+ "--config-name",
9
+ "llmbc_box-close-v2_o4-mini_qwen2-0.5b-instruct.yaml",
10
+ "training.seed=45"
11
+ ],
12
+ "program": "/work/u1131674/LLM-BC/./train.py",
13
+ "codePath": "train.py",
14
+ "git": {
15
+ "remote": "https://github.com/CHYang25/LLM-BC.git",
16
+ "commit": "155285319c80dc3cf3f63496e1795b29a54a9377"
17
+ },
18
+ "email": "chris920325@gmail.com",
19
+ "root": "/work/u1131674/LLM-BC/data/outputs/2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2",
20
+ "host": "hgpn05",
21
+ "username": "u1131674",
22
+ "executable": "/home/u1131674/.conda/envs/llm-bc/bin/python3",
23
+ "codePathLocal": "train.py",
24
+ "cpu_count": 112,
25
+ "cpu_count_logical": 112,
26
+ "gpu": "NVIDIA H100 80GB HBM3",
27
+ "gpu_count": 1,
28
+ "disk": {
29
+ "/": {
30
+ "total": "1918024196096",
31
+ "used": "287118954496"
32
+ }
33
+ },
34
+ "memory": {
35
+ "total": "2163617767424"
36
+ },
37
+ "cpu": {
38
+ "count": 112,
39
+ "countLogical": 112
40
+ },
41
+ "gpu_nvidia": [
42
+ {
43
+ "name": "NVIDIA H100 80GB HBM3",
44
+ "memoryTotal": "85520809984",
45
+ "cudaCores": 16896,
46
+ "architecture": "Hopper"
47
+ }
48
+ ],
49
+ "slurm": {
50
+ "cluster_name": "hpc",
51
+ "conf": "/etc/slurm/slurm.conf",
52
+ "cpu_bind": "quiet,mask_cpu:0x0000F000001C0100000000000000",
53
+ "cpu_bind_list": "0x0000F000001C0100000000000000",
54
+ "cpu_bind_type": "mask_cpu:",
55
+ "cpu_bind_verbose": "quiet",
56
+ "cpus_on_node": "8",
57
+ "cpus_per_task": "8",
58
+ "distribution": "cyclic,pack",
59
+ "gpus_on_node": "1",
60
+ "gpus_per_node": "1",
61
+ "gtids": "0",
62
+ "job_account": "mst114558",
63
+ "job_cpus_per_node": "8",
64
+ "job_end_time": "1772319396",
65
+ "job_gid": "106773",
66
+ "job_group": "MST114558",
67
+ "job_id": "124194",
68
+ "job_name": "python3",
69
+ "job_nodelist": "hgpn05",
70
+ "job_num_nodes": "1",
71
+ "job_partition": "normal",
72
+ "job_qos": "normal",
73
+ "job_start_time": "1772146596",
74
+ "job_uid": "41408",
75
+ "job_user": "u1131674",
76
+ "jobid": "124194",
77
+ "launch_node_ipaddr": "172.21.101.1",
78
+ "localid": "0",
79
+ "mem_per_node": "204800",
80
+ "nnodes": "1",
81
+ "nodeid": "0",
82
+ "nodelist": "hgpn05",
83
+ "nprocs": "1",
84
+ "ntasks": "1",
85
+ "prio_process": "0",
86
+ "procid": "0",
87
+ "srun_comm_host": "172.21.101.1",
88
+ "srun_comm_port": "41745",
89
+ "step_gpus": "4",
90
+ "step_id": "0",
91
+ "step_launcher_port": "41745",
92
+ "step_nodelist": "hgpn05",
93
+ "step_num_nodes": "1",
94
+ "step_num_tasks": "1",
95
+ "step_tasks_per_node": "1",
96
+ "stepid": "0",
97
+ "submit_dir": "/work/u1131674/LLM-BC",
98
+ "submit_host": "cbi-lgn01",
99
+ "task_pid": "1285863",
100
+ "tasks_per_node": "1",
101
+ "topology_addr": "ibsw1.hgpn05",
102
+ "topology_addr_pattern": "switch.node",
103
+ "tres_per_task": "cpu:8",
104
+ "umask": "0022"
105
+ },
106
+ "cudaVersion": "12.4"
107
+ }
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/files/wandb-summary.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"train/success_rate":0.9,"train_loss_bc":4.764880213770084e-05,"_wandb":{"runtime":68534},"test/success_rate":0.76,"global_step":237236,"grad_norm":0.01001900527626276,"test/mean_score":6.7717438534985055,"_timestamp":1.7722151509243326e+09,"lr":2.8077998259767867e-11,"train_loss_llm":0.043227363377809525,"train_action_mse_error":4.231345155858435e-05,"_runtime":68534.322141401,"test/cumulative_reward":13.066209712619024,"val_loss":0.0010201800614595413,"train/mean_score":7.989104326341963,"train_loss":0.00014833795020679371,"epoch":1000,"_step":237236,"train/cumulative_reward":15.759684587524779}
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/logs/debug-core.log ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-02-27T06:56:58.324896123+08:00","level":"INFO","msg":"started logging, with flags","port-filename":"/tmp/tmpbo9895mj/port-1285863.txt","pid":1285863,"debug":false,"disable-analytics":false}
2
+ {"time":"2026-02-27T06:56:58.324917661+08:00","level":"INFO","msg":"FeatureState","shutdownOnParentExitEnabled":false}
3
+ {"time":"2026-02-27T06:56:58.32525177+08:00","level":"INFO","msg":"Will exit if parent process dies.","ppid":1285863}
4
+ {"time":"2026-02-27T06:56:58.325246832+08:00","level":"INFO","msg":"server is running","addr":{"IP":"127.0.0.1","Port":41827,"Zone":""}}
5
+ {"time":"2026-02-27T06:56:58.517882072+08:00","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"127.0.0.1:35620"}
6
+ {"time":"2026-02-27T06:56:58.961740182+08:00","level":"INFO","msg":"handleInformInit: received","streamId":"y2drrt7x","id":"127.0.0.1:35620"}
7
+ {"time":"2026-02-27T06:56:59.078699634+08:00","level":"INFO","msg":"handleInformInit: stream started","streamId":"y2drrt7x","id":"127.0.0.1:35620"}
8
+ {"time":"2026-02-28T01:59:13.283276571+08:00","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"127.0.0.1:35620"}
9
+ {"time":"2026-02-28T01:59:13.283355844+08:00","level":"INFO","msg":"server is shutting down"}
10
+ {"time":"2026-02-28T01:59:13.283345125+08:00","level":"INFO","msg":"connection: Close: initiating connection closure","id":"127.0.0.1:35620"}
11
+ {"time":"2026-02-28T01:59:13.283456187+08:00","level":"INFO","msg":"connection: Close: connection successfully closed","id":"127.0.0.1:35620"}
12
+ {"time":"2026-02-28T01:59:19.338511341+08:00","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"127.0.0.1:35620"}
13
+ {"time":"2026-02-28T01:59:19.338529887+08:00","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"127.0.0.1:35620"}
14
+ {"time":"2026-02-28T01:59:19.338540836+08:00","level":"INFO","msg":"server is closed"}
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/logs/debug-internal.log ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-02-27T06:56:58.963004774+08:00","level":"INFO","msg":"using version","core version":"0.18.6"}
2
+ {"time":"2026-02-27T06:56:58.963014266+08:00","level":"INFO","msg":"created symlink","path":"/work/u1131674/LLM-BC/data/outputs/2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/logs/debug-core.log"}
3
+ {"time":"2026-02-27T06:56:59.07867238+08:00","level":"INFO","msg":"created new stream","id":"y2drrt7x"}
4
+ {"time":"2026-02-27T06:56:59.078695479+08:00","level":"INFO","msg":"stream: started","id":"y2drrt7x"}
5
+ {"time":"2026-02-27T06:56:59.078723031+08:00","level":"INFO","msg":"sender: started","stream_id":"y2drrt7x"}
6
+ {"time":"2026-02-27T06:56:59.078712125+08:00","level":"INFO","msg":"writer: Do: started","stream_id":{"value":"y2drrt7x"}}
7
+ {"time":"2026-02-27T06:56:59.078723394+08:00","level":"INFO","msg":"handler: started","stream_id":{"value":"y2drrt7x"}}
8
+ {"time":"2026-02-27T06:56:59.81316194+08:00","level":"INFO","msg":"Starting system monitor"}
9
+ {"time":"2026-02-28T01:59:13.283337585+08:00","level":"INFO","msg":"stream: closing","id":"y2drrt7x"}
10
+ {"time":"2026-02-28T01:59:13.283363707+08:00","level":"INFO","msg":"Stopping system monitor"}
11
+ {"time":"2026-02-28T01:59:13.283834312+08:00","level":"INFO","msg":"Stopped system monitor"}
12
+ {"time":"2026-02-28T01:59:18.731651769+08:00","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
13
+ {"time":"2026-02-28T01:59:19.338161326+08:00","level":"INFO","msg":"handler: closed","stream_id":{"value":"y2drrt7x"}}
14
+ {"time":"2026-02-28T01:59:19.3382042+08:00","level":"INFO","msg":"writer: Close: closed","stream_id":{"value":"y2drrt7x"}}
15
+ {"time":"2026-02-28T01:59:19.33824045+08:00","level":"INFO","msg":"sender: closed","stream_id":"y2drrt7x"}
16
+ {"time":"2026-02-28T01:59:19.338261081+08:00","level":"INFO","msg":"stream: closed","id":"y2drrt7x"}
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/logs/debug.log ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-02-27 06:56:58,958 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Current SDK version is 0.18.6
2
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Configure stats pid to 1285863
3
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Loading settings from /home/u1131674/.config/wandb/settings
4
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Loading settings from /work/u1131674/LLM-BC/wandb/settings
5
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Loading settings from environment variables: {}
6
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Applying setup settings: {'mode': 'online', '_disable_service': None}
7
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Inferring run settings from compute environment: {'program_relpath': 'train.py', 'program_abspath': '/work/u1131674/LLM-BC/train.py', 'program': '/work/u1131674/LLM-BC/./train.py'}
8
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_setup.py:_flush():79] Applying login settings: {}
9
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_init.py:_log_setup():533] Logging user logs to /work/u1131674/LLM-BC/data/outputs/2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/logs/debug.log
10
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_init.py:_log_setup():534] Logging internal logs to /work/u1131674/LLM-BC/data/outputs/2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/logs/debug-internal.log
11
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_init.py:init():619] calling init triggers
12
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_init.py:init():626] wandb.init called with sweep_config: {}
13
+ config: {'name': 'train_llmbc_lowdim', '_target_': 'llmbc.workspace.train_llmbc_lowdim_workspace.TrainLLMBCLowdimWorkspace', 'obs_dim': 9, 'action_dim': 4, 'task_name': 'box-close-v2', 'exp_name': 'default', 'model_name': 'Qwen/Qwen2-0.5B-Instruct', 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1, 'n_latency_steps': 0, 'past_action_visible': False, 'llm_orig_expert_feedback': True, 'llm_do_sample': False, 'policy': {'_target_': 'llmbc.policy.llmbc_lowdim_policy.LLMBCLowdimPolicy', 'model': {'_target_': 'llmbc.model.policy.policy_mlp.PolicyMLP', 'input_size': 9, 'hidden_size': [256, 256], 'output_size': 4, 'activation': 'relu', 'n_obs_steps': 1, 'n_action_steps': 1}, 'obs_dim': 9, 'action_dim': 4, 'llm_discriminator': {'_target_': 'llmbc.discriminator.llm_ce_discriminator.LLMCEDiscriminator', 'task_id': 'box-close-v2', 'llm_translator': {'_target_': 'llmbc.translator.llm_translator.LLMTranslator', 'cfg': {'name': 'Qwen/Qwen2-0.5B-Instruct', 'model_name': 'Qwen2-0.5B-Instruct', 'config_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2Config', 'causal_lm_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2ForCausalLM', 'use_quantization': False, 'use_joint_mlp_projector': True, 'llm_mode': 'ete-finetuned', 'finetune_mode': 'orig', 'checkpoint': 'data/outputs/2026.02.20/17.05.08_train_llm_lowdim_box-close-v2/Qwen/Qwen2-0.5B-Instruct-finetuned-box-close-v2/checkpoint-17670', 'max_length': 100, 'lora_config': {'r': 32, 'lora_alpha': 64, 'lora_dropout': 0.05, 'bias': 'none', 'task_type': 'CAUSAL_LM'}, 'prompter': {'_target_': 'llmbc.translator.prompter.qwen2_prompter.Qwen2Prompter', 'use_joint_mlp_projector': True}, 'hydra': {'job': {'override_dirname': 'Qwen/Qwen2-0.5B-Instruct'}, 'run': {'dir': 'data/outputs/2026.02.27/06.56.39_Qwen/Qwen2-0.5B-Instruct'}}}, 'obs_dim': 9, 'action_dim': 4, 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1}}, 'loss_bc_weight': 1.0, 'loss_llm_weight': 0.001, 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1, 'normalize_llm_loss': True}, 'dataloader': {'batch_size': 16, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'val_dataloader': {'batch_size': 16, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'optimizer': {'_target_': 'torch.optim.AdamW', 'lr': 0.01, 'betas': [0.95, 0.999], 'eps': 1e-08, 'weight_decay': 1e-06}, 'training': {'device': 'cuda:0', 'seed': 45, 'debug': False, 'resume': False, 'lr_scheduler': 'cosine', 'lr_warmup_steps': 10, 'num_epochs': 1001, 'gradient_accumulate_every': 8, 'grad_norm_clip': 0.5, 'rollout_every': 5, 'checkpoint_every': 5, 'val_every': 1, 'sample_every': 5, 'sample_max_batch': 128, 'max_train_steps': None, 'max_val_steps': None, 'tqdm_interval_sec': 1.0}, 'logging': {'project': 'box-close-v2-training-llm-backbone', 'resume': True, 'mode': 'online', 'name': '2026.02.27-06.56.39_train_llmbc_lowdim_box-close-v2', 'tags': ['train_llmbc_lowdim', 'box-close-v2', 'default'], 'id': None, 'group': None}, 'checkpoint': {'topk': {'monitor_key': 'test_success_rate', 'mode': 'max', 'k': 5, 'format_str': 'epoch={epoch:04d}-test_success_rate={test_success_rate:.3f}.ckpt'}, 'save_last_ckpt': True, 'save_last_snapshot': False}, 'multi_run': {'run_dir': 'data/outputs/2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2', 'wandb_name_base': '2026.02.27-06.56.39_train_llmbc_lowdim_box-close-v2'}, 'task': {'name': 'box-close-v2', 'obs_dim': 9, 'action_dim': 4, 'env_runner': {'_target_': 'llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner', 'env_name': 'llf-metaworld-box-close-v2', 'n_train': 10, 'n_test': 50, 'n_envs': 10, 'max_steps': 30, 'n_obs_steps': 1, 'n_action_steps': 1, 'instruction_type': 'b', 'feedback_type': ['hp', 'hn', 'fp'], 'visual': False, 'discount': 0.9}, 'dataset': {'_target_': 'llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset', 'data_path': 'datasets/box-close-v2-o4-mini.pt', 'data_path2': 'datasets/box-close-v2-o4-mini.pt', 'horizon': 1, 'pad_before': 0, 'pad_after': 0, 'obs_eef_target': True, 'use_manual_normalizer': False, 'val_ratio': 0.1, 'dummy_normalizer': True}, 'instructor': {'_target_': 'llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor'}}, 'llm': {'name': 'Qwen/Qwen2-0.5B-Instruct', 'model_name': 'Qwen2-0.5B-Instruct', 'config_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2Config', 'causal_lm_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2ForCausalLM', 'use_quantization': False, 'use_joint_mlp_projector': True, 'llm_mode': 'ete-finetuned', 'finetune_mode': 'orig', 'checkpoint': 'data/outputs/2026.02.20/17.05.08_train_llm_lowdim_box-close-v2/Qwen/Qwen2-0.5B-Instruct-finetuned-box-close-v2/checkpoint-17670', 'max_length': 100, 'lora_config': {'r': 32, 'lora_alpha': 64, 'lora_dropout': 0.05, 'bias': 'none', 'task_type': 'CAUSAL_LM'}, 'prompter': {'_target_': 'llmbc.translator.prompter.qwen2_prompter.Qwen2Prompter', 'use_joint_mlp_projector': True}, 'hydra': {'job': {'override_dirname': 'Qwen/Qwen2-0.5B-Instruct'}, 'run': {'dir': 'data/outputs/2026.02.27/06.56.39_Qwen/Qwen2-0.5B-Instruct'}}}}
14
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_init.py:init():669] starting backend
15
+ 2026-02-27 06:56:58,959 INFO MainThread:1285863 [wandb_init.py:init():673] sending inform_init request
16
+ 2026-02-27 06:56:58,960 INFO MainThread:1285863 [backend.py:_multiprocessing_setup():104] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
17
+ 2026-02-27 06:56:58,961 INFO MainThread:1285863 [wandb_init.py:init():686] backend started and connected
18
+ 2026-02-27 06:56:58,969 INFO MainThread:1285863 [wandb_init.py:init():781] updated telemetry
19
+ 2026-02-27 06:56:59,021 INFO MainThread:1285863 [wandb_init.py:init():814] communicating run to backend with 90.0 second timeout
20
+ 2026-02-27 06:56:59,808 INFO MainThread:1285863 [wandb_init.py:init():867] starting run threads in backend
21
+ 2026-02-27 06:57:00,340 INFO MainThread:1285863 [wandb_run.py:_console_start():2451] atexit reg
22
+ 2026-02-27 06:57:00,340 INFO MainThread:1285863 [wandb_run.py:_redirect():2299] redirect: wrap_raw
23
+ 2026-02-27 06:57:00,340 INFO MainThread:1285863 [wandb_run.py:_redirect():2364] Wrapping output streams.
24
+ 2026-02-27 06:57:00,340 INFO MainThread:1285863 [wandb_run.py:_redirect():2389] Redirects installed.
25
+ 2026-02-27 06:57:00,343 INFO MainThread:1285863 [wandb_init.py:init():911] run started, returning control to user process
26
+ 2026-02-27 06:57:00,343 INFO MainThread:1285863 [wandb_run.py:_config_callback():1389] config_cb None None {'output_dir': '/work/u1131674/LLM-BC/data/outputs/2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2'}
27
+ 2026-02-28 01:59:13,283 WARNING MsgRouterThr:1285863 [router.py:message_loop():75] message_loop has been closed
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_065658-y2drrt7x/run-y2drrt7x.wandb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8cffe769ce5752ff5b38266363f81feab3b6b225efa2d1490b820e48fbf479dc
3
+ size 216034465
2026.02.27/06.56.39_train_llmbc_lowdim_box-close-v2/wandb/wandb-resume.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"run_id": "y2drrt7x"}
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/.hydra/config.yaml ADDED
@@ -0,0 +1,163 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: train_llmbc_lowdim
2
+ _target_: llmbc.workspace.train_llmbc_lowdim_workspace.TrainLLMBCLowdimWorkspace
3
+ obs_dim: ${task.obs_dim}
4
+ action_dim: ${task.action_dim}
5
+ task_name: ${task.name}
6
+ exp_name: default
7
+ model_name: ${llm.name}
8
+ horizon: 1
9
+ n_obs_steps: 1
10
+ n_action_steps: 1
11
+ n_latency_steps: 0
12
+ past_action_visible: false
13
+ llm_orig_expert_feedback: true
14
+ llm_do_sample: false
15
+ policy:
16
+ _target_: llmbc.policy.llmbc_lowdim_policy.LLMBCLowdimPolicy
17
+ model:
18
+ _target_: llmbc.model.policy.policy_mlp.PolicyMLP
19
+ input_size: ${eval:'${n_obs_steps}*${obs_dim}'}
20
+ hidden_size:
21
+ - 256
22
+ - 256
23
+ output_size: ${eval:'${n_action_steps}*${action_dim}'}
24
+ activation: relu
25
+ n_obs_steps: ${n_obs_steps}
26
+ n_action_steps: ${n_action_steps}
27
+ obs_dim: ${obs_dim}
28
+ action_dim: ${action_dim}
29
+ llm_discriminator:
30
+ _target_: llmbc.discriminator.llm_ce_discriminator.LLMCEDiscriminator
31
+ task_id: ${task_name}
32
+ llm_translator:
33
+ _target_: llmbc.translator.llm_translator.LLMTranslator
34
+ cfg: ${llm}
35
+ obs_dim: ${task.obs_dim}
36
+ action_dim: ${task.action_dim}
37
+ horizon: ${horizon}
38
+ n_obs_steps: ${n_obs_steps}
39
+ n_action_steps: ${n_action_steps}
40
+ loss_bc_weight: 1.0
41
+ loss_llm_weight: 0.001
42
+ horizon: ${horizon}
43
+ n_obs_steps: ${n_obs_steps}
44
+ n_action_steps: ${n_action_steps}
45
+ normalize_llm_loss: true
46
+ dataloader:
47
+ batch_size: 16
48
+ num_workers: 0
49
+ shuffle: true
50
+ pin_memory: false
51
+ persistent_workers: false
52
+ val_dataloader:
53
+ batch_size: 16
54
+ num_workers: 0
55
+ shuffle: true
56
+ pin_memory: false
57
+ persistent_workers: false
58
+ optimizer:
59
+ _target_: torch.optim.AdamW
60
+ lr: 0.01
61
+ betas:
62
+ - 0.95
63
+ - 0.999
64
+ eps: 1.0e-08
65
+ weight_decay: 1.0e-06
66
+ training:
67
+ device: cuda:0
68
+ seed: 46
69
+ debug: false
70
+ resume: false
71
+ lr_scheduler: cosine
72
+ lr_warmup_steps: 10
73
+ num_epochs: 1001
74
+ gradient_accumulate_every: 8
75
+ grad_norm_clip: 0.5
76
+ rollout_every: 5
77
+ checkpoint_every: 5
78
+ val_every: 1
79
+ sample_every: 5
80
+ sample_max_batch: 128
81
+ max_train_steps: null
82
+ max_val_steps: null
83
+ tqdm_interval_sec: 1.0
84
+ logging:
85
+ project: ${task.name}-training-llm-backbone
86
+ resume: true
87
+ mode: online
88
+ name: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
89
+ tags:
90
+ - ${name}
91
+ - ${task_name}
92
+ - ${exp_name}
93
+ id: null
94
+ group: null
95
+ checkpoint:
96
+ topk:
97
+ monitor_key: test_success_rate
98
+ mode: max
99
+ k: 5
100
+ format_str: epoch={epoch:04d}-test_success_rate={test_success_rate:.3f}.ckpt
101
+ save_last_ckpt: true
102
+ save_last_snapshot: false
103
+ multi_run:
104
+ run_dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
105
+ wandb_name_base: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
106
+ task:
107
+ name: box-close-v2
108
+ obs_dim: 9
109
+ action_dim: 4
110
+ env_runner:
111
+ _target_: llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner
112
+ env_name: llf-metaworld-box-close-v2
113
+ n_train: 10
114
+ n_test: 50
115
+ n_envs: 10
116
+ max_steps: 30
117
+ n_obs_steps: ${n_obs_steps}
118
+ n_action_steps: ${n_action_steps}
119
+ instruction_type: b
120
+ feedback_type:
121
+ - hp
122
+ - hn
123
+ - fp
124
+ visual: false
125
+ discount: 0.9
126
+ dataset:
127
+ _target_: llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset
128
+ data_path: datasets/box-close-v2-o4-mini.pt
129
+ data_path2: datasets/box-close-v2-o4-mini.pt
130
+ horizon: ${horizon}
131
+ pad_before: ${eval:'${n_obs_steps}-1'}
132
+ pad_after: ${eval:'${n_action_steps}-1'}
133
+ obs_eef_target: true
134
+ use_manual_normalizer: false
135
+ val_ratio: 0.1
136
+ dummy_normalizer: true
137
+ instructor:
138
+ _target_: llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor
139
+ llm:
140
+ name: Qwen/Qwen2-0.5B-Instruct
141
+ model_name: Qwen2-0.5B-Instruct
142
+ config_target: llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2Config
143
+ causal_lm_target: llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2ForCausalLM
144
+ use_quantization: false
145
+ use_joint_mlp_projector: true
146
+ llm_mode: ete-finetuned
147
+ finetune_mode: orig
148
+ checkpoint: data/outputs/2026.02.20/17.05.08_train_llm_lowdim_box-close-v2/Qwen/Qwen2-0.5B-Instruct-finetuned-box-close-v2/checkpoint-17670
149
+ max_length: 100
150
+ lora_config:
151
+ r: 32
152
+ lora_alpha: 64
153
+ lora_dropout: 0.05
154
+ bias: none
155
+ task_type: CAUSAL_LM
156
+ prompter:
157
+ _target_: llmbc.translator.prompter.qwen2_prompter.Qwen2Prompter
158
+ use_joint_mlp_projector: true
159
+ hydra:
160
+ job:
161
+ override_dirname: ${model_name}
162
+ run:
163
+ dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${model_name}
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/.hydra/hydra.yaml ADDED
@@ -0,0 +1,155 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
4
+ sweep:
5
+ dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
9
+ sweeper:
10
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
11
+ max_batch_size: null
12
+ params: null
13
+ help:
14
+ app_name: ${hydra.job.name}
15
+ header: '${hydra.help.app_name} is powered by Hydra.
16
+
17
+ '
18
+ footer: 'Powered by Hydra (https://hydra.cc)
19
+
20
+ Use --hydra-help to view Hydra specific help
21
+
22
+ '
23
+ template: '${hydra.help.header}
24
+
25
+ == Configuration groups ==
26
+
27
+ Compose your configuration from those groups (group=option)
28
+
29
+
30
+ $APP_CONFIG_GROUPS
31
+
32
+
33
+ == Config ==
34
+
35
+ Override anything in the config (foo.bar=value)
36
+
37
+
38
+ $CONFIG
39
+
40
+
41
+ ${hydra.help.footer}
42
+
43
+ '
44
+ hydra_help:
45
+ template: 'Hydra (${hydra.runtime.version})
46
+
47
+ See https://hydra.cc for more info.
48
+
49
+
50
+ == Flags ==
51
+
52
+ $FLAGS_HELP
53
+
54
+
55
+ == Configuration groups ==
56
+
57
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
58
+ to command line)
59
+
60
+
61
+ $HYDRA_CONFIG_GROUPS
62
+
63
+
64
+ Use ''--cfg hydra'' to Show the Hydra config.
65
+
66
+ '
67
+ hydra_help: ???
68
+ hydra_logging:
69
+ version: 1
70
+ formatters:
71
+ simple:
72
+ format: '[%(asctime)s][HYDRA] %(message)s'
73
+ handlers:
74
+ console:
75
+ class: logging.StreamHandler
76
+ formatter: simple
77
+ stream: ext://sys.stdout
78
+ root:
79
+ level: INFO
80
+ handlers:
81
+ - console
82
+ loggers:
83
+ logging_example:
84
+ level: DEBUG
85
+ disable_existing_loggers: false
86
+ job_logging:
87
+ version: 1
88
+ formatters:
89
+ simple:
90
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
91
+ handlers:
92
+ console:
93
+ class: logging.StreamHandler
94
+ formatter: simple
95
+ stream: ext://sys.stdout
96
+ file:
97
+ class: logging.FileHandler
98
+ formatter: simple
99
+ filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
100
+ root:
101
+ level: INFO
102
+ handlers:
103
+ - console
104
+ - file
105
+ disable_existing_loggers: false
106
+ env: {}
107
+ mode: RUN
108
+ searchpath: []
109
+ callbacks: {}
110
+ output_subdir: .hydra
111
+ overrides:
112
+ hydra:
113
+ - hydra.mode=RUN
114
+ task:
115
+ - training.seed=46
116
+ job:
117
+ name: train
118
+ chdir: null
119
+ override_dirname: training.seed=46
120
+ id: ???
121
+ num: ???
122
+ config_name: llmbc_box-close-v2_o4-mini_qwen2-0.5b-instruct.yaml
123
+ env_set: {}
124
+ env_copy: []
125
+ config:
126
+ override_dirname:
127
+ kv_sep: '='
128
+ item_sep: ','
129
+ exclude_keys: []
130
+ runtime:
131
+ version: 1.2.0
132
+ version_base: '1.2'
133
+ cwd: /work/u1131674/LLM-BC
134
+ config_sources:
135
+ - path: hydra.conf
136
+ schema: pkg
137
+ provider: hydra
138
+ - path: /work/u1131674/LLM-BC/config/llm_backbone
139
+ schema: file
140
+ provider: main
141
+ - path: ''
142
+ schema: structured
143
+ provider: schema
144
+ output_dir: /work/u1131674/LLM-BC/data/outputs/2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2
145
+ choices:
146
+ hydra/env: default
147
+ hydra/callbacks: null
148
+ hydra/job_logging: default
149
+ hydra/hydra_logging: default
150
+ hydra/hydra_help: default
151
+ hydra/help: default
152
+ hydra/sweeper: basic
153
+ hydra/launcher: basic
154
+ hydra/output: default
155
+ verbose: false
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/.hydra/overrides.yaml ADDED
@@ -0,0 +1 @@
 
 
1
+ - training.seed=46
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0245-test_success_rate=0.900.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1286a07b2e74190b551c00bd3d66481caac1959f69676d9fdfe9634a9641f12
3
+ size 864520
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0355-test_success_rate=0.860.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e9e118037042b9ae9d8c19e767c75b889a33d377e7efa874ada855f787277461
3
+ size 864520
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0710-test_success_rate=0.880.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c4930e7f51f19c05a8ff4257b8e8d9fe94dff6186ddab6f825bfc8a175fbede
3
+ size 864520
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0735-test_success_rate=0.900.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:12d49267431c378fe9a343bf63a48e960bc61c86f6b094f0ad3af530f07617e2
3
+ size 864520
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0905-test_success_rate=0.880.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4fd02d3c68d5bbf0d1df7588e04320a9df5e6d376f9519aebd0082f9d4437201
3
+ size 864520
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/checkpoints/latest.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:66a2f493a53f54a5b82379ffb5ba3db7e5c15b615879aeb7a013ffa6e859d335
3
+ size 864520
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/logs.json.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6603424a3f7793a35f58c0ceb80c5b75be2e8e8e0067916315d173bcf59b7349
3
+ size 52090530
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/train.log ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ [2026-02-27 07:17:41,140][numexpr.utils][INFO] - Note: detected 112 virtual cores but NumExpr set to maximum of 64, check "NUMEXPR_MAX_THREADS" environment variable.
2
+ [2026-02-27 07:17:41,140][numexpr.utils][INFO] - Note: NumExpr detected 112 cores but "NUMEXPR_MAX_THREADS" not set, so enforcing safe limit of 16.
3
+ [2026-02-27 07:17:41,140][numexpr.utils][INFO] - NumExpr defaulting to 16 threads.
4
+ [2026-02-27 07:17:47,727][datasets][INFO] - PyTorch version 2.2.2 available.
5
+ [2026-02-27 07:17:47,728][datasets][INFO] - TensorFlow version 2.15.1 available.
6
+ [2026-02-27 07:17:47,730][datasets][INFO] - JAX version 0.4.30 available.
7
+ [2026-02-27 07:17:57,803][absl][INFO] - MUJOCO_GL=osmesa, attempting to import specified OpenGL backend.
8
+ [2026-02-27 07:17:57,814][absl][INFO] - MuJoCo library version is: 2.3.7
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/debug-internal.log ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-02-27T07:17:58.907834279+08:00","level":"INFO","msg":"using version","core version":"0.18.6"}
2
+ {"time":"2026-02-27T07:17:58.907846558+08:00","level":"INFO","msg":"created symlink","path":"/work/u1131674/LLM-BC/data/outputs/2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/logs/debug-core.log"}
3
+ {"time":"2026-02-27T07:17:59.0238555+08:00","level":"INFO","msg":"created new stream","id":"318q5q4f"}
4
+ {"time":"2026-02-27T07:17:59.02388384+08:00","level":"INFO","msg":"stream: started","id":"318q5q4f"}
5
+ {"time":"2026-02-27T07:17:59.023912225+08:00","level":"INFO","msg":"sender: started","stream_id":"318q5q4f"}
6
+ {"time":"2026-02-27T07:17:59.023898184+08:00","level":"INFO","msg":"writer: Do: started","stream_id":{"value":"318q5q4f"}}
7
+ {"time":"2026-02-27T07:17:59.023906225+08:00","level":"INFO","msg":"handler: started","stream_id":{"value":"318q5q4f"}}
8
+ {"time":"2026-02-27T07:17:59.835488935+08:00","level":"INFO","msg":"Starting system monitor"}
9
+ {"time":"2026-02-28T02:32:59.956571601+08:00","level":"INFO","msg":"stream: closing","id":"318q5q4f"}
10
+ {"time":"2026-02-28T02:32:59.956600943+08:00","level":"INFO","msg":"Stopping system monitor"}
11
+ {"time":"2026-02-28T02:32:59.957022078+08:00","level":"INFO","msg":"Stopped system monitor"}
12
+ {"time":"2026-02-28T02:33:01.372096213+08:00","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
13
+ {"time":"2026-02-28T02:33:01.668699753+08:00","level":"INFO","msg":"handler: closed","stream_id":{"value":"318q5q4f"}}
14
+ {"time":"2026-02-28T02:33:01.668743922+08:00","level":"INFO","msg":"writer: Close: closed","stream_id":{"value":"318q5q4f"}}
15
+ {"time":"2026-02-28T02:33:01.66877286+08:00","level":"INFO","msg":"sender: closed","stream_id":"318q5q4f"}
16
+ {"time":"2026-02-28T02:33:01.668795289+08:00","level":"INFO","msg":"stream: closed","id":"318q5q4f"}
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/debug.log ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Current SDK version is 0.18.6
2
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Configure stats pid to 1783449
3
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Loading settings from /home/u1131674/.config/wandb/settings
4
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Loading settings from /work/u1131674/LLM-BC/wandb/settings
5
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Loading settings from environment variables: {}
6
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Applying setup settings: {'mode': 'online', '_disable_service': None}
7
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Inferring run settings from compute environment: {'program_relpath': 'train.py', 'program_abspath': '/work/u1131674/LLM-BC/train.py', 'program': '/work/u1131674/LLM-BC/./train.py'}
8
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Applying login settings: {}
9
+ 2026-02-27 07:17:58,904 INFO MainThread:1783449 [wandb_init.py:_log_setup():533] Logging user logs to /work/u1131674/LLM-BC/data/outputs/2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/logs/debug.log
10
+ 2026-02-27 07:17:58,904 INFO MainThread:1783449 [wandb_init.py:_log_setup():534] Logging internal logs to /work/u1131674/LLM-BC/data/outputs/2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/logs/debug-internal.log
11
+ 2026-02-27 07:17:58,904 INFO MainThread:1783449 [wandb_init.py:init():619] calling init triggers
12
+ 2026-02-27 07:17:58,904 INFO MainThread:1783449 [wandb_init.py:init():626] wandb.init called with sweep_config: {}
13
+ config: {'name': 'train_llmbc_lowdim', '_target_': 'llmbc.workspace.train_llmbc_lowdim_workspace.TrainLLMBCLowdimWorkspace', 'obs_dim': 9, 'action_dim': 4, 'task_name': 'box-close-v2', 'exp_name': 'default', 'model_name': 'Qwen/Qwen2-0.5B-Instruct', 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1, 'n_latency_steps': 0, 'past_action_visible': False, 'llm_orig_expert_feedback': True, 'llm_do_sample': False, 'policy': {'_target_': 'llmbc.policy.llmbc_lowdim_policy.LLMBCLowdimPolicy', 'model': {'_target_': 'llmbc.model.policy.policy_mlp.PolicyMLP', 'input_size': 9, 'hidden_size': [256, 256], 'output_size': 4, 'activation': 'relu', 'n_obs_steps': 1, 'n_action_steps': 1}, 'obs_dim': 9, 'action_dim': 4, 'llm_discriminator': {'_target_': 'llmbc.discriminator.llm_ce_discriminator.LLMCEDiscriminator', 'task_id': 'box-close-v2', 'llm_translator': {'_target_': 'llmbc.translator.llm_translator.LLMTranslator', 'cfg': {'name': 'Qwen/Qwen2-0.5B-Instruct', 'model_name': 'Qwen2-0.5B-Instruct', 'config_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2Config', 'causal_lm_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2ForCausalLM', 'use_quantization': False, 'use_joint_mlp_projector': True, 'llm_mode': 'ete-finetuned', 'finetune_mode': 'orig', 'checkpoint': 'data/outputs/2026.02.20/17.05.08_train_llm_lowdim_box-close-v2/Qwen/Qwen2-0.5B-Instruct-finetuned-box-close-v2/checkpoint-17670', 'max_length': 100, 'lora_config': {'r': 32, 'lora_alpha': 64, 'lora_dropout': 0.05, 'bias': 'none', 'task_type': 'CAUSAL_LM'}, 'prompter': {'_target_': 'llmbc.translator.prompter.qwen2_prompter.Qwen2Prompter', 'use_joint_mlp_projector': True}, 'hydra': {'job': {'override_dirname': 'Qwen/Qwen2-0.5B-Instruct'}, 'run': {'dir': 'data/outputs/2026.02.27/07.17.38_Qwen/Qwen2-0.5B-Instruct'}}}, 'obs_dim': 9, 'action_dim': 4, 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1}}, 'loss_bc_weight': 1.0, 'loss_llm_weight': 0.001, 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1, 'normalize_llm_loss': True}, 'dataloader': {'batch_size': 16, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'val_dataloader': {'batch_size': 16, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'optimizer': {'_target_': 'torch.optim.AdamW', 'lr': 0.01, 'betas': [0.95, 0.999], 'eps': 1e-08, 'weight_decay': 1e-06}, 'training': {'device': 'cuda:0', 'seed': 46, 'debug': False, 'resume': False, 'lr_scheduler': 'cosine', 'lr_warmup_steps': 10, 'num_epochs': 1001, 'gradient_accumulate_every': 8, 'grad_norm_clip': 0.5, 'rollout_every': 5, 'checkpoint_every': 5, 'val_every': 1, 'sample_every': 5, 'sample_max_batch': 128, 'max_train_steps': None, 'max_val_steps': None, 'tqdm_interval_sec': 1.0}, 'logging': {'project': 'box-close-v2-training-llm-backbone', 'resume': True, 'mode': 'online', 'name': '2026.02.27-07.17.38_train_llmbc_lowdim_box-close-v2', 'tags': ['train_llmbc_lowdim', 'box-close-v2', 'default'], 'id': None, 'group': None}, 'checkpoint': {'topk': {'monitor_key': 'test_success_rate', 'mode': 'max', 'k': 5, 'format_str': 'epoch={epoch:04d}-test_success_rate={test_success_rate:.3f}.ckpt'}, 'save_last_ckpt': True, 'save_last_snapshot': False}, 'multi_run': {'run_dir': 'data/outputs/2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2', 'wandb_name_base': '2026.02.27-07.17.38_train_llmbc_lowdim_box-close-v2'}, 'task': {'name': 'box-close-v2', 'obs_dim': 9, 'action_dim': 4, 'env_runner': {'_target_': 'llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner', 'env_name': 'llf-metaworld-box-close-v2', 'n_train': 10, 'n_test': 50, 'n_envs': 10, 'max_steps': 30, 'n_obs_steps': 1, 'n_action_steps': 1, 'instruction_type': 'b', 'feedback_type': ['hp', 'hn', 'fp'], 'visual': False, 'discount': 0.9}, 'dataset': {'_target_': 'llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset', 'data_path': 'datasets/box-close-v2-o4-mini.pt', 'data_path2': 'datasets/box-close-v2-o4-mini.pt', 'horizon': 1, 'pad_before': 0, 'pad_after': 0, 'obs_eef_target': True, 'use_manual_normalizer': False, 'val_ratio': 0.1, 'dummy_normalizer': True}, 'instructor': {'_target_': 'llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor'}}, 'llm': {'name': 'Qwen/Qwen2-0.5B-Instruct', 'model_name': 'Qwen2-0.5B-Instruct', 'config_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2Config', 'causal_lm_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2ForCausalLM', 'use_quantization': False, 'use_joint_mlp_projector': True, 'llm_mode': 'ete-finetuned', 'finetune_mode': 'orig', 'checkpoint': 'data/outputs/2026.02.20/17.05.08_train_llm_lowdim_box-close-v2/Qwen/Qwen2-0.5B-Instruct-finetuned-box-close-v2/checkpoint-17670', 'max_length': 100, 'lora_config': {'r': 32, 'lora_alpha': 64, 'lora_dropout': 0.05, 'bias': 'none', 'task_type': 'CAUSAL_LM'}, 'prompter': {'_target_': 'llmbc.translator.prompter.qwen2_prompter.Qwen2Prompter', 'use_joint_mlp_projector': True}, 'hydra': {'job': {'override_dirname': 'Qwen/Qwen2-0.5B-Instruct'}, 'run': {'dir': 'data/outputs/2026.02.27/07.17.38_Qwen/Qwen2-0.5B-Instruct'}}}}
14
+ 2026-02-27 07:17:58,904 INFO MainThread:1783449 [wandb_init.py:init():669] starting backend
15
+ 2026-02-27 07:17:58,904 INFO MainThread:1783449 [wandb_init.py:init():673] sending inform_init request
16
+ 2026-02-27 07:17:58,905 INFO MainThread:1783449 [backend.py:_multiprocessing_setup():104] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
17
+ 2026-02-27 07:17:58,906 INFO MainThread:1783449 [wandb_init.py:init():686] backend started and connected
18
+ 2026-02-27 07:17:58,915 INFO MainThread:1783449 [wandb_init.py:init():781] updated telemetry
19
+ 2026-02-27 07:17:58,969 INFO MainThread:1783449 [wandb_init.py:init():814] communicating run to backend with 90.0 second timeout
20
+ 2026-02-27 07:17:59,830 INFO MainThread:1783449 [wandb_init.py:init():867] starting run threads in backend
21
+ 2026-02-27 07:18:00,377 INFO MainThread:1783449 [wandb_run.py:_console_start():2451] atexit reg
22
+ 2026-02-27 07:18:00,377 INFO MainThread:1783449 [wandb_run.py:_redirect():2299] redirect: wrap_raw
23
+ 2026-02-27 07:18:00,377 INFO MainThread:1783449 [wandb_run.py:_redirect():2364] Wrapping output streams.
24
+ 2026-02-27 07:18:00,377 INFO MainThread:1783449 [wandb_run.py:_redirect():2389] Redirects installed.
25
+ 2026-02-27 07:18:00,380 INFO MainThread:1783449 [wandb_init.py:init():911] run started, returning control to user process
26
+ 2026-02-27 07:18:00,381 INFO MainThread:1783449 [wandb_run.py:_config_callback():1389] config_cb None None {'output_dir': '/work/u1131674/LLM-BC/data/outputs/2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2'}
27
+ 2026-02-28 02:32:59,956 WARNING MsgRouterThr:1783449 [router.py:message_loop():75] message_loop has been closed
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/files/config.yaml ADDED
@@ -0,0 +1,271 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _target_:
2
+ value: llmbc.workspace.train_llmbc_lowdim_workspace.TrainLLMBCLowdimWorkspace
3
+ _wandb:
4
+ value:
5
+ cli_version: 0.18.6
6
+ m: []
7
+ python_version: 3.9.25
8
+ t:
9
+ "1":
10
+ - 1
11
+ - 2
12
+ - 3
13
+ - 5
14
+ - 11
15
+ - 12
16
+ - 41
17
+ - 49
18
+ - 50
19
+ - 51
20
+ - 53
21
+ - 55
22
+ - 71
23
+ - 83
24
+ - 95
25
+ - 98
26
+ - 100
27
+ - 105
28
+ "2":
29
+ - 1
30
+ - 2
31
+ - 3
32
+ - 5
33
+ - 11
34
+ - 12
35
+ - 41
36
+ - 49
37
+ - 50
38
+ - 51
39
+ - 53
40
+ - 55
41
+ - 71
42
+ - 83
43
+ - 95
44
+ - 98
45
+ - 100
46
+ - 105
47
+ "3":
48
+ - 13
49
+ - 15
50
+ - 16
51
+ - 23
52
+ - 55
53
+ - 61
54
+ "4": 3.9.25
55
+ "5": 0.18.6
56
+ "6": 4.47.1
57
+ "8":
58
+ - 5
59
+ "12": 0.18.6
60
+ "13": linux-x86_64
61
+ action_dim:
62
+ value: 4
63
+ checkpoint:
64
+ value:
65
+ save_last_ckpt: true
66
+ save_last_snapshot: false
67
+ topk:
68
+ format_str: epoch={epoch:04d}-test_success_rate={test_success_rate:.3f}.ckpt
69
+ k: 5
70
+ mode: max
71
+ monitor_key: test_success_rate
72
+ dataloader:
73
+ value:
74
+ batch_size: 16
75
+ num_workers: 0
76
+ persistent_workers: false
77
+ pin_memory: false
78
+ shuffle: true
79
+ exp_name:
80
+ value: default
81
+ horizon:
82
+ value: 1
83
+ llm:
84
+ value:
85
+ causal_lm_target: llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2ForCausalLM
86
+ checkpoint: data/outputs/2026.02.20/17.05.08_train_llm_lowdim_box-close-v2/Qwen/Qwen2-0.5B-Instruct-finetuned-box-close-v2/checkpoint-17670
87
+ config_target: llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2Config
88
+ finetune_mode: orig
89
+ hydra:
90
+ job:
91
+ override_dirname: Qwen/Qwen2-0.5B-Instruct
92
+ run:
93
+ dir: data/outputs/2026.02.27/07.17.38_Qwen/Qwen2-0.5B-Instruct
94
+ llm_mode: ete-finetuned
95
+ lora_config:
96
+ bias: none
97
+ lora_alpha: 64
98
+ lora_dropout: 0.05
99
+ r: 32
100
+ task_type: CAUSAL_LM
101
+ max_length: 100
102
+ model_name: Qwen2-0.5B-Instruct
103
+ name: Qwen/Qwen2-0.5B-Instruct
104
+ prompter:
105
+ _target_: llmbc.translator.prompter.qwen2_prompter.Qwen2Prompter
106
+ use_joint_mlp_projector: true
107
+ use_joint_mlp_projector: true
108
+ use_quantization: false
109
+ llm_do_sample:
110
+ value: false
111
+ llm_orig_expert_feedback:
112
+ value: true
113
+ logging:
114
+ value:
115
+ group: null
116
+ id: null
117
+ mode: online
118
+ name: 2026.02.27-07.17.38_train_llmbc_lowdim_box-close-v2
119
+ project: box-close-v2-training-llm-backbone
120
+ resume: true
121
+ tags:
122
+ - train_llmbc_lowdim
123
+ - box-close-v2
124
+ - default
125
+ model_name:
126
+ value: Qwen/Qwen2-0.5B-Instruct
127
+ multi_run:
128
+ value:
129
+ run_dir: data/outputs/2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2
130
+ wandb_name_base: 2026.02.27-07.17.38_train_llmbc_lowdim_box-close-v2
131
+ n_action_steps:
132
+ value: 1
133
+ n_latency_steps:
134
+ value: 0
135
+ n_obs_steps:
136
+ value: 1
137
+ name:
138
+ value: train_llmbc_lowdim
139
+ obs_dim:
140
+ value: 9
141
+ optimizer:
142
+ value:
143
+ _target_: torch.optim.AdamW
144
+ betas:
145
+ - 0.95
146
+ - 0.999
147
+ eps: 1e-08
148
+ lr: 0.01
149
+ weight_decay: 1e-06
150
+ output_dir:
151
+ value: /work/u1131674/LLM-BC/data/outputs/2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2
152
+ past_action_visible:
153
+ value: false
154
+ policy:
155
+ value:
156
+ _target_: llmbc.policy.llmbc_lowdim_policy.LLMBCLowdimPolicy
157
+ action_dim: 4
158
+ horizon: 1
159
+ llm_discriminator:
160
+ _target_: llmbc.discriminator.llm_ce_discriminator.LLMCEDiscriminator
161
+ llm_translator:
162
+ _target_: llmbc.translator.llm_translator.LLMTranslator
163
+ action_dim: 4
164
+ cfg:
165
+ causal_lm_target: llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2ForCausalLM
166
+ checkpoint: data/outputs/2026.02.20/17.05.08_train_llm_lowdim_box-close-v2/Qwen/Qwen2-0.5B-Instruct-finetuned-box-close-v2/checkpoint-17670
167
+ config_target: llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2Config
168
+ finetune_mode: orig
169
+ hydra:
170
+ job:
171
+ override_dirname: Qwen/Qwen2-0.5B-Instruct
172
+ run:
173
+ dir: data/outputs/2026.02.27/07.17.38_Qwen/Qwen2-0.5B-Instruct
174
+ llm_mode: ete-finetuned
175
+ lora_config:
176
+ bias: none
177
+ lora_alpha: 64
178
+ lora_dropout: 0.05
179
+ r: 32
180
+ task_type: CAUSAL_LM
181
+ max_length: 100
182
+ model_name: Qwen2-0.5B-Instruct
183
+ name: Qwen/Qwen2-0.5B-Instruct
184
+ prompter:
185
+ _target_: llmbc.translator.prompter.qwen2_prompter.Qwen2Prompter
186
+ use_joint_mlp_projector: true
187
+ use_joint_mlp_projector: true
188
+ use_quantization: false
189
+ horizon: 1
190
+ n_action_steps: 1
191
+ n_obs_steps: 1
192
+ obs_dim: 9
193
+ task_id: box-close-v2
194
+ loss_bc_weight: 1
195
+ loss_llm_weight: 0.001
196
+ model:
197
+ _target_: llmbc.model.policy.policy_mlp.PolicyMLP
198
+ activation: relu
199
+ hidden_size:
200
+ - 256
201
+ - 256
202
+ input_size: 9
203
+ n_action_steps: 1
204
+ n_obs_steps: 1
205
+ output_size: 4
206
+ n_action_steps: 1
207
+ n_obs_steps: 1
208
+ normalize_llm_loss: true
209
+ obs_dim: 9
210
+ task:
211
+ value:
212
+ action_dim: 4
213
+ dataset:
214
+ _target_: llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset
215
+ data_path: datasets/box-close-v2-o4-mini.pt
216
+ data_path2: datasets/box-close-v2-o4-mini.pt
217
+ dummy_normalizer: true
218
+ horizon: 1
219
+ obs_eef_target: true
220
+ pad_after: 0
221
+ pad_before: 0
222
+ use_manual_normalizer: false
223
+ val_ratio: 0.1
224
+ env_runner:
225
+ _target_: llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner
226
+ discount: 0.9
227
+ env_name: llf-metaworld-box-close-v2
228
+ feedback_type:
229
+ - hp
230
+ - hn
231
+ - fp
232
+ instruction_type: b
233
+ max_steps: 30
234
+ n_action_steps: 1
235
+ n_envs: 10
236
+ n_obs_steps: 1
237
+ n_test: 50
238
+ n_train: 10
239
+ visual: false
240
+ instructor:
241
+ _target_: llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor
242
+ name: box-close-v2
243
+ obs_dim: 9
244
+ task_name:
245
+ value: box-close-v2
246
+ training:
247
+ value:
248
+ checkpoint_every: 5
249
+ debug: false
250
+ device: cuda:0
251
+ grad_norm_clip: 0.5
252
+ gradient_accumulate_every: 8
253
+ lr_scheduler: cosine
254
+ lr_warmup_steps: 10
255
+ max_train_steps: null
256
+ max_val_steps: null
257
+ num_epochs: 1001
258
+ resume: false
259
+ rollout_every: 5
260
+ sample_every: 5
261
+ sample_max_batch: 128
262
+ seed: 46
263
+ tqdm_interval_sec: 1
264
+ val_every: 1
265
+ val_dataloader:
266
+ value:
267
+ batch_size: 16
268
+ num_workers: 0
269
+ persistent_workers: false
270
+ pin_memory: false
271
+ shuffle: true
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/files/output.log ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ Eval MetaworldLowdimRunner 1/6: 0%| | 0/30 [00:00<?, ?it/s]/work/u1131674/LLM-BC/llmbc/common/llfbench_util.py:39: UserWarning: Creating a tensor from a list of numpy.ndarrays is extremely slow. Please consider converting the list to a single numpy.ndarray with numpy.array() before converting to a tensor. (Triggered internally at ../torch/csrc/utils/tensor_new.cpp:275.)
2
+ obs = torch.tensor(obs, dtype=torch.float32).unsqueeze(dim=0).to(device)
3
+
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/files/wandb-metadata.json ADDED
@@ -0,0 +1,107 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-4.18.0-513.24.1.el8_9.x86_64-x86_64-with-glibc2.28",
3
+ "python": "3.9.25",
4
+ "startedAt": "2026-02-26T23:17:58.906178Z",
5
+ "args": [
6
+ "--config-path",
7
+ "config/llm_backbone",
8
+ "--config-name",
9
+ "llmbc_box-close-v2_o4-mini_qwen2-0.5b-instruct.yaml",
10
+ "training.seed=46"
11
+ ],
12
+ "program": "/work/u1131674/LLM-BC/./train.py",
13
+ "codePath": "train.py",
14
+ "git": {
15
+ "remote": "https://github.com/CHYang25/LLM-BC.git",
16
+ "commit": "155285319c80dc3cf3f63496e1795b29a54a9377"
17
+ },
18
+ "email": "chris920325@gmail.com",
19
+ "root": "/work/u1131674/LLM-BC/data/outputs/2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2",
20
+ "host": "hgpn18",
21
+ "username": "u1131674",
22
+ "executable": "/home/u1131674/.conda/envs/llm-bc/bin/python3",
23
+ "codePathLocal": "train.py",
24
+ "cpu_count": 112,
25
+ "cpu_count_logical": 112,
26
+ "gpu": "NVIDIA H100 80GB HBM3",
27
+ "gpu_count": 1,
28
+ "disk": {
29
+ "/": {
30
+ "total": "1918024196096",
31
+ "used": "227001356288"
32
+ }
33
+ },
34
+ "memory": {
35
+ "total": "2163686170624"
36
+ },
37
+ "cpu": {
38
+ "count": 112,
39
+ "countLogical": 112
40
+ },
41
+ "gpu_nvidia": [
42
+ {
43
+ "name": "NVIDIA H100 80GB HBM3",
44
+ "memoryTotal": "85520809984",
45
+ "cudaCores": 16896,
46
+ "architecture": "Hopper"
47
+ }
48
+ ],
49
+ "slurm": {
50
+ "cluster_name": "hpc",
51
+ "conf": "/etc/slurm/slurm.conf",
52
+ "cpu_bind": "quiet,mask_cpu:0x00F0000000001E00000000000000",
53
+ "cpu_bind_list": "0x00F0000000001E00000000000000",
54
+ "cpu_bind_type": "mask_cpu:",
55
+ "cpu_bind_verbose": "quiet",
56
+ "cpus_on_node": "8",
57
+ "cpus_per_task": "8",
58
+ "distribution": "cyclic,pack",
59
+ "gpus_on_node": "1",
60
+ "gpus_per_node": "1",
61
+ "gtids": "0",
62
+ "job_account": "mst114558",
63
+ "job_cpus_per_node": "8",
64
+ "job_end_time": "1772320656",
65
+ "job_gid": "106773",
66
+ "job_group": "MST114558",
67
+ "job_id": "124195",
68
+ "job_name": "python3",
69
+ "job_nodelist": "hgpn18",
70
+ "job_num_nodes": "1",
71
+ "job_partition": "normal",
72
+ "job_qos": "normal",
73
+ "job_start_time": "1772147856",
74
+ "job_uid": "41408",
75
+ "job_user": "u1131674",
76
+ "jobid": "124195",
77
+ "launch_node_ipaddr": "172.21.101.1",
78
+ "localid": "0",
79
+ "mem_per_node": "204800",
80
+ "nnodes": "1",
81
+ "nodeid": "0",
82
+ "nodelist": "hgpn18",
83
+ "nprocs": "1",
84
+ "ntasks": "1",
85
+ "prio_process": "0",
86
+ "procid": "0",
87
+ "srun_comm_host": "172.21.101.1",
88
+ "srun_comm_port": "46455",
89
+ "step_gpus": "5",
90
+ "step_id": "0",
91
+ "step_launcher_port": "46455",
92
+ "step_nodelist": "hgpn18",
93
+ "step_num_nodes": "1",
94
+ "step_num_tasks": "1",
95
+ "step_tasks_per_node": "1",
96
+ "stepid": "0",
97
+ "submit_dir": "/work/u1131674/LLM-BC",
98
+ "submit_host": "cbi-lgn01",
99
+ "task_pid": "1783449",
100
+ "tasks_per_node": "1",
101
+ "topology_addr": "ibsw1.hgpn18",
102
+ "topology_addr_pattern": "switch.node",
103
+ "tres_per_task": "cpu:8",
104
+ "umask": "0022"
105
+ },
106
+ "cudaVersion": "12.4"
107
+ }
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/files/wandb-summary.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"train/success_rate":0.9,"epoch":1000,"_step":239238,"train_loss":0.00030461582503041703,"grad_norm":0.012254353612661362,"lr":2.7610336239547452e-11,"test/mean_score":6.500273968629853,"train/mean_score":7.373033225625792,"train/cumulative_reward":13.32605421766263,"test/cumulative_reward":12.876683728156026,"val_loss":0.00016958703054115176,"train_loss_llm":0.0001508244313299656,"_runtime":69301.05040989,"_timestamp":1.7722171776507635e+09,"train_loss_bc":5.699055691366084e-06,"_wandb":{"runtime":69301},"test/success_rate":0.7,"global_step":239238,"train_action_mse_error":1.5991279724403284e-05}
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/logs/debug-core.log ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-02-27T07:17:58.186223614+08:00","level":"INFO","msg":"started logging, with flags","port-filename":"/tmp/tmpqiwzdbsx/port-1783449.txt","pid":1783449,"debug":false,"disable-analytics":false}
2
+ {"time":"2026-02-27T07:17:58.186243292+08:00","level":"INFO","msg":"FeatureState","shutdownOnParentExitEnabled":false}
3
+ {"time":"2026-02-27T07:17:58.186586009+08:00","level":"INFO","msg":"Will exit if parent process dies.","ppid":1783449}
4
+ {"time":"2026-02-27T07:17:58.186581492+08:00","level":"INFO","msg":"server is running","addr":{"IP":"127.0.0.1","Port":40087,"Zone":""}}
5
+ {"time":"2026-02-27T07:17:58.38035843+08:00","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"127.0.0.1:33640"}
6
+ {"time":"2026-02-27T07:17:58.906733227+08:00","level":"INFO","msg":"handleInformInit: received","streamId":"318q5q4f","id":"127.0.0.1:33640"}
7
+ {"time":"2026-02-27T07:17:59.023889783+08:00","level":"INFO","msg":"handleInformInit: stream started","streamId":"318q5q4f","id":"127.0.0.1:33640"}
8
+ {"time":"2026-02-28T02:32:59.956509095+08:00","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"127.0.0.1:33640"}
9
+ {"time":"2026-02-28T02:32:59.95658987+08:00","level":"INFO","msg":"server is shutting down"}
10
+ {"time":"2026-02-28T02:32:59.956576286+08:00","level":"INFO","msg":"connection: Close: initiating connection closure","id":"127.0.0.1:33640"}
11
+ {"time":"2026-02-28T02:32:59.956675328+08:00","level":"INFO","msg":"connection: Close: connection successfully closed","id":"127.0.0.1:33640"}
12
+ {"time":"2026-02-28T02:33:01.669014177+08:00","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"127.0.0.1:33640"}
13
+ {"time":"2026-02-28T02:33:01.669031782+08:00","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"127.0.0.1:33640"}
14
+ {"time":"2026-02-28T02:33:01.669043684+08:00","level":"INFO","msg":"server is closed"}
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/logs/debug-internal.log ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-02-27T07:17:58.907834279+08:00","level":"INFO","msg":"using version","core version":"0.18.6"}
2
+ {"time":"2026-02-27T07:17:58.907846558+08:00","level":"INFO","msg":"created symlink","path":"/work/u1131674/LLM-BC/data/outputs/2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/logs/debug-core.log"}
3
+ {"time":"2026-02-27T07:17:59.0238555+08:00","level":"INFO","msg":"created new stream","id":"318q5q4f"}
4
+ {"time":"2026-02-27T07:17:59.02388384+08:00","level":"INFO","msg":"stream: started","id":"318q5q4f"}
5
+ {"time":"2026-02-27T07:17:59.023912225+08:00","level":"INFO","msg":"sender: started","stream_id":"318q5q4f"}
6
+ {"time":"2026-02-27T07:17:59.023898184+08:00","level":"INFO","msg":"writer: Do: started","stream_id":{"value":"318q5q4f"}}
7
+ {"time":"2026-02-27T07:17:59.023906225+08:00","level":"INFO","msg":"handler: started","stream_id":{"value":"318q5q4f"}}
8
+ {"time":"2026-02-27T07:17:59.835488935+08:00","level":"INFO","msg":"Starting system monitor"}
9
+ {"time":"2026-02-28T02:32:59.956571601+08:00","level":"INFO","msg":"stream: closing","id":"318q5q4f"}
10
+ {"time":"2026-02-28T02:32:59.956600943+08:00","level":"INFO","msg":"Stopping system monitor"}
11
+ {"time":"2026-02-28T02:32:59.957022078+08:00","level":"INFO","msg":"Stopped system monitor"}
12
+ {"time":"2026-02-28T02:33:01.372096213+08:00","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
13
+ {"time":"2026-02-28T02:33:01.668699753+08:00","level":"INFO","msg":"handler: closed","stream_id":{"value":"318q5q4f"}}
14
+ {"time":"2026-02-28T02:33:01.668743922+08:00","level":"INFO","msg":"writer: Close: closed","stream_id":{"value":"318q5q4f"}}
15
+ {"time":"2026-02-28T02:33:01.66877286+08:00","level":"INFO","msg":"sender: closed","stream_id":"318q5q4f"}
16
+ {"time":"2026-02-28T02:33:01.668795289+08:00","level":"INFO","msg":"stream: closed","id":"318q5q4f"}
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/logs/debug.log ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Current SDK version is 0.18.6
2
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Configure stats pid to 1783449
3
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Loading settings from /home/u1131674/.config/wandb/settings
4
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Loading settings from /work/u1131674/LLM-BC/wandb/settings
5
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Loading settings from environment variables: {}
6
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Applying setup settings: {'mode': 'online', '_disable_service': None}
7
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Inferring run settings from compute environment: {'program_relpath': 'train.py', 'program_abspath': '/work/u1131674/LLM-BC/train.py', 'program': '/work/u1131674/LLM-BC/./train.py'}
8
+ 2026-02-27 07:17:58,903 INFO MainThread:1783449 [wandb_setup.py:_flush():79] Applying login settings: {}
9
+ 2026-02-27 07:17:58,904 INFO MainThread:1783449 [wandb_init.py:_log_setup():533] Logging user logs to /work/u1131674/LLM-BC/data/outputs/2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/logs/debug.log
10
+ 2026-02-27 07:17:58,904 INFO MainThread:1783449 [wandb_init.py:_log_setup():534] Logging internal logs to /work/u1131674/LLM-BC/data/outputs/2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/logs/debug-internal.log
11
+ 2026-02-27 07:17:58,904 INFO MainThread:1783449 [wandb_init.py:init():619] calling init triggers
12
+ 2026-02-27 07:17:58,904 INFO MainThread:1783449 [wandb_init.py:init():626] wandb.init called with sweep_config: {}
13
+ config: {'name': 'train_llmbc_lowdim', '_target_': 'llmbc.workspace.train_llmbc_lowdim_workspace.TrainLLMBCLowdimWorkspace', 'obs_dim': 9, 'action_dim': 4, 'task_name': 'box-close-v2', 'exp_name': 'default', 'model_name': 'Qwen/Qwen2-0.5B-Instruct', 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1, 'n_latency_steps': 0, 'past_action_visible': False, 'llm_orig_expert_feedback': True, 'llm_do_sample': False, 'policy': {'_target_': 'llmbc.policy.llmbc_lowdim_policy.LLMBCLowdimPolicy', 'model': {'_target_': 'llmbc.model.policy.policy_mlp.PolicyMLP', 'input_size': 9, 'hidden_size': [256, 256], 'output_size': 4, 'activation': 'relu', 'n_obs_steps': 1, 'n_action_steps': 1}, 'obs_dim': 9, 'action_dim': 4, 'llm_discriminator': {'_target_': 'llmbc.discriminator.llm_ce_discriminator.LLMCEDiscriminator', 'task_id': 'box-close-v2', 'llm_translator': {'_target_': 'llmbc.translator.llm_translator.LLMTranslator', 'cfg': {'name': 'Qwen/Qwen2-0.5B-Instruct', 'model_name': 'Qwen2-0.5B-Instruct', 'config_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2Config', 'causal_lm_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2ForCausalLM', 'use_quantization': False, 'use_joint_mlp_projector': True, 'llm_mode': 'ete-finetuned', 'finetune_mode': 'orig', 'checkpoint': 'data/outputs/2026.02.20/17.05.08_train_llm_lowdim_box-close-v2/Qwen/Qwen2-0.5B-Instruct-finetuned-box-close-v2/checkpoint-17670', 'max_length': 100, 'lora_config': {'r': 32, 'lora_alpha': 64, 'lora_dropout': 0.05, 'bias': 'none', 'task_type': 'CAUSAL_LM'}, 'prompter': {'_target_': 'llmbc.translator.prompter.qwen2_prompter.Qwen2Prompter', 'use_joint_mlp_projector': True}, 'hydra': {'job': {'override_dirname': 'Qwen/Qwen2-0.5B-Instruct'}, 'run': {'dir': 'data/outputs/2026.02.27/07.17.38_Qwen/Qwen2-0.5B-Instruct'}}}, 'obs_dim': 9, 'action_dim': 4, 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1}}, 'loss_bc_weight': 1.0, 'loss_llm_weight': 0.001, 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1, 'normalize_llm_loss': True}, 'dataloader': {'batch_size': 16, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'val_dataloader': {'batch_size': 16, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'optimizer': {'_target_': 'torch.optim.AdamW', 'lr': 0.01, 'betas': [0.95, 0.999], 'eps': 1e-08, 'weight_decay': 1e-06}, 'training': {'device': 'cuda:0', 'seed': 46, 'debug': False, 'resume': False, 'lr_scheduler': 'cosine', 'lr_warmup_steps': 10, 'num_epochs': 1001, 'gradient_accumulate_every': 8, 'grad_norm_clip': 0.5, 'rollout_every': 5, 'checkpoint_every': 5, 'val_every': 1, 'sample_every': 5, 'sample_max_batch': 128, 'max_train_steps': None, 'max_val_steps': None, 'tqdm_interval_sec': 1.0}, 'logging': {'project': 'box-close-v2-training-llm-backbone', 'resume': True, 'mode': 'online', 'name': '2026.02.27-07.17.38_train_llmbc_lowdim_box-close-v2', 'tags': ['train_llmbc_lowdim', 'box-close-v2', 'default'], 'id': None, 'group': None}, 'checkpoint': {'topk': {'monitor_key': 'test_success_rate', 'mode': 'max', 'k': 5, 'format_str': 'epoch={epoch:04d}-test_success_rate={test_success_rate:.3f}.ckpt'}, 'save_last_ckpt': True, 'save_last_snapshot': False}, 'multi_run': {'run_dir': 'data/outputs/2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2', 'wandb_name_base': '2026.02.27-07.17.38_train_llmbc_lowdim_box-close-v2'}, 'task': {'name': 'box-close-v2', 'obs_dim': 9, 'action_dim': 4, 'env_runner': {'_target_': 'llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner', 'env_name': 'llf-metaworld-box-close-v2', 'n_train': 10, 'n_test': 50, 'n_envs': 10, 'max_steps': 30, 'n_obs_steps': 1, 'n_action_steps': 1, 'instruction_type': 'b', 'feedback_type': ['hp', 'hn', 'fp'], 'visual': False, 'discount': 0.9}, 'dataset': {'_target_': 'llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset', 'data_path': 'datasets/box-close-v2-o4-mini.pt', 'data_path2': 'datasets/box-close-v2-o4-mini.pt', 'horizon': 1, 'pad_before': 0, 'pad_after': 0, 'obs_eef_target': True, 'use_manual_normalizer': False, 'val_ratio': 0.1, 'dummy_normalizer': True}, 'instructor': {'_target_': 'llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor'}}, 'llm': {'name': 'Qwen/Qwen2-0.5B-Instruct', 'model_name': 'Qwen2-0.5B-Instruct', 'config_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2Config', 'causal_lm_target': 'llmbc.model.llm.qwen2_lowdim_model.LowdimQwen2ForCausalLM', 'use_quantization': False, 'use_joint_mlp_projector': True, 'llm_mode': 'ete-finetuned', 'finetune_mode': 'orig', 'checkpoint': 'data/outputs/2026.02.20/17.05.08_train_llm_lowdim_box-close-v2/Qwen/Qwen2-0.5B-Instruct-finetuned-box-close-v2/checkpoint-17670', 'max_length': 100, 'lora_config': {'r': 32, 'lora_alpha': 64, 'lora_dropout': 0.05, 'bias': 'none', 'task_type': 'CAUSAL_LM'}, 'prompter': {'_target_': 'llmbc.translator.prompter.qwen2_prompter.Qwen2Prompter', 'use_joint_mlp_projector': True}, 'hydra': {'job': {'override_dirname': 'Qwen/Qwen2-0.5B-Instruct'}, 'run': {'dir': 'data/outputs/2026.02.27/07.17.38_Qwen/Qwen2-0.5B-Instruct'}}}}
14
+ 2026-02-27 07:17:58,904 INFO MainThread:1783449 [wandb_init.py:init():669] starting backend
15
+ 2026-02-27 07:17:58,904 INFO MainThread:1783449 [wandb_init.py:init():673] sending inform_init request
16
+ 2026-02-27 07:17:58,905 INFO MainThread:1783449 [backend.py:_multiprocessing_setup():104] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
17
+ 2026-02-27 07:17:58,906 INFO MainThread:1783449 [wandb_init.py:init():686] backend started and connected
18
+ 2026-02-27 07:17:58,915 INFO MainThread:1783449 [wandb_init.py:init():781] updated telemetry
19
+ 2026-02-27 07:17:58,969 INFO MainThread:1783449 [wandb_init.py:init():814] communicating run to backend with 90.0 second timeout
20
+ 2026-02-27 07:17:59,830 INFO MainThread:1783449 [wandb_init.py:init():867] starting run threads in backend
21
+ 2026-02-27 07:18:00,377 INFO MainThread:1783449 [wandb_run.py:_console_start():2451] atexit reg
22
+ 2026-02-27 07:18:00,377 INFO MainThread:1783449 [wandb_run.py:_redirect():2299] redirect: wrap_raw
23
+ 2026-02-27 07:18:00,377 INFO MainThread:1783449 [wandb_run.py:_redirect():2364] Wrapping output streams.
24
+ 2026-02-27 07:18:00,377 INFO MainThread:1783449 [wandb_run.py:_redirect():2389] Redirects installed.
25
+ 2026-02-27 07:18:00,380 INFO MainThread:1783449 [wandb_init.py:init():911] run started, returning control to user process
26
+ 2026-02-27 07:18:00,381 INFO MainThread:1783449 [wandb_run.py:_config_callback():1389] config_cb None None {'output_dir': '/work/u1131674/LLM-BC/data/outputs/2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2'}
27
+ 2026-02-28 02:32:59,956 WARNING MsgRouterThr:1783449 [router.py:message_loop():75] message_loop has been closed
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/run-20260227_071758-318q5q4f/run-318q5q4f.wandb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eae7533c26903fee92ee7969e6dfc6f8247b0f8d44d313e5cb65f7fc23c5d9df
3
+ size 218773585
2026.02.27/07.17.38_train_llmbc_lowdim_box-close-v2/wandb/wandb-resume.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"run_id": "318q5q4f"}