Upload folder using huggingface_hub

.gitattributes

@@ -307,3 +307,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 2026.10.12/11.25.44_train_mhcbc_classifier_model_FrankaKitchen-v1/wandb/run-20261012_112546-hz07rjhi/run-hz07rjhi.wandb filter=lfs diff=lfs merge=lfs -text
 2026.10.18/18.08.03_train_rbc_reward_model_FrankaKitchen-v1/logs.json.txt filter=lfs diff=lfs merge=lfs -text
 2026.10.18/18.08.03_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261018_180805-hknktw63/run-hknktw63.wandb filter=lfs diff=lfs merge=lfs -text
+2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/logs.json.txt filter=lfs diff=lfs merge=lfs -text
+2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261028_135422-ux76l2ws/run-ux76l2ws.wandb filter=lfs diff=lfs merge=lfs -text

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/.hydra/config.yaml

@@ -0,0 +1,125 @@
+name: train_rbc_reward_model
+_target_: llmbc.workspace.train_rbc_reward_workspace.TrainRBCRewardWorkspace
+obs_dim: ${task.obs_dim}
+action_dim: ${task.action_dim}
+task_name: ${task.name}
+exp_name: default
+horizon: 16
+n_obs_steps: 2
+n_action_steps: 8
+n_latency_steps: 0
+past_action_visible: false
+reward_model:
+  _target_: llmbc.policy.rbc_lowdim_policy.RBCLowdimRewardModel
+  input_size: ${eval:'${n_obs_steps}*${obs_dim}+${horizon}*${action_dim}'}
+  hidden_size:
+  - 512
+  - 512
+  - 512
+  obs_dim: ${obs_dim}
+  action_dim: ${action_dim}
+  n_obs_steps: ${n_obs_steps}
+  n_action_steps: ${horizon}
+  reward_dim: 1
+dataloader:
+  batch_size: 1024
+  num_workers: 0
+  shuffle: true
+  pin_memory: false
+  persistent_workers: false
+val_dataloader:
+  batch_size: 1024
+  num_workers: 0
+  shuffle: true
+  pin_memory: false
+  persistent_workers: false
+optimizer:
+  _target_: torch.optim.Adam
+  lr: 1.0e-05
+  betas:
+  - 0.95
+  - 0.999
+  eps: 1.0e-08
+  weight_decay: 1.0e-06
+training:
+  device: cuda
+  seed: 42
+  debug: false
+  resume: true
+  lr_scheduler: cosine
+  lr_warmup_steps: 0
+  num_epochs: 500
+  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: null
+  max_val_steps: null
+  tqdm_interval_sec: 1.0
+logging:
+  project: rbc_module_training
+  resume: true
+  mode: online
+  name: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
+  tags:
+  - ${name}
+  - ${task_name}
+  - ${exp_name}
+  id: null
+  group: null
+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/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+  wandb_name_base: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
+task:
+  name: FrankaKitchen-v1
+  obs_dim: 59
+  action_dim: 9
+  lang_multistep_merger:
+    _target_: llfbench.envs.kitchen.multistep_merger.franka_kitchen_merger.KitchenMultistepMerger
+  env_runner:
+    _target_: llmbc.env_runner.kitchen_lowdim_runner.KitchenLowdimRunner
+    env_name: llf-kitchen-FrankaKitchen-v1
+    n_train: 10
+    n_test: 50
+    n_envs: 10
+    max_steps: 500
+    n_obs_steps: ${n_obs_steps}
+    n_action_steps: ${n_action_steps}
+    instruction_type: b
+    feedback_type:
+    - hp
+    - hn
+    - fp
+    tasks_to_complete:
+    - microwave
+    - kettle
+    - light switch
+    - slide cabinet
+    visual: false
+    discount: 0.99
+    inference_expert_action: false
+    lang_multistep_merger:
+      _target_: llfbench.envs.kitchen.multistep_merger.franka_kitchen_merger.KitchenMultistepMerger
+  dataset:
+    _target_: llmbc.dataset.kitchen_lowdim_dataset.KitchenLowdimDataset
+    data_path: datasets/FrankaKitchen-v1-general-freward.pt
+    data_path2: datasets/FrankaKitchen-v1-freward.pt
+    horizon: ${horizon}
+    pad_before: ${eval:'${n_obs_steps}-1'}
+    pad_after: ${eval:'${n_action_steps}-1'}
+    obs_eef_target: true
+    use_manual_normalizer: false
+    val_ratio: 0.05
+    dummy_normalizer: false
+  instructor:
+    _target_: llmbc.translator.instructor.kitchen_instructor.FrankaKitchen_v1_instructor.FrankaKitchenV1Instructor

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/.hydra/hydra.yaml

