Upload folder using huggingface_hub
Browse files- .gitattributes +1 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/.hydra/config.yaml +112 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/.hydra/hydra.yaml +155 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/.hydra/overrides.yaml +1 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/checkpoints/epoch=9750-val_loss=0.023.ckpt +3 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/checkpoints/epoch=9800-val_loss=0.023.ckpt +3 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/checkpoints/epoch=9850-val_loss=0.023.ckpt +3 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/checkpoints/epoch=9900-val_loss=0.023.ckpt +3 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/checkpoints/epoch=9950-val_loss=0.023.ckpt +3 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/checkpoints/latest.ckpt +3 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/logs.json.txt +0 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/train_rbc_reward_workspace.log +0 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/debug-internal.log +16 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/debug.log +27 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/files/config.yaml +170 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/files/output.log +1 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/files/wandb-metadata.json +49 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/files/wandb-summary.json +1 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/logs/debug-core.log +14 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/logs/debug-internal.log +16 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/logs/debug.log +27 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/run-f5kmrluf.wandb +3 -0
- 2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/wandb-resume.json +1 -0
.gitattributes
CHANGED
|
@@ -106,3 +106,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 106 |
2025.10.15/14.25.50_train_rbc_reward_model_box-close-v2/wandb/run-20251015_142552-k1pb90ja/run-k1pb90ja.wandb filter=lfs diff=lfs merge=lfs -text
|
| 107 |
2026.01.05/13.53.06_train_rbc_reward_model_box-close-v2/wandb/run-20260105_135308-lwrt8yen/run-lwrt8yen.wandb filter=lfs diff=lfs merge=lfs -text
|
| 108 |
2026.01.05/13.53.26_train_rbc_reward_model_box-close-v2/wandb/run-20260105_135328-t2o4vy3f/run-t2o4vy3f.wandb filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 106 |
2025.10.15/14.25.50_train_rbc_reward_model_box-close-v2/wandb/run-20251015_142552-k1pb90ja/run-k1pb90ja.wandb filter=lfs diff=lfs merge=lfs -text
|
| 107 |
2026.01.05/13.53.06_train_rbc_reward_model_box-close-v2/wandb/run-20260105_135308-lwrt8yen/run-lwrt8yen.wandb filter=lfs diff=lfs merge=lfs -text
|
| 108 |
2026.01.05/13.53.26_train_rbc_reward_model_box-close-v2/wandb/run-20260105_135328-t2o4vy3f/run-t2o4vy3f.wandb filter=lfs diff=lfs merge=lfs -text
|
| 109 |
+
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/run-f5kmrluf.wandb filter=lfs diff=lfs merge=lfs -text
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/.hydra/config.yaml
ADDED
|
@@ -0,0 +1,112 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: train_rbc_reward_model
|
| 2 |
+
_target_: llmbc.workspace.train_rbc_reward_workspace.TrainRBCRewardWorkspace
|
| 3 |
+
obs_dim: ${task.obs_dim}
|
| 4 |
+
action_dim: ${task.action_dim}
|
| 5 |
+
task_name: ${task.name}
|
| 6 |
+
exp_name: default
|
| 7 |
+
horizon: 1
|
| 8 |
+
n_obs_steps: 1
|
| 9 |
+
n_action_steps: 1
|
| 10 |
+
n_latency_steps: 0
|
| 11 |
+
past_action_visible: false
|
| 12 |
+
reward_model:
|
| 13 |
+
_target_: llmbc.policy.rbc_lowdim_policy.RBCLowdimRewardModel
|
| 14 |
+
input_size: ${eval:'${n_obs_steps}*${obs_dim}+${n_action_steps}*${action_dim}'}
|
| 15 |
+
hidden_size:
|
| 16 |
+
- 128
|
| 17 |
+
obs_dim: ${obs_dim}
|
| 18 |
+
action_dim: ${action_dim}
|
| 19 |
+
n_obs_steps: ${n_obs_steps}
|
| 20 |
+
n_action_steps: ${n_action_steps}
|
| 21 |
+
dataloader:
|
| 22 |
+
batch_size: 256
|
| 23 |
+
num_workers: 0
|
| 24 |
+
shuffle: true
|
| 25 |
+
pin_memory: false
|
| 26 |
+
persistent_workers: false
|
| 27 |
+
val_dataloader:
|
| 28 |
+
batch_size: 256
|
| 29 |
+
num_workers: 0
|
| 30 |
+
shuffle: true
|
| 31 |
+
pin_memory: false
|
| 32 |
+
persistent_workers: false
|
| 33 |
+
optimizer:
|
| 34 |
+
_target_: torch.optim.Adam
|
| 35 |
+
lr: 5.0e-06
|
| 36 |
+
betas:
|
| 37 |
+
- 0.95
|
| 38 |
+
- 0.999
|
| 39 |
+
eps: 1.0e-08
|
| 40 |
+
weight_decay: 1.0e-06
|
| 41 |
+
training:
|
| 42 |
+
device: cuda
|
| 43 |
+
seed: 42
|
| 44 |
+
debug: false
|
| 45 |
+
resume: true
|
| 46 |
+
lr_scheduler: cosine
|
| 47 |
+
lr_warmup_steps: 0
|
| 48 |
+
num_epochs: 10000
|
| 49 |
+
gradient_accumulate_every: 1
|
| 50 |
+
grad_norm_clip: 0.5
|
| 51 |
+
checkpoint_every: 50
|
| 52 |
+
val_every: 1
|
| 53 |
+
sample_every: 50
|
| 54 |
+
sample_max_batch: 1000
|
| 55 |
+
max_train_steps: null
|
| 56 |
+
max_val_steps: null
|
| 57 |
+
tqdm_interval_sec: 1.0
|
| 58 |
+
logging:
|
| 59 |
+
project: rbc_module_training
|
| 60 |
+
resume: true
|
| 61 |
+
mode: online
|
| 62 |
+
name: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
|
| 63 |
+
tags:
|
| 64 |
+
- ${name}
|
| 65 |
+
- ${task_name}
|
| 66 |
+
- ${exp_name}
|
| 67 |
+
id: null
|
| 68 |
+
group: null
|
| 69 |
+
checkpoint:
|
| 70 |
+
topk:
|
| 71 |
+
monitor_key: val_loss
|
| 72 |
+
mode: min
|
| 73 |
+
k: 5
|
| 74 |
+
format_str: epoch={epoch:04d}-val_loss={val_loss:.3f}.ckpt
|
| 75 |
+
save_last_ckpt: true
|
| 76 |
+
save_last_snapshot: false
|
| 77 |
+
multi_run:
|
| 78 |
+
run_dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
|
| 79 |
+
wandb_name_base: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
|
| 80 |
+
task:
|
| 81 |
+
name: box-close-v2
|
| 82 |
+
obs_dim: 9
|
| 83 |
+
action_dim: 4
|
| 84 |
+
env_runner:
|
| 85 |
+
_target_: llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner
|
| 86 |
+
env_name: llf-metaworld-box-close-v2
|
| 87 |
+
n_train: 10
|
| 88 |
+
n_test: 50
|
| 89 |
+
n_envs: 10
|
| 90 |
+
max_steps: 30
|
| 91 |
+
n_obs_steps: ${n_obs_steps}
|
| 92 |
+
n_action_steps: ${n_action_steps}
|
| 93 |
+
instruction_type: b
|
| 94 |
+
feedback_type:
|
| 95 |
+
- hp
|
| 96 |
+
- hn
|
| 97 |
+
- fp
|
| 98 |
+
visual: false
|
| 99 |
+
discount: 0.9
|
| 100 |
+
dataset:
|
| 101 |
+
_target_: llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset
|
| 102 |
+
data_path: datasets/box-close-v2-general-reward-split0.02.pt
|
| 103 |
+
data_path2: datasets/box-close-v2-reward-split0.02.pt
|
| 104 |
+
horizon: ${horizon}
|
| 105 |
+
pad_before: ${eval:'${n_obs_steps}-1'}
|
| 106 |
+
pad_after: ${eval:'${n_action_steps}-1'}
|
| 107 |
+
obs_eef_target: true
|
| 108 |
+
use_manual_normalizer: false
|
| 109 |
+
val_ratio: 0.1
|
| 110 |
+
dummy_normalizer: false
|
| 111 |
+
instructor:
|
| 112 |
+
_target_: llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor
|
2026.01.05/19.09.31_train_rbc_reward_model_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 |
+
job:
|
| 116 |
+
name: train_rbc_reward_workspace
|
| 117 |
+
chdir: null
|
| 118 |
+
override_dirname: ''
|
| 119 |
+
id: ???
|
| 120 |
+
num: ???
|
| 121 |
+
config_name: train_rbc_reward_workspace
|
| 122 |
+
env_set: {}
|
| 123 |
+
env_copy: []
|
| 124 |
+
config:
|
| 125 |
+
override_dirname:
|
| 126 |
+
kv_sep: '='
|
| 127 |
+
item_sep: ','
|
| 128 |
+
exclude_keys: []
|
| 129 |
+
runtime:
|
| 130 |
+
version: 1.2.0
|
| 131 |
+
version_base: '1.2'
|
| 132 |
+
cwd: /tmp2/chyang/workspace/LLM-BC
|
| 133 |
+
config_sources:
|
| 134 |
+
- path: hydra.conf
|
| 135 |
+
schema: pkg
|
| 136 |
+
provider: hydra
|
| 137 |
+
- path: /tmp2/chyang/workspace/LLM-BC/llmbc/config
|
| 138 |
+
schema: file
|
| 139 |
+
provider: main
|
| 140 |
+
- path: ''
|
| 141 |
+
schema: structured
|
| 142 |
+
provider: schema
|
| 143 |
+
output_dir: /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2
|
| 144 |
+
choices:
|
| 145 |
+
task: box-close-v2
|
| 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.01.05/19.09.31_train_rbc_reward_model_box-close-v2/.hydra/overrides.yaml
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
[]
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/checkpoints/epoch=9750-val_loss=0.023.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7b0ed303dbdbd98604d49518e7de3235dc5c32cc1de13a5edfaad2a8321c1698
|
| 3 |
+
size 47932
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/checkpoints/epoch=9800-val_loss=0.023.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:413fddfdb2fc8fc10174aad73e98fdd1d4822752f87caefd20d24de4ac3e2a72
|
| 3 |
+
size 47932
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/checkpoints/epoch=9850-val_loss=0.023.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6de62074914816052c1c27a82a13933de3ca708a2c9935f935dd7dde0dfce574
|
| 3 |
+
size 47932
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/checkpoints/epoch=9900-val_loss=0.023.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:af4ad3d15e3ac57947508a0cc0c1b19b65771109ce79ede56bfed8fe1cb9128f
|
| 3 |
+
size 47932
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/checkpoints/epoch=9950-val_loss=0.023.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a94dc9a2e2b5911f3f85ca37b1121808e5c74dfc3aff71981d672d8c6d733649
|
| 3 |
+
size 47932
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/checkpoints/latest.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a94dc9a2e2b5911f3f85ca37b1121808e5c74dfc3aff71981d672d8c6d733649
|
| 3 |
+
size 47932
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/logs.json.txt
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/train_rbc_reward_workspace.log
ADDED
|
File without changes
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/debug-internal.log
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{"time":"2026-01-05T19:09:33.469014101+08:00","level":"INFO","msg":"using version","core version":"0.18.6"}
|
| 2 |
+
{"time":"2026-01-05T19:09:33.469038682+08:00","level":"INFO","msg":"created symlink","path":"/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/logs/debug-core.log"}
|
| 3 |
+
{"time":"2026-01-05T19:09:33.578729199+08:00","level":"INFO","msg":"created new stream","id":"f5kmrluf"}
|
| 4 |
+
{"time":"2026-01-05T19:09:33.578780561+08:00","level":"INFO","msg":"stream: started","id":"f5kmrluf"}
|
| 5 |
+
{"time":"2026-01-05T19:09:33.578893262+08:00","level":"INFO","msg":"sender: started","stream_id":"f5kmrluf"}
|
| 6 |
+
{"time":"2026-01-05T19:09:33.578806052+08:00","level":"INFO","msg":"writer: Do: started","stream_id":{"value":"f5kmrluf"}}
|
| 7 |
+
{"time":"2026-01-05T19:09:33.578868514+08:00","level":"INFO","msg":"handler: started","stream_id":{"value":"f5kmrluf"}}
|
| 8 |
+
{"time":"2026-01-05T19:09:34.295619023+08:00","level":"INFO","msg":"Starting system monitor"}
|
| 9 |
+
{"time":"2026-01-05T19:15:21.576746822+08:00","level":"INFO","msg":"stream: closing","id":"f5kmrluf"}
|
| 10 |
+
{"time":"2026-01-05T19:15:21.576787059+08:00","level":"INFO","msg":"Stopping system monitor"}
|
| 11 |
+
{"time":"2026-01-05T19:15:21.577575522+08:00","level":"INFO","msg":"Stopped system monitor"}
|
| 12 |
+
{"time":"2026-01-05T19:15:22.630438342+08:00","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
|
| 13 |
+
{"time":"2026-01-05T19:15:22.920151712+08:00","level":"INFO","msg":"handler: closed","stream_id":{"value":"f5kmrluf"}}
|
| 14 |
+
{"time":"2026-01-05T19:15:22.920233247+08:00","level":"INFO","msg":"sender: closed","stream_id":"f5kmrluf"}
|
| 15 |
+
{"time":"2026-01-05T19:15:22.920232896+08:00","level":"INFO","msg":"writer: Close: closed","stream_id":{"value":"f5kmrluf"}}
|
| 16 |
+
{"time":"2026-01-05T19:15:22.920391499+08:00","level":"INFO","msg":"stream: closed","id":"f5kmrluf"}
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/debug.log
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Current SDK version is 0.18.6
|
| 2 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Configure stats pid to 101140
|
| 3 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Loading settings from /home/chyang/.config/wandb/settings
|
| 4 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Loading settings from /tmp2/chyang/workspace/LLM-BC/wandb/settings
|
| 5 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Loading settings from environment variables: {}
|
| 6 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Applying setup settings: {'mode': 'online', '_disable_service': None}
|
| 7 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Inferring run settings from compute environment: {'program_relpath': 'llmbc/workspace/train_rbc_reward_workspace.py', 'program_abspath': '/tmp2/chyang/workspace/LLM-BC/llmbc/workspace/train_rbc_reward_workspace.py', 'program': '/tmp2/chyang/workspace/LLM-BC/./llmbc/workspace/train_rbc_reward_workspace.py'}
|
| 8 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Applying login settings: {}
|
| 9 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_init.py:_log_setup():533] Logging user logs to /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/logs/debug.log
|
| 10 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_init.py:_log_setup():534] Logging internal logs to /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/logs/debug-internal.log
|
| 11 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_init.py:init():619] calling init triggers
|
| 12 |
+
2026-01-05 19:09:33,465 INFO MainThread:101140 [wandb_init.py:init():626] wandb.init called with sweep_config: {}
|
| 13 |
+
config: {'name': 'train_rbc_reward_model', '_target_': 'llmbc.workspace.train_rbc_reward_workspace.TrainRBCRewardWorkspace', 'obs_dim': 9, 'action_dim': 4, 'task_name': 'box-close-v2', 'exp_name': 'default', 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1, 'n_latency_steps': 0, 'past_action_visible': False, 'reward_model': {'_target_': 'llmbc.policy.rbc_lowdim_policy.RBCLowdimRewardModel', 'input_size': 13, 'hidden_size': [128], 'obs_dim': 9, 'action_dim': 4, 'n_obs_steps': 1, 'n_action_steps': 1}, 'dataloader': {'batch_size': 256, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'val_dataloader': {'batch_size': 256, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'optimizer': {'_target_': 'torch.optim.Adam', 'lr': 5e-06, 'betas': [0.95, 0.999], 'eps': 1e-08, 'weight_decay': 1e-06}, 'training': {'device': 'cuda', 'seed': 42, 'debug': False, 'resume': True, 'lr_scheduler': 'cosine', 'lr_warmup_steps': 0, 'num_epochs': 10000, 'gradient_accumulate_every': 1, 'grad_norm_clip': 0.5, 'checkpoint_every': 50, 'val_every': 1, 'sample_every': 50, 'sample_max_batch': 1000, 'max_train_steps': None, 'max_val_steps': None, 'tqdm_interval_sec': 1.0}, 'logging': {'project': 'rbc_module_training', 'resume': True, 'mode': 'online', 'name': '2026.01.05-19.09.32_train_rbc_reward_model_box-close-v2', 'tags': ['train_rbc_reward_model', 'box-close-v2', 'default'], 'id': None, 'group': None}, 'checkpoint': {'topk': {'monitor_key': 'val_loss', 'mode': 'min', 'k': 5, 'format_str': 'epoch={epoch:04d}-val_loss={val_loss:.3f}.ckpt'}, 'save_last_ckpt': True, 'save_last_snapshot': False}, 'multi_run': {'run_dir': 'data/outputs/2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2', 'wandb_name_base': '2026.01.05-19.09.32_train_rbc_reward_model_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-general-reward-split0.02.pt', 'data_path2': 'datasets/box-close-v2-reward-split0.02.pt', 'horizon': 1, 'pad_before': 0, 'pad_after': 0, 'obs_eef_target': True, 'use_manual_normalizer': False, 'val_ratio': 0.1, 'dummy_normalizer': False}, 'instructor': {'_target_': 'llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor'}}}
|
| 14 |
+
2026-01-05 19:09:33,465 INFO MainThread:101140 [wandb_init.py:init():669] starting backend
|
| 15 |
+
2026-01-05 19:09:33,465 INFO MainThread:101140 [wandb_init.py:init():673] sending inform_init request
|
| 16 |
+
2026-01-05 19:09:33,465 INFO MainThread:101140 [backend.py:_multiprocessing_setup():104] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
|
| 17 |
+
2026-01-05 19:09:33,466 INFO MainThread:101140 [wandb_init.py:init():686] backend started and connected
|
| 18 |
+
2026-01-05 19:09:33,469 INFO MainThread:101140 [wandb_init.py:init():781] updated telemetry
|
| 19 |
+
2026-01-05 19:09:33,491 INFO MainThread:101140 [wandb_init.py:init():814] communicating run to backend with 90.0 second timeout
|
| 20 |
+
2026-01-05 19:09:34,290 INFO MainThread:101140 [wandb_init.py:init():867] starting run threads in backend
|
| 21 |
+
2026-01-05 19:09:34,456 INFO MainThread:101140 [wandb_run.py:_console_start():2451] atexit reg
|
| 22 |
+
2026-01-05 19:09:34,456 INFO MainThread:101140 [wandb_run.py:_redirect():2299] redirect: wrap_raw
|
| 23 |
+
2026-01-05 19:09:34,457 INFO MainThread:101140 [wandb_run.py:_redirect():2364] Wrapping output streams.
|
| 24 |
+
2026-01-05 19:09:34,457 INFO MainThread:101140 [wandb_run.py:_redirect():2389] Redirects installed.
|
| 25 |
+
2026-01-05 19:09:34,458 INFO MainThread:101140 [wandb_init.py:init():911] run started, returning control to user process
|
| 26 |
+
2026-01-05 19:09:34,459 INFO MainThread:101140 [wandb_run.py:_config_callback():1389] config_cb None None {'output_dir': '/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2'}
|
| 27 |
+
2026-01-05 19:15:21,576 WARNING MsgRouterThr:101140 [router.py:message_loop():75] message_loop has been closed
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/files/config.yaml
ADDED
|
@@ -0,0 +1,170 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
_target_:
|
| 2 |
+
value: llmbc.workspace.train_rbc_reward_workspace.TrainRBCRewardWorkspace
|
| 3 |
+
_wandb:
|
| 4 |
+
value:
|
| 5 |
+
cli_version: 0.18.6
|
| 6 |
+
m: []
|
| 7 |
+
python_version: 3.9.21
|
| 8 |
+
t:
|
| 9 |
+
"1":
|
| 10 |
+
- 1
|
| 11 |
+
- 11
|
| 12 |
+
- 49
|
| 13 |
+
- 50
|
| 14 |
+
- 55
|
| 15 |
+
- 83
|
| 16 |
+
- 105
|
| 17 |
+
"2":
|
| 18 |
+
- 1
|
| 19 |
+
- 11
|
| 20 |
+
- 49
|
| 21 |
+
- 50
|
| 22 |
+
- 55
|
| 23 |
+
- 83
|
| 24 |
+
- 105
|
| 25 |
+
"3":
|
| 26 |
+
- 13
|
| 27 |
+
- 15
|
| 28 |
+
- 16
|
| 29 |
+
- 23
|
| 30 |
+
- 55
|
| 31 |
+
- 61
|
| 32 |
+
"4": 3.9.21
|
| 33 |
+
"5": 0.18.6
|
| 34 |
+
"6": 4.47.1
|
| 35 |
+
"8":
|
| 36 |
+
- 5
|
| 37 |
+
"12": 0.18.6
|
| 38 |
+
"13": linux-x86_64
|
| 39 |
+
action_dim:
|
| 40 |
+
value: 4
|
| 41 |
+
checkpoint:
|
| 42 |
+
value:
|
| 43 |
+
save_last_ckpt: true
|
| 44 |
+
save_last_snapshot: false
|
| 45 |
+
topk:
|
| 46 |
+
format_str: epoch={epoch:04d}-val_loss={val_loss:.3f}.ckpt
|
| 47 |
+
k: 5
|
| 48 |
+
mode: min
|
| 49 |
+
monitor_key: val_loss
|
| 50 |
+
dataloader:
|
| 51 |
+
value:
|
| 52 |
+
batch_size: 256
|
| 53 |
+
num_workers: 0
|
| 54 |
+
persistent_workers: false
|
| 55 |
+
pin_memory: false
|
| 56 |
+
shuffle: true
|
| 57 |
+
exp_name:
|
| 58 |
+
value: default
|
| 59 |
+
horizon:
|
| 60 |
+
value: 1
|
| 61 |
+
logging:
|
| 62 |
+
value:
|
| 63 |
+
group: null
|
| 64 |
+
id: null
|
| 65 |
+
mode: online
|
| 66 |
+
name: 2026.01.05-19.09.32_train_rbc_reward_model_box-close-v2
|
| 67 |
+
project: rbc_module_training
|
| 68 |
+
resume: true
|
| 69 |
+
tags:
|
| 70 |
+
- train_rbc_reward_model
|
| 71 |
+
- box-close-v2
|
| 72 |
+
- default
|
| 73 |
+
multi_run:
|
| 74 |
+
value:
|
| 75 |
+
run_dir: data/outputs/2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2
|
| 76 |
+
wandb_name_base: 2026.01.05-19.09.32_train_rbc_reward_model_box-close-v2
|
| 77 |
+
n_action_steps:
|
| 78 |
+
value: 1
|
| 79 |
+
n_latency_steps:
|
| 80 |
+
value: 0
|
| 81 |
+
n_obs_steps:
|
| 82 |
+
value: 1
|
| 83 |
+
name:
|
| 84 |
+
value: train_rbc_reward_model
|
| 85 |
+
obs_dim:
|
| 86 |
+
value: 9
|
| 87 |
+
optimizer:
|
| 88 |
+
value:
|
| 89 |
+
_target_: torch.optim.Adam
|
| 90 |
+
betas:
|
| 91 |
+
- 0.95
|
| 92 |
+
- 0.999
|
| 93 |
+
eps: 1e-08
|
| 94 |
+
lr: 5e-06
|
| 95 |
+
weight_decay: 1e-06
|
| 96 |
+
output_dir:
|
| 97 |
+
value: /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2
|
| 98 |
+
past_action_visible:
|
| 99 |
+
value: false
|
| 100 |
+
reward_model:
|
| 101 |
+
value:
|
| 102 |
+
_target_: llmbc.policy.rbc_lowdim_policy.RBCLowdimRewardModel
|
| 103 |
+
action_dim: 4
|
| 104 |
+
hidden_size:
|
| 105 |
+
- 128
|
| 106 |
+
input_size: 13
|
| 107 |
+
n_action_steps: 1
|
| 108 |
+
n_obs_steps: 1
|
| 109 |
+
obs_dim: 9
|
| 110 |
+
task:
|
| 111 |
+
value:
|
| 112 |
+
action_dim: 4
|
| 113 |
+
dataset:
|
| 114 |
+
_target_: llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset
|
| 115 |
+
data_path: datasets/box-close-v2-general-reward-split0.02.pt
|
| 116 |
+
data_path2: datasets/box-close-v2-reward-split0.02.pt
|
| 117 |
+
dummy_normalizer: false
|
| 118 |
+
horizon: 1
|
| 119 |
+
obs_eef_target: true
|
| 120 |
+
pad_after: 0
|
| 121 |
+
pad_before: 0
|
| 122 |
+
use_manual_normalizer: false
|
| 123 |
+
val_ratio: 0.1
|
| 124 |
+
env_runner:
|
| 125 |
+
_target_: llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner
|
| 126 |
+
discount: 0.9
|
| 127 |
+
env_name: llf-metaworld-box-close-v2
|
| 128 |
+
feedback_type:
|
| 129 |
+
- hp
|
| 130 |
+
- hn
|
| 131 |
+
- fp
|
| 132 |
+
instruction_type: b
|
| 133 |
+
max_steps: 30
|
| 134 |
+
n_action_steps: 1
|
| 135 |
+
n_envs: 10
|
| 136 |
+
n_obs_steps: 1
|
| 137 |
+
n_test: 50
|
| 138 |
+
n_train: 10
|
| 139 |
+
visual: false
|
| 140 |
+
instructor:
|
| 141 |
+
_target_: llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor
|
| 142 |
+
name: box-close-v2
|
| 143 |
+
obs_dim: 9
|
| 144 |
+
task_name:
|
| 145 |
+
value: box-close-v2
|
| 146 |
+
training:
|
| 147 |
+
value:
|
| 148 |
+
checkpoint_every: 50
|
| 149 |
+
debug: false
|
| 150 |
+
device: cuda
|
| 151 |
+
grad_norm_clip: 0.5
|
| 152 |
+
gradient_accumulate_every: 1
|
| 153 |
+
lr_scheduler: cosine
|
| 154 |
+
lr_warmup_steps: 0
|
| 155 |
+
max_train_steps: null
|
| 156 |
+
max_val_steps: null
|
| 157 |
+
num_epochs: 10000
|
| 158 |
+
resume: true
|
| 159 |
+
sample_every: 50
|
| 160 |
+
sample_max_batch: 1000
|
| 161 |
+
seed: 42
|
| 162 |
+
tqdm_interval_sec: 1
|
| 163 |
+
val_every: 1
|
| 164 |
+
val_dataloader:
|
| 165 |
+
value:
|
| 166 |
+
batch_size: 256
|
| 167 |
+
num_workers: 0
|
| 168 |
+
persistent_workers: false
|
| 169 |
+
pin_memory: false
|
| 170 |
+
shuffle: true
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/files/output.log
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/files/wandb-metadata.json
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"os": "Linux-5.15.0-138-generic-x86_64-with-glibc2.31",
|
| 3 |
+
"python": "3.9.21",
|
| 4 |
+
"startedAt": "2026-01-05T11:09:33.466339Z",
|
| 5 |
+
"program": "/tmp2/chyang/workspace/LLM-BC/./llmbc/workspace/train_rbc_reward_workspace.py",
|
| 6 |
+
"codePath": "llmbc/workspace/train_rbc_reward_workspace.py",
|
| 7 |
+
"git": {
|
| 8 |
+
"remote": "https://github.com/CHYang25/LLM-BC.git",
|
| 9 |
+
"commit": "f7001fc0c1467efdbf9301e78bc936338d67bf0a"
|
| 10 |
+
},
|
| 11 |
+
"email": "chris920325@gmail.com",
|
| 12 |
+
"root": "/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2",
|
| 13 |
+
"host": "rllab1012",
|
| 14 |
+
"username": "chyang",
|
| 15 |
+
"executable": "/home/chyang/miniconda3/envs/llm-bc/bin/python3",
|
| 16 |
+
"codePathLocal": "llmbc/workspace/train_rbc_reward_workspace.py",
|
| 17 |
+
"cpu_count": 10,
|
| 18 |
+
"cpu_count_logical": 20,
|
| 19 |
+
"gpu": "NVIDIA GeForce RTX 4070 Ti",
|
| 20 |
+
"gpu_count": 2,
|
| 21 |
+
"disk": {
|
| 22 |
+
"/": {
|
| 23 |
+
"total": "982820896768",
|
| 24 |
+
"used": "215198687232"
|
| 25 |
+
}
|
| 26 |
+
},
|
| 27 |
+
"memory": {
|
| 28 |
+
"total": "134733287424"
|
| 29 |
+
},
|
| 30 |
+
"cpu": {
|
| 31 |
+
"count": 10,
|
| 32 |
+
"countLogical": 20
|
| 33 |
+
},
|
| 34 |
+
"gpu_nvidia": [
|
| 35 |
+
{
|
| 36 |
+
"name": "NVIDIA GeForce RTX 4070 Ti",
|
| 37 |
+
"memoryTotal": "12878610432",
|
| 38 |
+
"cudaCores": 7680,
|
| 39 |
+
"architecture": "Ada"
|
| 40 |
+
},
|
| 41 |
+
{
|
| 42 |
+
"name": "NVIDIA GeForce RTX 4070 Ti",
|
| 43 |
+
"memoryTotal": "12878610432",
|
| 44 |
+
"cudaCores": 7680,
|
| 45 |
+
"architecture": "Ada"
|
| 46 |
+
}
|
| 47 |
+
],
|
| 48 |
+
"cudaVersion": "12.9"
|
| 49 |
+
}
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/files/wandb-summary.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"grad_norm":0.11417969316244125,"train_loss":0.013938702798138062,"lr":0,"train_batch_mse_error":0.04234898090362549,"_timestamp":1.7676117215757732e+09,"_runtime":348.110437019,"_step":59999,"_wandb":{"runtime":348},"val_loss":0.0233740396797657,"epoch":9999,"global_step":59999}
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/logs/debug-core.log
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{"time":"2026-01-05T19:09:32.866769723+08:00","level":"INFO","msg":"started logging, with flags","port-filename":"/tmp/tmponstbrpa/port-101140.txt","pid":101140,"debug":false,"disable-analytics":false}
|
| 2 |
+
{"time":"2026-01-05T19:09:32.866817602+08:00","level":"INFO","msg":"FeatureState","shutdownOnParentExitEnabled":false}
|
| 3 |
+
{"time":"2026-01-05T19:09:32.871725619+08:00","level":"INFO","msg":"Will exit if parent process dies.","ppid":101140}
|
| 4 |
+
{"time":"2026-01-05T19:09:32.871660419+08:00","level":"INFO","msg":"server is running","addr":{"IP":"127.0.0.1","Port":46291,"Zone":""}}
|
| 5 |
+
{"time":"2026-01-05T19:09:33.05762616+08:00","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"127.0.0.1:50020"}
|
| 6 |
+
{"time":"2026-01-05T19:09:33.468793326+08:00","level":"INFO","msg":"handleInformInit: received","streamId":"f5kmrluf","id":"127.0.0.1:50020"}
|
| 7 |
+
{"time":"2026-01-05T19:09:33.578790867+08:00","level":"INFO","msg":"handleInformInit: stream started","streamId":"f5kmrluf","id":"127.0.0.1:50020"}
|
| 8 |
+
{"time":"2026-01-05T19:15:21.57665803+08:00","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"127.0.0.1:50020"}
|
| 9 |
+
{"time":"2026-01-05T19:15:21.576750574+08:00","level":"INFO","msg":"server is shutting down"}
|
| 10 |
+
{"time":"2026-01-05T19:15:21.576743838+08:00","level":"INFO","msg":"connection: Close: initiating connection closure","id":"127.0.0.1:50020"}
|
| 11 |
+
{"time":"2026-01-05T19:15:21.576866783+08:00","level":"INFO","msg":"connection: Close: connection successfully closed","id":"127.0.0.1:50020"}
|
| 12 |
+
{"time":"2026-01-05T19:15:22.920487318+08:00","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"127.0.0.1:50020"}
|
| 13 |
+
{"time":"2026-01-05T19:15:22.920527959+08:00","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"127.0.0.1:50020"}
|
| 14 |
+
{"time":"2026-01-05T19:15:22.920564361+08:00","level":"INFO","msg":"server is closed"}
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/logs/debug-internal.log
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{"time":"2026-01-05T19:09:33.469014101+08:00","level":"INFO","msg":"using version","core version":"0.18.6"}
|
| 2 |
+
{"time":"2026-01-05T19:09:33.469038682+08:00","level":"INFO","msg":"created symlink","path":"/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/logs/debug-core.log"}
|
| 3 |
+
{"time":"2026-01-05T19:09:33.578729199+08:00","level":"INFO","msg":"created new stream","id":"f5kmrluf"}
|
| 4 |
+
{"time":"2026-01-05T19:09:33.578780561+08:00","level":"INFO","msg":"stream: started","id":"f5kmrluf"}
|
| 5 |
+
{"time":"2026-01-05T19:09:33.578893262+08:00","level":"INFO","msg":"sender: started","stream_id":"f5kmrluf"}
|
| 6 |
+
{"time":"2026-01-05T19:09:33.578806052+08:00","level":"INFO","msg":"writer: Do: started","stream_id":{"value":"f5kmrluf"}}
|
| 7 |
+
{"time":"2026-01-05T19:09:33.578868514+08:00","level":"INFO","msg":"handler: started","stream_id":{"value":"f5kmrluf"}}
|
| 8 |
+
{"time":"2026-01-05T19:09:34.295619023+08:00","level":"INFO","msg":"Starting system monitor"}
|
| 9 |
+
{"time":"2026-01-05T19:15:21.576746822+08:00","level":"INFO","msg":"stream: closing","id":"f5kmrluf"}
|
| 10 |
+
{"time":"2026-01-05T19:15:21.576787059+08:00","level":"INFO","msg":"Stopping system monitor"}
|
| 11 |
+
{"time":"2026-01-05T19:15:21.577575522+08:00","level":"INFO","msg":"Stopped system monitor"}
|
| 12 |
+
{"time":"2026-01-05T19:15:22.630438342+08:00","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
|
| 13 |
+
{"time":"2026-01-05T19:15:22.920151712+08:00","level":"INFO","msg":"handler: closed","stream_id":{"value":"f5kmrluf"}}
|
| 14 |
+
{"time":"2026-01-05T19:15:22.920233247+08:00","level":"INFO","msg":"sender: closed","stream_id":"f5kmrluf"}
|
| 15 |
+
{"time":"2026-01-05T19:15:22.920232896+08:00","level":"INFO","msg":"writer: Close: closed","stream_id":{"value":"f5kmrluf"}}
|
| 16 |
+
{"time":"2026-01-05T19:15:22.920391499+08:00","level":"INFO","msg":"stream: closed","id":"f5kmrluf"}
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/logs/debug.log
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Current SDK version is 0.18.6
|
| 2 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Configure stats pid to 101140
|
| 3 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Loading settings from /home/chyang/.config/wandb/settings
|
| 4 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Loading settings from /tmp2/chyang/workspace/LLM-BC/wandb/settings
|
| 5 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Loading settings from environment variables: {}
|
| 6 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Applying setup settings: {'mode': 'online', '_disable_service': None}
|
| 7 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Inferring run settings from compute environment: {'program_relpath': 'llmbc/workspace/train_rbc_reward_workspace.py', 'program_abspath': '/tmp2/chyang/workspace/LLM-BC/llmbc/workspace/train_rbc_reward_workspace.py', 'program': '/tmp2/chyang/workspace/LLM-BC/./llmbc/workspace/train_rbc_reward_workspace.py'}
|
| 8 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_setup.py:_flush():79] Applying login settings: {}
|
| 9 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_init.py:_log_setup():533] Logging user logs to /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/logs/debug.log
|
| 10 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_init.py:_log_setup():534] Logging internal logs to /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/logs/debug-internal.log
|
| 11 |
+
2026-01-05 19:09:33,464 INFO MainThread:101140 [wandb_init.py:init():619] calling init triggers
|
| 12 |
+
2026-01-05 19:09:33,465 INFO MainThread:101140 [wandb_init.py:init():626] wandb.init called with sweep_config: {}
|
| 13 |
+
config: {'name': 'train_rbc_reward_model', '_target_': 'llmbc.workspace.train_rbc_reward_workspace.TrainRBCRewardWorkspace', 'obs_dim': 9, 'action_dim': 4, 'task_name': 'box-close-v2', 'exp_name': 'default', 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1, 'n_latency_steps': 0, 'past_action_visible': False, 'reward_model': {'_target_': 'llmbc.policy.rbc_lowdim_policy.RBCLowdimRewardModel', 'input_size': 13, 'hidden_size': [128], 'obs_dim': 9, 'action_dim': 4, 'n_obs_steps': 1, 'n_action_steps': 1}, 'dataloader': {'batch_size': 256, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'val_dataloader': {'batch_size': 256, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'optimizer': {'_target_': 'torch.optim.Adam', 'lr': 5e-06, 'betas': [0.95, 0.999], 'eps': 1e-08, 'weight_decay': 1e-06}, 'training': {'device': 'cuda', 'seed': 42, 'debug': False, 'resume': True, 'lr_scheduler': 'cosine', 'lr_warmup_steps': 0, 'num_epochs': 10000, 'gradient_accumulate_every': 1, 'grad_norm_clip': 0.5, 'checkpoint_every': 50, 'val_every': 1, 'sample_every': 50, 'sample_max_batch': 1000, 'max_train_steps': None, 'max_val_steps': None, 'tqdm_interval_sec': 1.0}, 'logging': {'project': 'rbc_module_training', 'resume': True, 'mode': 'online', 'name': '2026.01.05-19.09.32_train_rbc_reward_model_box-close-v2', 'tags': ['train_rbc_reward_model', 'box-close-v2', 'default'], 'id': None, 'group': None}, 'checkpoint': {'topk': {'monitor_key': 'val_loss', 'mode': 'min', 'k': 5, 'format_str': 'epoch={epoch:04d}-val_loss={val_loss:.3f}.ckpt'}, 'save_last_ckpt': True, 'save_last_snapshot': False}, 'multi_run': {'run_dir': 'data/outputs/2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2', 'wandb_name_base': '2026.01.05-19.09.32_train_rbc_reward_model_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-general-reward-split0.02.pt', 'data_path2': 'datasets/box-close-v2-reward-split0.02.pt', 'horizon': 1, 'pad_before': 0, 'pad_after': 0, 'obs_eef_target': True, 'use_manual_normalizer': False, 'val_ratio': 0.1, 'dummy_normalizer': False}, 'instructor': {'_target_': 'llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor'}}}
|
| 14 |
+
2026-01-05 19:09:33,465 INFO MainThread:101140 [wandb_init.py:init():669] starting backend
|
| 15 |
+
2026-01-05 19:09:33,465 INFO MainThread:101140 [wandb_init.py:init():673] sending inform_init request
|
| 16 |
+
2026-01-05 19:09:33,465 INFO MainThread:101140 [backend.py:_multiprocessing_setup():104] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
|
| 17 |
+
2026-01-05 19:09:33,466 INFO MainThread:101140 [wandb_init.py:init():686] backend started and connected
|
| 18 |
+
2026-01-05 19:09:33,469 INFO MainThread:101140 [wandb_init.py:init():781] updated telemetry
|
| 19 |
+
2026-01-05 19:09:33,491 INFO MainThread:101140 [wandb_init.py:init():814] communicating run to backend with 90.0 second timeout
|
| 20 |
+
2026-01-05 19:09:34,290 INFO MainThread:101140 [wandb_init.py:init():867] starting run threads in backend
|
| 21 |
+
2026-01-05 19:09:34,456 INFO MainThread:101140 [wandb_run.py:_console_start():2451] atexit reg
|
| 22 |
+
2026-01-05 19:09:34,456 INFO MainThread:101140 [wandb_run.py:_redirect():2299] redirect: wrap_raw
|
| 23 |
+
2026-01-05 19:09:34,457 INFO MainThread:101140 [wandb_run.py:_redirect():2364] Wrapping output streams.
|
| 24 |
+
2026-01-05 19:09:34,457 INFO MainThread:101140 [wandb_run.py:_redirect():2389] Redirects installed.
|
| 25 |
+
2026-01-05 19:09:34,458 INFO MainThread:101140 [wandb_init.py:init():911] run started, returning control to user process
|
| 26 |
+
2026-01-05 19:09:34,459 INFO MainThread:101140 [wandb_run.py:_config_callback():1389] config_cb None None {'output_dir': '/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2'}
|
| 27 |
+
2026-01-05 19:15:21,576 WARNING MsgRouterThr:101140 [router.py:message_loop():75] message_loop has been closed
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/run-20260105_190933-f5kmrluf/run-f5kmrluf.wandb
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:346e7ba4920fc7674eb50498db875d6695a8587ec706d6ff96d01fdde93a5060
|
| 3 |
+
size 44046595
|
2026.01.05/19.09.31_train_rbc_reward_model_box-close-v2/wandb/wandb-resume.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"run_id": "f5kmrluf"}
|