Upload folder using huggingface_hub
Browse files- .gitattributes +2 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/.hydra/config.yaml +163 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/.hydra/hydra.yaml +155 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/.hydra/overrides.yaml +1 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0275-test_success_rate=0.920.ckpt +3 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0330-test_success_rate=0.920.ckpt +3 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0540-test_success_rate=0.900.ckpt +3 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0575-test_success_rate=0.900.ckpt +3 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0800-test_success_rate=0.920.ckpt +3 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/checkpoints/latest.ckpt +3 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/logs.json.txt +3 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/train.log +8 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/debug-internal.log +19 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/debug.log +27 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/files/config.yaml +271 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/files/output.log +3 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/files/wandb-metadata.json +107 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/files/wandb-summary.json +1 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/logs/debug-core.log +14 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/logs/debug-internal.log +19 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/logs/debug.log +27 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/run-4cc5hfsj.wandb +3 -0
- 2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/wandb-resume.json +1 -0
.gitattributes
CHANGED
|
@@ -114,3 +114,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 114 |
2026.02.24/23.11.45_train_llm_lowdim_box-close-v2/wandb/run-20260224_231158-zzc3dyq3/run-zzc3dyq3.wandb filter=lfs diff=lfs merge=lfs -text
|
| 115 |
2026.02.24/23.25.46_train_llmbc_lowdim_box-close-v2/logs.json.txt 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
|
|
|
|
|
|
|
|
|
| 114 |
2026.02.24/23.11.45_train_llm_lowdim_box-close-v2/wandb/run-20260224_231158-zzc3dyq3/run-zzc3dyq3.wandb filter=lfs diff=lfs merge=lfs -text
|
| 115 |
2026.02.24/23.25.46_train_llmbc_lowdim_box-close-v2/logs.json.txt 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
|
2026.02.25/02.36.16_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: 44
|
| 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.25/02.36.16_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=44
|
| 116 |
+
job:
|
| 117 |
+
name: train
|
| 118 |
+
chdir: null
|
| 119 |
+
override_dirname: training.seed=44
|
| 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.25/02.36.16_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.25/02.36.16_train_llmbc_lowdim_box-close-v2/.hydra/overrides.yaml
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
- training.seed=44
|
2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0275-test_success_rate=0.920.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:575049231fae0225ad34ad7affc42bfd4a8734e2fb1fe9b303de99f91028711b
|
| 3 |
+
size 864520
|
2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0330-test_success_rate=0.920.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:60ea27e71b33e818cd4aec878bd3fda6c50b0e79aa861176b61b574184e84e36
|
| 3 |
+
size 864520
|
2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0540-test_success_rate=0.900.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:045f16c7d373f7d35001a14e1a6d2470a7bf7d75f61ea5d3f4cdd37978d20829
|
| 3 |
+
size 864520
|
2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0575-test_success_rate=0.900.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7d522e87b27cd4e1f9c50205eb56833bfba9fd87b97bdf720584393ae8781be6
|
| 3 |
+
size 864520
|
2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/checkpoints/epoch=0800-test_success_rate=0.920.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b1ca1c80f6fd5e76bcab0d63d49862071b04fe3f5fbecd263e5815c670fce064
|
| 3 |
+
size 864520
|
2026.02.25/02.36.16_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:5104c2c83b44a8ce8818bd25cdd0cc352f59ff5ad15ea3bd19e275e780dde7bb
|
| 3 |
+
size 864520
|
2026.02.25/02.36.16_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:249211c9accb179c2875f87fc3ee4605eed6e4a8287a26b903d10ad66340c427
|
| 3 |
+
size 51663812
|
2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/train.log
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[2026-02-25 02:36:19,344][numexpr.utils][INFO] - Note: detected 112 virtual cores but NumExpr set to maximum of 64, check "NUMEXPR_MAX_THREADS" environment variable.
|
| 2 |
+
[2026-02-25 02:36:19,344][numexpr.utils][INFO] - Note: NumExpr detected 112 cores but "NUMEXPR_MAX_THREADS" not set, so enforcing safe limit of 16.
|
| 3 |
+
[2026-02-25 02:36:19,344][numexpr.utils][INFO] - NumExpr defaulting to 16 threads.
|
| 4 |
+
[2026-02-25 02:36:26,289][datasets][INFO] - PyTorch version 2.2.2 available.
|
| 5 |
+
[2026-02-25 02:36:26,290][datasets][INFO] - TensorFlow version 2.15.1 available.
|
| 6 |
+
[2026-02-25 02:36:26,291][datasets][INFO] - JAX version 0.4.30 available.
|
| 7 |
+
[2026-02-25 02:36:36,807][absl][INFO] - MUJOCO_GL=osmesa, attempting to import specified OpenGL backend.
|
| 8 |
+
[2026-02-25 02:36:36,818][absl][INFO] - MuJoCo library version is: 2.3.7
|
2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/debug-internal.log
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{"time":"2026-02-25T02:36:37.952235833+08:00","level":"INFO","msg":"using version","core version":"0.18.6"}
|
| 2 |
+
{"time":"2026-02-25T02:36:37.952245446+08:00","level":"INFO","msg":"created symlink","path":"/work/u1131674/LLM-BC/data/outputs/2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/logs/debug-core.log"}
|
| 3 |
+
{"time":"2026-02-25T02:36:38.068144621+08:00","level":"INFO","msg":"created new stream","id":"4cc5hfsj"}
|
| 4 |
+
{"time":"2026-02-25T02:36:38.068174212+08:00","level":"INFO","msg":"stream: started","id":"4cc5hfsj"}
|
| 5 |
+
{"time":"2026-02-25T02:36:38.068196478+08:00","level":"INFO","msg":"sender: started","stream_id":"4cc5hfsj"}
|
| 6 |
+
{"time":"2026-02-25T02:36:38.06818769+08:00","level":"INFO","msg":"writer: Do: started","stream_id":{"value":"4cc5hfsj"}}
|
| 7 |
+
{"time":"2026-02-25T02:36:38.068194158+08:00","level":"INFO","msg":"handler: started","stream_id":{"value":"4cc5hfsj"}}
|
| 8 |
+
{"time":"2026-02-25T02:36:38.940202457+08:00","level":"INFO","msg":"Starting system monitor"}
|
| 9 |
+
{"time":"2026-02-25T10:08:09.673545635+08:00","level":"INFO","msg":"api: retrying error","error":"Post \"https://api.wandb.ai/graphql\": context deadline exceeded (Client.Timeout exceeded while awaiting headers)"}
|
| 10 |
+
{"time":"2026-02-25T10:08:42.083258362+08:00","level":"INFO","msg":"api: retrying error","error":"Post \"https://api.wandb.ai/graphql\": context deadline exceeded (Client.Timeout exceeded while awaiting headers)"}
|
| 11 |
+
{"time":"2026-02-25T10:09:16.126605311+08:00","level":"INFO","msg":"api: retrying error","error":"Post \"https://api.wandb.ai/graphql\": context deadline exceeded (Client.Timeout exceeded while awaiting headers)"}
|
| 12 |
+
{"time":"2026-02-25T21:33:34.325540207+08:00","level":"INFO","msg":"stream: closing","id":"4cc5hfsj"}
|
| 13 |
+
{"time":"2026-02-25T21:33:34.325567921+08:00","level":"INFO","msg":"Stopping system monitor"}
|
| 14 |
+
{"time":"2026-02-25T21:33:34.326076063+08:00","level":"INFO","msg":"Stopped system monitor"}
|
| 15 |
+
{"time":"2026-02-25T21:33:36.648063951+08:00","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
|
| 16 |
+
{"time":"2026-02-25T21:33:37.182579356+08:00","level":"INFO","msg":"handler: closed","stream_id":{"value":"4cc5hfsj"}}
|
| 17 |
+
{"time":"2026-02-25T21:33:37.182622781+08:00","level":"INFO","msg":"writer: Close: closed","stream_id":{"value":"4cc5hfsj"}}
|
| 18 |
+
{"time":"2026-02-25T21:33:37.182632816+08:00","level":"INFO","msg":"sender: closed","stream_id":"4cc5hfsj"}
|
| 19 |
+
{"time":"2026-02-25T21:33:37.182673966+08:00","level":"INFO","msg":"stream: closed","id":"4cc5hfsj"}
|
2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/debug.log
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_setup.py:_flush():79] Current SDK version is 0.18.6
|
| 2 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_setup.py:_flush():79] Configure stats pid to 4059787
|
| 3 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_setup.py:_flush():79] Loading settings from /home/u1131674/.config/wandb/settings
|
| 4 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_setup.py:_flush():79] Loading settings from /work/u1131674/LLM-BC/wandb/settings
|
| 5 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_setup.py:_flush():79] Loading settings from environment variables: {}
|
| 6 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_setup.py:_flush():79] Applying setup settings: {'mode': 'online', '_disable_service': None}
|
| 7 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [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-25 02:36:37,947 INFO MainThread:4059787 [wandb_setup.py:_flush():79] Applying login settings: {}
|
| 9 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_init.py:_log_setup():533] Logging user logs to /work/u1131674/LLM-BC/data/outputs/2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/logs/debug.log
|
| 10 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_init.py:_log_setup():534] Logging internal logs to /work/u1131674/LLM-BC/data/outputs/2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/logs/debug-internal.log
|
| 11 |
+
2026-02-25 02:36:37,948 INFO MainThread:4059787 [wandb_init.py:init():619] calling init triggers
|
| 12 |
+
2026-02-25 02:36:37,948 INFO MainThread:4059787 [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.25/02.36.16_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': 44, '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.25-02.36.16_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.25/02.36.16_train_llmbc_lowdim_box-close-v2', 'wandb_name_base': '2026.02.25-02.36.16_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.25/02.36.16_Qwen/Qwen2-0.5B-Instruct'}}}}
|
| 14 |
+
2026-02-25 02:36:37,948 INFO MainThread:4059787 [wandb_init.py:init():669] starting backend
|
| 15 |
+
2026-02-25 02:36:37,948 INFO MainThread:4059787 [wandb_init.py:init():673] sending inform_init request
|
| 16 |
+
2026-02-25 02:36:37,949 INFO MainThread:4059787 [backend.py:_multiprocessing_setup():104] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
|
| 17 |
+
2026-02-25 02:36:37,949 INFO MainThread:4059787 [wandb_init.py:init():686] backend started and connected
|
| 18 |
+
2026-02-25 02:36:37,958 INFO MainThread:4059787 [wandb_init.py:init():781] updated telemetry
|
| 19 |
+
2026-02-25 02:36:38,007 INFO MainThread:4059787 [wandb_init.py:init():814] communicating run to backend with 90.0 second timeout
|
| 20 |
+
2026-02-25 02:36:38,935 INFO MainThread:4059787 [wandb_init.py:init():867] starting run threads in backend
|
| 21 |
+
2026-02-25 02:36:39,515 INFO MainThread:4059787 [wandb_run.py:_console_start():2451] atexit reg
|
| 22 |
+
2026-02-25 02:36:39,516 INFO MainThread:4059787 [wandb_run.py:_redirect():2299] redirect: wrap_raw
|
| 23 |
+
2026-02-25 02:36:39,516 INFO MainThread:4059787 [wandb_run.py:_redirect():2364] Wrapping output streams.
|
| 24 |
+
2026-02-25 02:36:39,516 INFO MainThread:4059787 [wandb_run.py:_redirect():2389] Redirects installed.
|
| 25 |
+
2026-02-25 02:36:39,519 INFO MainThread:4059787 [wandb_init.py:init():911] run started, returning control to user process
|
| 26 |
+
2026-02-25 02:36:39,519 INFO MainThread:4059787 [wandb_run.py:_config_callback():1389] config_cb None None {'output_dir': '/work/u1131674/LLM-BC/data/outputs/2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2'}
|
| 27 |
+
2026-02-25 21:33:34,325 WARNING MsgRouterThr:4059787 [router.py:message_loop():75] message_loop has been closed
|
2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/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.25/02.36.16_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.25-02.36.16_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.25/02.36.16_train_llmbc_lowdim_box-close-v2
|
| 130 |
+
wandb_name_base: 2026.02.25-02.36.16_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.25/02.36.16_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.25/02.36.16_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: 44
|
| 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.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/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.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/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-24T18:36:37.949969Z",
|
| 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=44"
|
| 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": "f11e83e6ed4c9801ff7539f6d1f1f0600ac57e56"
|
| 17 |
+
},
|
| 18 |
+
"email": "chris920325@gmail.com",
|
| 19 |
+
"root": "/work/u1131674/LLM-BC/data/outputs/2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2",
|
| 20 |
+
"host": "hgpn02",
|
| 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": "274651598848"
|
| 32 |
+
}
|
| 33 |
+
},
|
| 34 |
+
"memory": {
|
| 35 |
+
"total": "2163622846464"
|
| 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:0x00000000000000000000FE000001",
|
| 53 |
+
"cpu_bind_list": "0x00000000000000000000FE000001",
|
| 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": "1772130974",
|
| 65 |
+
"job_gid": "106773",
|
| 66 |
+
"job_group": "MST114558",
|
| 67 |
+
"job_id": "124193",
|
| 68 |
+
"job_name": "python3",
|
| 69 |
+
"job_nodelist": "hgpn02",
|
| 70 |
+
"job_num_nodes": "1",
|
| 71 |
+
"job_partition": "normal",
|
| 72 |
+
"job_qos": "normal",
|
| 73 |
+
"job_start_time": "1771958174",
|
| 74 |
+
"job_uid": "41408",
|
| 75 |
+
"job_user": "u1131674",
|
| 76 |
+
"jobid": "124193",
|
| 77 |
+
"launch_node_ipaddr": "172.21.101.1",
|
| 78 |
+
"localid": "0",
|
| 79 |
+
"mem_per_node": "204800",
|
| 80 |
+
"nnodes": "1",
|
| 81 |
+
"nodeid": "0",
|
| 82 |
+
"nodelist": "hgpn02",
|
| 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": "43349",
|
| 89 |
+
"step_gpus": "0",
|
| 90 |
+
"step_id": "0",
|
| 91 |
+
"step_launcher_port": "43349",
|
| 92 |
+
"step_nodelist": "hgpn02",
|
| 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": "4059787",
|
| 100 |
+
"tasks_per_node": "1",
|
| 101 |
+
"topology_addr": "ibsw1.hgpn02",
|
| 102 |
+
"topology_addr_pattern": "switch.node",
|
| 103 |
+
"tres_per_task": "cpu:8",
|
| 104 |
+
"umask": "0022"
|
| 105 |
+
},
|
| 106 |
+
"cudaVersion": "12.4"
|
| 107 |
+
}
|
2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/files/wandb-summary.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"train_loss_bc":7.778144208714366e-05,"test/cumulative_reward":13.970284531473615,"val_loss":0.00033129341318272054,"global_step":237236,"epoch":1000,"train/cumulative_reward":16.655863503460573,"train/success_rate":1,"train_action_mse_error":0.00010216646478511393,"test/mean_score":7.784985426388307,"grad_norm":0.008994531817734241,"_timestamp":1.7720264120929835e+09,"_runtime":68216.375586972,"_step":237236,"test/success_rate":0.88,"train_loss_llm":0.08394504338502884,"train_loss":0.0001783600668106742,"_wandb":{"runtime":68216},"lr":2.8077998259767867e-11,"train/mean_score":8.70553322948669}
|
2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/logs/debug-core.log
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{"time":"2026-02-25T02:36:37.194105041+08:00","level":"INFO","msg":"started logging, with flags","port-filename":"/tmp/tmphhck99_t/port-4059787.txt","pid":4059787,"debug":false,"disable-analytics":false}
|
| 2 |
+
{"time":"2026-02-25T02:36:37.194126812+08:00","level":"INFO","msg":"FeatureState","shutdownOnParentExitEnabled":false}
|
| 3 |
+
{"time":"2026-02-25T02:36:37.194587897+08:00","level":"INFO","msg":"Will exit if parent process dies.","ppid":4059787}
|
| 4 |
+
{"time":"2026-02-25T02:36:37.194581349+08:00","level":"INFO","msg":"server is running","addr":{"IP":"127.0.0.1","Port":42467,"Zone":""}}
|
| 5 |
+
{"time":"2026-02-25T02:36:37.387161574+08:00","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"127.0.0.1:60552"}
|
| 6 |
+
{"time":"2026-02-25T02:36:37.950490669+08:00","level":"INFO","msg":"handleInformInit: received","streamId":"4cc5hfsj","id":"127.0.0.1:60552"}
|
| 7 |
+
{"time":"2026-02-25T02:36:38.068180116+08:00","level":"INFO","msg":"handleInformInit: stream started","streamId":"4cc5hfsj","id":"127.0.0.1:60552"}
|
| 8 |
+
{"time":"2026-02-25T21:33:34.325476276+08:00","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"127.0.0.1:60552"}
|
| 9 |
+
{"time":"2026-02-25T21:33:34.325553848+08:00","level":"INFO","msg":"server is shutting down"}
|
| 10 |
+
{"time":"2026-02-25T21:33:34.325544398+08:00","level":"INFO","msg":"connection: Close: initiating connection closure","id":"127.0.0.1:60552"}
|
| 11 |
+
{"time":"2026-02-25T21:33:34.325635718+08:00","level":"INFO","msg":"connection: Close: connection successfully closed","id":"127.0.0.1:60552"}
|
| 12 |
+
{"time":"2026-02-25T21:33:37.182880749+08:00","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"127.0.0.1:60552"}
|
| 13 |
+
{"time":"2026-02-25T21:33:37.182896958+08:00","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"127.0.0.1:60552"}
|
| 14 |
+
{"time":"2026-02-25T21:33:37.182908024+08:00","level":"INFO","msg":"server is closed"}
|
2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/logs/debug-internal.log
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{"time":"2026-02-25T02:36:37.952235833+08:00","level":"INFO","msg":"using version","core version":"0.18.6"}
|
| 2 |
+
{"time":"2026-02-25T02:36:37.952245446+08:00","level":"INFO","msg":"created symlink","path":"/work/u1131674/LLM-BC/data/outputs/2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/logs/debug-core.log"}
|
| 3 |
+
{"time":"2026-02-25T02:36:38.068144621+08:00","level":"INFO","msg":"created new stream","id":"4cc5hfsj"}
|
| 4 |
+
{"time":"2026-02-25T02:36:38.068174212+08:00","level":"INFO","msg":"stream: started","id":"4cc5hfsj"}
|
| 5 |
+
{"time":"2026-02-25T02:36:38.068196478+08:00","level":"INFO","msg":"sender: started","stream_id":"4cc5hfsj"}
|
| 6 |
+
{"time":"2026-02-25T02:36:38.06818769+08:00","level":"INFO","msg":"writer: Do: started","stream_id":{"value":"4cc5hfsj"}}
|
| 7 |
+
{"time":"2026-02-25T02:36:38.068194158+08:00","level":"INFO","msg":"handler: started","stream_id":{"value":"4cc5hfsj"}}
|
| 8 |
+
{"time":"2026-02-25T02:36:38.940202457+08:00","level":"INFO","msg":"Starting system monitor"}
|
| 9 |
+
{"time":"2026-02-25T10:08:09.673545635+08:00","level":"INFO","msg":"api: retrying error","error":"Post \"https://api.wandb.ai/graphql\": context deadline exceeded (Client.Timeout exceeded while awaiting headers)"}
|
| 10 |
+
{"time":"2026-02-25T10:08:42.083258362+08:00","level":"INFO","msg":"api: retrying error","error":"Post \"https://api.wandb.ai/graphql\": context deadline exceeded (Client.Timeout exceeded while awaiting headers)"}
|
| 11 |
+
{"time":"2026-02-25T10:09:16.126605311+08:00","level":"INFO","msg":"api: retrying error","error":"Post \"https://api.wandb.ai/graphql\": context deadline exceeded (Client.Timeout exceeded while awaiting headers)"}
|
| 12 |
+
{"time":"2026-02-25T21:33:34.325540207+08:00","level":"INFO","msg":"stream: closing","id":"4cc5hfsj"}
|
| 13 |
+
{"time":"2026-02-25T21:33:34.325567921+08:00","level":"INFO","msg":"Stopping system monitor"}
|
| 14 |
+
{"time":"2026-02-25T21:33:34.326076063+08:00","level":"INFO","msg":"Stopped system monitor"}
|
| 15 |
+
{"time":"2026-02-25T21:33:36.648063951+08:00","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
|
| 16 |
+
{"time":"2026-02-25T21:33:37.182579356+08:00","level":"INFO","msg":"handler: closed","stream_id":{"value":"4cc5hfsj"}}
|
| 17 |
+
{"time":"2026-02-25T21:33:37.182622781+08:00","level":"INFO","msg":"writer: Close: closed","stream_id":{"value":"4cc5hfsj"}}
|
| 18 |
+
{"time":"2026-02-25T21:33:37.182632816+08:00","level":"INFO","msg":"sender: closed","stream_id":"4cc5hfsj"}
|
| 19 |
+
{"time":"2026-02-25T21:33:37.182673966+08:00","level":"INFO","msg":"stream: closed","id":"4cc5hfsj"}
|
2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/logs/debug.log
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_setup.py:_flush():79] Current SDK version is 0.18.6
|
| 2 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_setup.py:_flush():79] Configure stats pid to 4059787
|
| 3 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_setup.py:_flush():79] Loading settings from /home/u1131674/.config/wandb/settings
|
| 4 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_setup.py:_flush():79] Loading settings from /work/u1131674/LLM-BC/wandb/settings
|
| 5 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_setup.py:_flush():79] Loading settings from environment variables: {}
|
| 6 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_setup.py:_flush():79] Applying setup settings: {'mode': 'online', '_disable_service': None}
|
| 7 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [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-25 02:36:37,947 INFO MainThread:4059787 [wandb_setup.py:_flush():79] Applying login settings: {}
|
| 9 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_init.py:_log_setup():533] Logging user logs to /work/u1131674/LLM-BC/data/outputs/2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/logs/debug.log
|
| 10 |
+
2026-02-25 02:36:37,947 INFO MainThread:4059787 [wandb_init.py:_log_setup():534] Logging internal logs to /work/u1131674/LLM-BC/data/outputs/2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/logs/debug-internal.log
|
| 11 |
+
2026-02-25 02:36:37,948 INFO MainThread:4059787 [wandb_init.py:init():619] calling init triggers
|
| 12 |
+
2026-02-25 02:36:37,948 INFO MainThread:4059787 [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.25/02.36.16_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': 44, '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.25-02.36.16_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.25/02.36.16_train_llmbc_lowdim_box-close-v2', 'wandb_name_base': '2026.02.25-02.36.16_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.25/02.36.16_Qwen/Qwen2-0.5B-Instruct'}}}}
|
| 14 |
+
2026-02-25 02:36:37,948 INFO MainThread:4059787 [wandb_init.py:init():669] starting backend
|
| 15 |
+
2026-02-25 02:36:37,948 INFO MainThread:4059787 [wandb_init.py:init():673] sending inform_init request
|
| 16 |
+
2026-02-25 02:36:37,949 INFO MainThread:4059787 [backend.py:_multiprocessing_setup():104] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
|
| 17 |
+
2026-02-25 02:36:37,949 INFO MainThread:4059787 [wandb_init.py:init():686] backend started and connected
|
| 18 |
+
2026-02-25 02:36:37,958 INFO MainThread:4059787 [wandb_init.py:init():781] updated telemetry
|
| 19 |
+
2026-02-25 02:36:38,007 INFO MainThread:4059787 [wandb_init.py:init():814] communicating run to backend with 90.0 second timeout
|
| 20 |
+
2026-02-25 02:36:38,935 INFO MainThread:4059787 [wandb_init.py:init():867] starting run threads in backend
|
| 21 |
+
2026-02-25 02:36:39,515 INFO MainThread:4059787 [wandb_run.py:_console_start():2451] atexit reg
|
| 22 |
+
2026-02-25 02:36:39,516 INFO MainThread:4059787 [wandb_run.py:_redirect():2299] redirect: wrap_raw
|
| 23 |
+
2026-02-25 02:36:39,516 INFO MainThread:4059787 [wandb_run.py:_redirect():2364] Wrapping output streams.
|
| 24 |
+
2026-02-25 02:36:39,516 INFO MainThread:4059787 [wandb_run.py:_redirect():2389] Redirects installed.
|
| 25 |
+
2026-02-25 02:36:39,519 INFO MainThread:4059787 [wandb_init.py:init():911] run started, returning control to user process
|
| 26 |
+
2026-02-25 02:36:39,519 INFO MainThread:4059787 [wandb_run.py:_config_callback():1389] config_cb None None {'output_dir': '/work/u1131674/LLM-BC/data/outputs/2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2'}
|
| 27 |
+
2026-02-25 21:33:34,325 WARNING MsgRouterThr:4059787 [router.py:message_loop():75] message_loop has been closed
|
2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/run-20260225_023637-4cc5hfsj/run-4cc5hfsj.wandb
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:01dcf4134a815cc9b92c037bdbd1bb4f0fc308d6ecac00a7971df30d6db7a645
|
| 3 |
+
size 215620869
|
2026.02.25/02.36.16_train_llmbc_lowdim_box-close-v2/wandb/wandb-resume.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"run_id": "4cc5hfsj"}
|