@@ -0,0 +1,154 @@
+hydra:
+  run:
+    dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+  sweep:
+    dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+    subdir: ${hydra.job.num}
+  launcher:
+    _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
+  sweeper:
+    _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
+    max_batch_size: null
+    params: null
+  help:
+    app_name: ${hydra.job.name}
+    header: '${hydra.help.app_name} is powered by Hydra.
+
+      '
+    footer: 'Powered by Hydra (https://hydra.cc)
+
+      Use --hydra-help to view Hydra specific help
+
+      '
+    template: '${hydra.help.header}
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (group=option)
+
+
+      $APP_CONFIG_GROUPS
+
+
+      == Config ==
+
+      Override anything in the config (foo.bar=value)
+
+
+      $CONFIG
+
+
+      ${hydra.help.footer}
+
+      '
+  hydra_help:
+    template: 'Hydra (${hydra.runtime.version})
+
+      See https://hydra.cc for more info.
+
+
+      == Flags ==
+
+      $FLAGS_HELP
+
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (For example, append hydra/job_logging=disabled
+      to command line)
+
+
+      $HYDRA_CONFIG_GROUPS
+
+
+      Use ''--cfg hydra'' to Show the Hydra config.
+
+      '
+    hydra_help: ???
+  hydra_logging:
+    version: 1
+    formatters:
+      simple:
+        format: '[%(asctime)s][HYDRA] %(message)s'
+    handlers:
+      console:
+        class: logging.StreamHandler
+        formatter: simple
+        stream: ext://sys.stdout
+    root:
+      level: INFO
+      handlers:
+      - console
+    loggers:
+      logging_example:
+        level: DEBUG
+    disable_existing_loggers: false
+  job_logging:
+    version: 1
+    formatters:
+      simple:
+        format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
+    handlers:
+      console:
+        class: logging.StreamHandler
+        formatter: simple
+        stream: ext://sys.stdout
+      file:
+        class: logging.FileHandler
+        formatter: simple
+        filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
+    root:
+      level: INFO
+      handlers:
+      - console
+      - file
+    disable_existing_loggers: false
+  env: {}
+  mode: RUN
+  searchpath: []
+  callbacks: {}
+  output_subdir: .hydra
+  overrides:
+    hydra:
+    - hydra.mode=RUN
+    task: []
+  job:
+    name: train
+    chdir: null
+    override_dirname: ''
+    id: ???
+    num: ???
+    config_name: rbc_reward-freward_FrankaKitchen-v1.yaml
+    env_set: {}
+    env_copy: []
+    config:
+      override_dirname:
+        kv_sep: '='
+        item_sep: ','
+        exclude_keys: []
+  runtime:
+    version: 1.2.0
+    version_base: '1.2'
+    cwd: /tmp2/chyang/workspace/LLM-BC
+    config_sources:
+    - path: hydra.conf
+      schema: pkg
+      provider: hydra
+    - path: /tmp2/chyang/workspace/LLM-BC/config/main_table
+      schema: file
+      provider: main
+    - path: ''
+      schema: structured
+      provider: schema
+    output_dir: /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1
+    choices:
+      hydra/env: default
+      hydra/callbacks: null
+      hydra/job_logging: default
+      hydra/hydra_logging: default
+      hydra/hydra_help: default
+      hydra/help: default
+      hydra/sweeper: basic
+      hydra/launcher: basic
+      hydra/output: default
+  verbose: false

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/.hydra/overrides.yaml

@@ -0,0 +1 @@
+[]

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/checkpoints/epoch=0250-val_loss=0.661.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5645700a112bc0e4a89de4e54e90a2ff8280a69e1a31073bf815774a3682842d
+size 8049932

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/checkpoints/epoch=0300-val_loss=0.657.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1ca296090731bfc32e568882475c770aaef913787fee91edce94c686eb257d54
+size 8049932

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/checkpoints/epoch=0350-val_loss=0.658.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8146c40d3ddf23b3b02b814e5a53e54b997851ae60746c4096a9e0e98f7e7177
+size 8049932

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/checkpoints/epoch=0400-val_loss=0.653.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:39aeb90bc4bf0dd1321441409bd66fb71a28699aa72315eea3394b3d488614bf
+size 8049932

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/checkpoints/epoch=0450-val_loss=0.652.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:97b593b24b2d347c1d78ff8325f73a81b4a44ddb639bbee652f3663b27071e81
+size 8049932

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/checkpoints/latest.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:97b593b24b2d347c1d78ff8325f73a81b4a44ddb639bbee652f3663b27071e81
+size 8049932

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/logs.json.txt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c5f87183c7e82845066426d3f73ceea52543abc37799df7e38c8829b2ebfb806
+size 51087405

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/train.log

@@ -0,0 +1,2 @@
+[2026-10-28 13:54:19,823][numexpr.utils][INFO] - Note: NumExpr detected 40 cores but "NUMEXPR_MAX_THREADS" not set, so enforcing safe limit of 16.
+[2026-10-28 13:54:19,823][numexpr.utils][INFO] - NumExpr defaulting to 16 threads.

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/debug-internal.log

@@ -0,0 +1,16 @@
+{"time":"2026-10-28T13:54:22.04058983+08:00","level":"INFO","msg":"using version","core version":"0.18.6"}
+{"time":"2026-10-28T13:54:22.040597822+08:00","level":"INFO","msg":"created symlink","path":"/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261028_135422-ux76l2ws/logs/debug-core.log"}
+{"time":"2026-10-28T13:54:22.146045588+08:00","level":"INFO","msg":"created new stream","id":"ux76l2ws"}
+{"time":"2026-10-28T13:54:22.14635216+08:00","level":"INFO","msg":"stream: started","id":"ux76l2ws"}
+{"time":"2026-10-28T13:54:22.146365949+08:00","level":"INFO","msg":"sender: started","stream_id":"ux76l2ws"}
+{"time":"2026-10-28T13:54:22.146441597+08:00","level":"INFO","msg":"writer: Do: started","stream_id":{"value":"ux76l2ws"}}
+{"time":"2026-10-28T13:54:22.146454052+08:00","level":"INFO","msg":"handler: started","stream_id":{"value":"ux76l2ws"}}
+{"time":"2026-10-28T13:54:23.682925087+08:00","level":"INFO","msg":"Starting system monitor"}
+{"time":"2026-10-28T15:03:33.214215188+08:00","level":"INFO","msg":"stream: closing","id":"ux76l2ws"}
+{"time":"2026-10-28T15:03:33.214240966+08:00","level":"INFO","msg":"Stopping system monitor"}
+{"time":"2026-10-28T15:03:33.214801794+08:00","level":"INFO","msg":"Stopped system monitor"}
+{"time":"2026-10-28T15:03:36.392182341+08:00","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
+{"time":"2026-10-28T15:03:37.070412327+08:00","level":"INFO","msg":"handler: closed","stream_id":{"value":"ux76l2ws"}}
+{"time":"2026-10-28T15:03:37.07051065+08:00","level":"INFO","msg":"writer: Close: closed","stream_id":{"value":"ux76l2ws"}}
+{"time":"2026-10-28T15:03:37.07054647+08:00","level":"INFO","msg":"sender: closed","stream_id":"ux76l2ws"}
+{"time":"2026-10-28T15:03:37.070619326+08:00","level":"INFO","msg":"stream: closed","id":"ux76l2ws"}

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/debug.log

@@ -0,0 +1,27 @@
+2026-10-28 13:54:22,037 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Current SDK version is 0.18.6
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Configure stats pid to 3216302
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Loading settings from /home/chyang/.config/wandb/settings
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Loading settings from /tmp2/chyang/workspace/LLM-BC/wandb/settings
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Loading settings from environment variables: {}
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Applying setup settings: {'mode': 'online', '_disable_service': None}
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Inferring run settings from compute environment: {'program_relpath': 'train.py', 'program_abspath': '/tmp2/chyang/workspace/LLM-BC/train.py', 'program': '/tmp2/chyang/workspace/LLM-BC/./train.py'}
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Applying login settings: {}
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_init.py:_log_setup():533] Logging user logs to /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261028_135422-ux76l2ws/logs/debug.log
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_init.py:_log_setup():534] Logging internal logs to /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261028_135422-ux76l2ws/logs/debug-internal.log
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_init.py:init():619] calling init triggers
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_init.py:init():626] wandb.init called with sweep_config: {}
+config: {'name': 'train_rbc_reward_model', '_target_': 'llmbc.workspace.train_rbc_reward_workspace.TrainRBCRewardWorkspace', 'obs_dim': 59, 'action_dim': 9, 'task_name': 'FrankaKitchen-v1', 'exp_name': 'default', 'horizon': 16, 'n_obs_steps': 2, 'n_action_steps': 8, 'n_latency_steps': 0, 'past_action_visible': False, 'reward_model': {'_target_': 'llmbc.policy.rbc_lowdim_policy.RBCLowdimRewardModel', 'input_size': 262, 'hidden_size': [512, 512, 512], 'obs_dim': 59, 'action_dim': 9, 'n_obs_steps': 2, 'n_action_steps': 16, 'reward_dim': 1}, 'dataloader': {'batch_size': 1024, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'val_dataloader': {'batch_size': 1024, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'optimizer': {'_target_': 'torch.optim.Adam', 'lr': 1e-05, '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': 500, '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.10.28-13.54.19_train_rbc_reward_model_FrankaKitchen-v1', 'tags': ['train_rbc_reward_model', 'FrankaKitchen-v1', '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.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1', 'wandb_name_base': '2026.10.28-13.54.19_train_rbc_reward_model_FrankaKitchen-v1'}, 'task': {'name': 'FrankaKitchen-v1', 'obs_dim': 59, 'action_dim': 9, 'lang_multistep_merger': {'_target_': 'llfbench.envs.kitchen.multistep_merger.franka_kitchen_merger.KitchenMultistepMerger'}, 'env_runner': {'_target_': 'llmbc.env_runner.kitchen_lowdim_runner.KitchenLowdimRunner', 'env_name': 'llf-kitchen-FrankaKitchen-v1', 'n_train': 10, 'n_test': 50, 'n_envs': 10, 'max_steps': 500, 'n_obs_steps': 2, 'n_action_steps': 8, 'instruction_type': 'b', 'feedback_type': ['hp', 'hn', 'fp'], 'tasks_to_complete': ['microwave', 'kettle', 'light switch', 'slide cabinet'], 'visual': False, 'discount': 0.99, 'inference_expert_action': False, 'lang_multistep_merger': {'_target_': 'llfbench.envs.kitchen.multistep_merger.franka_kitchen_merger.KitchenMultistepMerger'}}, 'dataset': {'_target_': 'llmbc.dataset.kitchen_lowdim_dataset.KitchenLowdimDataset', 'data_path': 'datasets/FrankaKitchen-v1-general-freward.pt', 'data_path2': 'datasets/FrankaKitchen-v1-freward.pt', 'horizon': 16, 'pad_before': 1, 'pad_after': 7, 'obs_eef_target': True, 'use_manual_normalizer': False, 'val_ratio': 0.05, 'dummy_normalizer': False}, 'instructor': {'_target_': 'llmbc.translator.instructor.kitchen_instructor.FrankaKitchen_v1_instructor.FrankaKitchenV1Instructor'}}}
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_init.py:init():669] starting backend
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_init.py:init():673] sending inform_init request
+2026-10-28 13:54:22,039 INFO    MainThread:3216302 [backend.py:_multiprocessing_setup():104] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
+2026-10-28 13:54:22,039 INFO    MainThread:3216302 [wandb_init.py:init():686] backend started and connected
+2026-10-28 13:54:22,044 INFO    MainThread:3216302 [wandb_init.py:init():781] updated telemetry
+2026-10-28 13:54:22,068 INFO    MainThread:3216302 [wandb_init.py:init():814] communicating run to backend with 90.0 second timeout
+2026-10-28 13:54:23,679 INFO    MainThread:3216302 [wandb_init.py:init():867] starting run threads in backend
+2026-10-28 13:54:23,789 INFO    MainThread:3216302 [wandb_run.py:_console_start():2451] atexit reg
+2026-10-28 13:54:23,789 INFO    MainThread:3216302 [wandb_run.py:_redirect():2299] redirect: wrap_raw
+2026-10-28 13:54:23,789 INFO    MainThread:3216302 [wandb_run.py:_redirect():2364] Wrapping output streams.
+2026-10-28 13:54:23,790 INFO    MainThread:3216302 [wandb_run.py:_redirect():2389] Redirects installed.
+2026-10-28 13:54:23,792 INFO    MainThread:3216302 [wandb_init.py:init():911] run started, returning control to user process
+2026-10-28 13:54:23,792 INFO    MainThread:3216302 [wandb_run.py:_config_callback():1389] config_cb None None {'output_dir': '/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1'}
+2026-10-28 15:03:33,214 WARNING MsgRouterThr:3216302 [router.py:message_loop():75] message_loop has been closed

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261028_135422-ux76l2ws/files/config.yaml

@@ -0,0 +1,183 @@
+_target_:
+    value: llmbc.workspace.train_rbc_reward_workspace.TrainRBCRewardWorkspace
+_wandb:
+    value:
+        cli_version: 0.18.6
+        m: []
+        python_version: 3.9.23
+        t:
+            "1":
+                - 1
+                - 11
+                - 49
+                - 50
+                - 55
+                - 83
+                - 105
+            "2":
+                - 1
+                - 11
+                - 49
+                - 50
+                - 55
+                - 83
+                - 105
+            "3":
+                - 13
+                - 15
+                - 16
+                - 23
+                - 55
+                - 61
+            "4": 3.9.23
+            "5": 0.18.6
+            "6": 4.47.1
+            "8":
+                - 5
+            "12": 0.18.6
+            "13": linux-x86_64
+action_dim:
+    value: 9
+checkpoint:
+    value:
+        save_last_ckpt: true
+        save_last_snapshot: false
+        topk:
+            format_str: epoch={epoch:04d}-val_loss={val_loss:.3f}.ckpt
+            k: 5
+            mode: min
+            monitor_key: val_loss
+dataloader:
+    value:
+        batch_size: 1024
+        num_workers: 0
+        persistent_workers: false
+        pin_memory: false
+        shuffle: true
+exp_name:
+    value: default
+horizon:
+    value: 16
+logging:
+    value:
+        group: null
+        id: null
+        mode: online
+        name: 2026.10.28-13.54.19_train_rbc_reward_model_FrankaKitchen-v1
+        project: rbc_module_training
+        resume: true
+        tags:
+            - train_rbc_reward_model
+            - FrankaKitchen-v1
+            - default
+multi_run:
+    value:
+        run_dir: data/outputs/2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1
+        wandb_name_base: 2026.10.28-13.54.19_train_rbc_reward_model_FrankaKitchen-v1
+n_action_steps:
+    value: 8
+n_latency_steps:
+    value: 0
+n_obs_steps:
+    value: 2
+name:
+    value: train_rbc_reward_model
+obs_dim:
+    value: 59
+optimizer:
+    value:
+        _target_: torch.optim.Adam
+        betas:
+            - 0.95
+            - 0.999
+        eps: 1e-08
+        lr: 1e-05
+        weight_decay: 1e-06
+output_dir:
+    value: /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1
+past_action_visible:
+    value: false
+reward_model:
+    value:
+        _target_: llmbc.policy.rbc_lowdim_policy.RBCLowdimRewardModel
+        action_dim: 9
+        hidden_size:
+            - 512
+            - 512
+            - 512
+        input_size: 262
+        n_action_steps: 16
+        n_obs_steps: 2
+        obs_dim: 59
+        reward_dim: 1
+task:
+    value:
+        action_dim: 9
+        dataset:
+            _target_: llmbc.dataset.kitchen_lowdim_dataset.KitchenLowdimDataset
+            data_path: datasets/FrankaKitchen-v1-general-freward.pt
+            data_path2: datasets/FrankaKitchen-v1-freward.pt
+            dummy_normalizer: false
+            horizon: 16
+            obs_eef_target: true
+            pad_after: 7
+            pad_before: 1
+            use_manual_normalizer: false
+            val_ratio: 0.05
+        env_runner:
+            _target_: llmbc.env_runner.kitchen_lowdim_runner.KitchenLowdimRunner
+            discount: 0.99
+            env_name: llf-kitchen-FrankaKitchen-v1
+            feedback_type:
+                - hp
+                - hn
+                - fp
+            inference_expert_action: false
+            instruction_type: b
+            lang_multistep_merger:
+                _target_: llfbench.envs.kitchen.multistep_merger.franka_kitchen_merger.KitchenMultistepMerger
+            max_steps: 500
+            n_action_steps: 8
+            n_envs: 10
+            n_obs_steps: 2
+            n_test: 50
+            n_train: 10
+            tasks_to_complete:
+                - microwave
+                - kettle
+                - light switch
+                - slide cabinet
+            visual: false
+        instructor:
+            _target_: llmbc.translator.instructor.kitchen_instructor.FrankaKitchen_v1_instructor.FrankaKitchenV1Instructor
+        lang_multistep_merger:
+            _target_: llfbench.envs.kitchen.multistep_merger.franka_kitchen_merger.KitchenMultistepMerger
+        name: FrankaKitchen-v1
+        obs_dim: 59
+task_name:
+    value: FrankaKitchen-v1
+training:
+    value:
+        checkpoint_every: 50
+        debug: false
+        device: cuda
+        grad_norm_clip: 0.5
+        gradient_accumulate_every: 1
+        lr_scheduler: cosine
+        lr_warmup_steps: 0
+        max_train_steps: null
+        max_val_steps: null
+        num_epochs: 500
+        resume: true
+        sample_every: 50
+        sample_max_batch: 1000
+        seed: 42
+        tqdm_interval_sec: 1
+        val_every: 1
+val_dataloader:
+    value:
+        batch_size: 1024
+        num_workers: 0
+        persistent_workers: false
+        pin_memory: false
+        shuffle: true

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261028_135422-ux76l2ws/files/output.log

@@ -0,0 +1 @@
+                                                                                                                                                                                

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261028_135422-ux76l2ws/files/wandb-metadata.json

@@ -0,0 +1,55 @@
+{
+  "os": "Linux-6.8.0-139-generic-x86_64-with-glibc2.39",
+  "python": "3.9.23",
+  "startedAt": "2026-10-28T05:54:22.040028Z",
+  "args": [
+    "--config-path",
+    "config/main_table",
+    "--config-name",
+    "rbc_reward-freward_FrankaKitchen-v1.yaml"
+  ],
+  "program": "/tmp2/chyang/workspace/LLM-BC/./train.py",
+  "codePath": "train.py",
+  "git": {
+    "remote": "https://github.com/CHYang25/LLM-BC.git",
+    "commit": "30f5d5f1a1a4aaefacdd537c64b272114354098d"
+  },
+  "email": "chris920325@gmail.com",
+  "root": "/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1",
+  "host": "rllab1015",
+  "username": "chyang",
+  "executable": "/home/chyang/miniconda3/envs/llm-bc/bin/python3",
+  "codePathLocal": "train.py",
+  "cpu_count": 20,
+  "cpu_count_logical": 40,
+  "gpu": "NVIDIA GeForce RTX 4090",
+  "gpu_count": 2,
+  "disk": {
+    "/": {
+      "total": "1967317549056",
+      "used": "1370207358976"
+    }
+  },
+  "memory": {
+    "total": "66890878976"
+  },
+  "cpu": {
+    "count": 20,
+    "countLogical": 40
+  },
+  "gpu_nvidia": [
+    {
+      "name": "NVIDIA GeForce RTX 4090",
+      "memoryTotal": "25757220864",
+      "cudaCores": 16384,
+      "architecture": "Ada"
+    },
+    {
+      "name": "NVIDIA GeForce RTX 4090",
+      "memoryTotal": "25757220864",
+      "cudaCores": 16384,
+      "architecture": "Ada"
+    }
+  ],
+  "cudaVersion": "13.0"
+}

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261028_135422-ux76l2ws/files/wandb-summary.json

@@ -0,0 +1 @@
+{"grad_norm":13.23994255065918,"train_loss":0.5743371296301,"_runtime":4151.174204355,"_timestamp":1.7931710132054985e+09,"val_loss":0.652033269405365,"global_step":384499,"epoch":499,"lr":0,"_step":384499,"train_batch_mse_error":6.567255824419713,"_wandb":{"runtime":4151}}

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261028_135422-ux76l2ws/logs/debug-core.log

@@ -0,0 +1,14 @@
+{"time":"2026-10-28T13:54:21.336346658+08:00","level":"INFO","msg":"started logging, with flags","port-filename":"/tmp/tmpx8n3a1qq/port-3216302.txt","pid":3216302,"debug":false,"disable-analytics":false}
+{"time":"2026-10-28T13:54:21.336374483+08:00","level":"INFO","msg":"FeatureState","shutdownOnParentExitEnabled":false}
+{"time":"2026-10-28T13:54:21.33733982+08:00","level":"INFO","msg":"server is running","addr":{"IP":"127.0.0.1","Port":32833,"Zone":""}}
+{"time":"2026-10-28T13:54:21.337360618+08:00","level":"INFO","msg":"Will exit if parent process dies.","ppid":3216302}
+{"time":"2026-10-28T13:54:21.532738402+08:00","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"127.0.0.1:58182"}
+{"time":"2026-10-28T13:54:22.040473047+08:00","level":"INFO","msg":"handleInformInit: received","streamId":"ux76l2ws","id":"127.0.0.1:58182"}
+{"time":"2026-10-28T13:54:22.146381973+08:00","level":"INFO","msg":"handleInformInit: stream started","streamId":"ux76l2ws","id":"127.0.0.1:58182"}
+{"time":"2026-10-28T15:03:33.214152128+08:00","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"127.0.0.1:58182"}
+{"time":"2026-10-28T15:03:33.214215814+08:00","level":"INFO","msg":"server is shutting down"}
+{"time":"2026-10-28T15:03:33.214210041+08:00","level":"INFO","msg":"connection: Close: initiating connection closure","id":"127.0.0.1:58182"}
+{"time":"2026-10-28T15:03:33.214332235+08:00","level":"INFO","msg":"connection: Close: connection successfully closed","id":"127.0.0.1:58182"}
+{"time":"2026-10-28T15:03:37.070727689+08:00","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"127.0.0.1:58182"}
+{"time":"2026-10-28T15:03:37.070769339+08:00","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"127.0.0.1:58182"}
+{"time":"2026-10-28T15:03:37.070794255+08:00","level":"INFO","msg":"server is closed"}

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261028_135422-ux76l2ws/logs/debug-internal.log

@@ -0,0 +1,16 @@
+{"time":"2026-10-28T13:54:22.04058983+08:00","level":"INFO","msg":"using version","core version":"0.18.6"}
+{"time":"2026-10-28T13:54:22.040597822+08:00","level":"INFO","msg":"created symlink","path":"/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261028_135422-ux76l2ws/logs/debug-core.log"}
+{"time":"2026-10-28T13:54:22.146045588+08:00","level":"INFO","msg":"created new stream","id":"ux76l2ws"}
+{"time":"2026-10-28T13:54:22.14635216+08:00","level":"INFO","msg":"stream: started","id":"ux76l2ws"}
+{"time":"2026-10-28T13:54:22.146365949+08:00","level":"INFO","msg":"sender: started","stream_id":"ux76l2ws"}
+{"time":"2026-10-28T13:54:22.146441597+08:00","level":"INFO","msg":"writer: Do: started","stream_id":{"value":"ux76l2ws"}}
+{"time":"2026-10-28T13:54:22.146454052+08:00","level":"INFO","msg":"handler: started","stream_id":{"value":"ux76l2ws"}}
+{"time":"2026-10-28T13:54:23.682925087+08:00","level":"INFO","msg":"Starting system monitor"}
+{"time":"2026-10-28T15:03:33.214215188+08:00","level":"INFO","msg":"stream: closing","id":"ux76l2ws"}
+{"time":"2026-10-28T15:03:33.214240966+08:00","level":"INFO","msg":"Stopping system monitor"}
+{"time":"2026-10-28T15:03:33.214801794+08:00","level":"INFO","msg":"Stopped system monitor"}
+{"time":"2026-10-28T15:03:36.392182341+08:00","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
+{"time":"2026-10-28T15:03:37.070412327+08:00","level":"INFO","msg":"handler: closed","stream_id":{"value":"ux76l2ws"}}
+{"time":"2026-10-28T15:03:37.07051065+08:00","level":"INFO","msg":"writer: Close: closed","stream_id":{"value":"ux76l2ws"}}
+{"time":"2026-10-28T15:03:37.07054647+08:00","level":"INFO","msg":"sender: closed","stream_id":"ux76l2ws"}
+{"time":"2026-10-28T15:03:37.070619326+08:00","level":"INFO","msg":"stream: closed","id":"ux76l2ws"}

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261028_135422-ux76l2ws/logs/debug.log

@@ -0,0 +1,27 @@
+2026-10-28 13:54:22,037 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Current SDK version is 0.18.6
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Configure stats pid to 3216302
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Loading settings from /home/chyang/.config/wandb/settings
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Loading settings from /tmp2/chyang/workspace/LLM-BC/wandb/settings
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Loading settings from environment variables: {}
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Applying setup settings: {'mode': 'online', '_disable_service': None}
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Inferring run settings from compute environment: {'program_relpath': 'train.py', 'program_abspath': '/tmp2/chyang/workspace/LLM-BC/train.py', 'program': '/tmp2/chyang/workspace/LLM-BC/./train.py'}
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_setup.py:_flush():79] Applying login settings: {}
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_init.py:_log_setup():533] Logging user logs to /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261028_135422-ux76l2ws/logs/debug.log
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_init.py:_log_setup():534] Logging internal logs to /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261028_135422-ux76l2ws/logs/debug-internal.log
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_init.py:init():619] calling init triggers
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_init.py:init():626] wandb.init called with sweep_config: {}
+config: {'name': 'train_rbc_reward_model', '_target_': 'llmbc.workspace.train_rbc_reward_workspace.TrainRBCRewardWorkspace', 'obs_dim': 59, 'action_dim': 9, 'task_name': 'FrankaKitchen-v1', 'exp_name': 'default', 'horizon': 16, 'n_obs_steps': 2, 'n_action_steps': 8, 'n_latency_steps': 0, 'past_action_visible': False, 'reward_model': {'_target_': 'llmbc.policy.rbc_lowdim_policy.RBCLowdimRewardModel', 'input_size': 262, 'hidden_size': [512, 512, 512], 'obs_dim': 59, 'action_dim': 9, 'n_obs_steps': 2, 'n_action_steps': 16, 'reward_dim': 1}, 'dataloader': {'batch_size': 1024, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'val_dataloader': {'batch_size': 1024, 'num_workers': 0, 'shuffle': True, 'pin_memory': False, 'persistent_workers': False}, 'optimizer': {'_target_': 'torch.optim.Adam', 'lr': 1e-05, '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': 500, '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.10.28-13.54.19_train_rbc_reward_model_FrankaKitchen-v1', 'tags': ['train_rbc_reward_model', 'FrankaKitchen-v1', '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.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1', 'wandb_name_base': '2026.10.28-13.54.19_train_rbc_reward_model_FrankaKitchen-v1'}, 'task': {'name': 'FrankaKitchen-v1', 'obs_dim': 59, 'action_dim': 9, 'lang_multistep_merger': {'_target_': 'llfbench.envs.kitchen.multistep_merger.franka_kitchen_merger.KitchenMultistepMerger'}, 'env_runner': {'_target_': 'llmbc.env_runner.kitchen_lowdim_runner.KitchenLowdimRunner', 'env_name': 'llf-kitchen-FrankaKitchen-v1', 'n_train': 10, 'n_test': 50, 'n_envs': 10, 'max_steps': 500, 'n_obs_steps': 2, 'n_action_steps': 8, 'instruction_type': 'b', 'feedback_type': ['hp', 'hn', 'fp'], 'tasks_to_complete': ['microwave', 'kettle', 'light switch', 'slide cabinet'], 'visual': False, 'discount': 0.99, 'inference_expert_action': False, 'lang_multistep_merger': {'_target_': 'llfbench.envs.kitchen.multistep_merger.franka_kitchen_merger.KitchenMultistepMerger'}}, 'dataset': {'_target_': 'llmbc.dataset.kitchen_lowdim_dataset.KitchenLowdimDataset', 'data_path': 'datasets/FrankaKitchen-v1-general-freward.pt', 'data_path2': 'datasets/FrankaKitchen-v1-freward.pt', 'horizon': 16, 'pad_before': 1, 'pad_after': 7, 'obs_eef_target': True, 'use_manual_normalizer': False, 'val_ratio': 0.05, 'dummy_normalizer': False}, 'instructor': {'_target_': 'llmbc.translator.instructor.kitchen_instructor.FrankaKitchen_v1_instructor.FrankaKitchenV1Instructor'}}}
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_init.py:init():669] starting backend
+2026-10-28 13:54:22,038 INFO    MainThread:3216302 [wandb_init.py:init():673] sending inform_init request
+2026-10-28 13:54:22,039 INFO    MainThread:3216302 [backend.py:_multiprocessing_setup():104] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
+2026-10-28 13:54:22,039 INFO    MainThread:3216302 [wandb_init.py:init():686] backend started and connected
+2026-10-28 13:54:22,044 INFO    MainThread:3216302 [wandb_init.py:init():781] updated telemetry
+2026-10-28 13:54:22,068 INFO    MainThread:3216302 [wandb_init.py:init():814] communicating run to backend with 90.0 second timeout
+2026-10-28 13:54:23,679 INFO    MainThread:3216302 [wandb_init.py:init():867] starting run threads in backend
+2026-10-28 13:54:23,789 INFO    MainThread:3216302 [wandb_run.py:_console_start():2451] atexit reg
+2026-10-28 13:54:23,789 INFO    MainThread:3216302 [wandb_run.py:_redirect():2299] redirect: wrap_raw
+2026-10-28 13:54:23,789 INFO    MainThread:3216302 [wandb_run.py:_redirect():2364] Wrapping output streams.
+2026-10-28 13:54:23,790 INFO    MainThread:3216302 [wandb_run.py:_redirect():2389] Redirects installed.
+2026-10-28 13:54:23,792 INFO    MainThread:3216302 [wandb_init.py:init():911] run started, returning control to user process
+2026-10-28 13:54:23,792 INFO    MainThread:3216302 [wandb_run.py:_config_callback():1389] config_cb None None {'output_dir': '/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1'}
+2026-10-28 15:03:33,214 WARNING MsgRouterThr:3216302 [router.py:message_loop():75] message_loop has been closed

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/run-20261028_135422-ux76l2ws/run-ux76l2ws.wandb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:790fd0d28500657d7b6294cb0320e8f4e957a65f610f46dfec18bae128615cef
+size 201695703

2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/wandb/wandb-resume.json

@@ -0,0 +1 @@
+{"run_id": "ux76l2ws"}