CHYang25 commited on
Commit
6efed04
·
verified ·
1 Parent(s): 805cd41

Upload folder using huggingface_hub

Browse files
Files changed (43) hide show
  1. .gitattributes +1 -0
  2. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/.hydra/config.yaml +115 -0
  3. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/.hydra/hydra.yaml +154 -0
  4. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/.hydra/overrides.yaml +1 -0
  5. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/config.json +35 -0
  6. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/generation_config.json +6 -0
  7. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/mlp_projector.bin +3 -0
  8. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/model.safetensors +3 -0
  9. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/optimizer.pt +3 -0
  10. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/rng_state.pth +3 -0
  11. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/scheduler.pt +3 -0
  12. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/special_tokens_map.json +24 -0
  13. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/tokenizer.json +0 -0
  14. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/tokenizer_config.json +215 -0
  15. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/trainer_state.json +0 -0
  16. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/training_args.bin +3 -0
  17. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/config.json +35 -0
  18. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/generation_config.json +6 -0
  19. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/mlp_projector.bin +3 -0
  20. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/model.safetensors +3 -0
  21. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/optimizer.pt +3 -0
  22. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/rng_state.pth +3 -0
  23. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/scheduler.pt +3 -0
  24. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/special_tokens_map.json +24 -0
  25. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/tokenizer.json +0 -0
  26. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/tokenizer_config.json +215 -0
  27. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/trainer_state.json +0 -0
  28. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/training_args.bin +3 -0
  29. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/config.json +35 -0
  30. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/generation_config.json +6 -0
  31. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/model.safetensors +3 -0
  32. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/normalizer.pt +3 -0
  33. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/train.log +11 -0
  34. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/debug-internal.log +17 -0
  35. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/debug.log +35 -0
  36. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/files/config.yaml +708 -0
  37. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/files/output.log +0 -0
  38. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/files/wandb-metadata.json +55 -0
  39. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/files/wandb-summary.json +1 -0
  40. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/logs/debug-core.log +16 -0
  41. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/logs/debug-internal.log +17 -0
  42. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/logs/debug.log +35 -0
  43. 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/run-d0nbcloa.wandb +3 -0
.gitattributes CHANGED
@@ -310,3 +310,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
310
  2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/logs.json.txt filter=lfs diff=lfs merge=lfs -text
311
  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
312
  2026.11.07/11.47.11_train_llm_lowdim_box-close-v2/wandb/run-20261107_114716-n4c76l8i/run-n4c76l8i.wandb filter=lfs diff=lfs merge=lfs -text
 
 
310
  2026.10.28/13.54.19_train_rbc_reward_model_FrankaKitchen-v1/logs.json.txt filter=lfs diff=lfs merge=lfs -text
311
  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
312
  2026.11.07/11.47.11_train_llm_lowdim_box-close-v2/wandb/run-20261107_114716-n4c76l8i/run-n4c76l8i.wandb filter=lfs diff=lfs merge=lfs -text
313
+ 2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/run-d0nbcloa.wandb filter=lfs diff=lfs merge=lfs -text
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/.hydra/config.yaml ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: train_llm_lowdim
2
+ _target_: llmbc.workspace.train_llm_workspace.TrainLLMWorkspace
3
+ obs_dim: ${task.obs_dim}
4
+ action_dim: ${task.action_dim}
5
+ horizon: 1
6
+ n_obs_steps: 1
7
+ n_action_steps: 1
8
+ task_name: ${task.name}
9
+ exp_name: train llm
10
+ model_name: ${llm.name}
11
+ use_quantization: ${llm.use_quantization}
12
+ lora_config: ${llm.lora_config}
13
+ dataset:
14
+ test_data_ratio: 0.01
15
+ debug: false
16
+ training:
17
+ seed: 42
18
+ per_device_train_batch_size: 128
19
+ per_device_eval_batch_size: 128
20
+ gradient_accumulation_steps: 1
21
+ optim: paged_adamw_32bit
22
+ num_train_epochs: 10
23
+ eval_strategy: steps
24
+ logging_steps: 1
25
+ warmup_steps: 100
26
+ logging_strategy: steps
27
+ learning_rate: 0.0001
28
+ fp16: false
29
+ bf16: true
30
+ tf32: true
31
+ group_by_length: true
32
+ report_to: wandb
33
+ save_steps: 5000
34
+ eval_steps: 10
35
+ use_joint_mlp_projector: ${llm.use_joint_mlp_projector}
36
+ joint_obs_action_mlp_lr: 0.0001
37
+ trainer:
38
+ obs_dim: ${obs_dim}
39
+ action_dim: ${action_dim}
40
+ use_joint_mlp_projector: ${llm.use_joint_mlp_projector}
41
+ max_seq_length: ${llm.max_length}
42
+ dataset_text_field: text
43
+ packing: false
44
+ logging:
45
+ project: llm_module_finetuning
46
+ resume: true
47
+ mode: online
48
+ name: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
49
+ tags:
50
+ - ${name}
51
+ - ${task_name}
52
+ - ${exp_name}
53
+ id: null
54
+ group: null
55
+ multi_run:
56
+ run_dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
57
+ wandb_name_base: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
58
+ task:
59
+ name: box-close-v2
60
+ obs_dim: 9
61
+ action_dim: 4
62
+ env_runner:
63
+ _target_: llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner
64
+ env_name: llf-metaworld-box-close-v2
65
+ n_train: 10
66
+ n_test: 50
67
+ n_envs: 10
68
+ max_steps: 30
69
+ n_obs_steps: ${n_obs_steps}
70
+ n_action_steps: ${n_action_steps}
71
+ instruction_type: b
72
+ feedback_type:
73
+ - hp
74
+ - hn
75
+ - fp
76
+ visual: false
77
+ discount: 0.9
78
+ dataset:
79
+ _target_: llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset
80
+ data_path: datasets/box-close-v2-general.pt
81
+ data_path2: datasets/box-close-v2.pt
82
+ horizon: ${horizon}
83
+ pad_before: ${eval:'${n_obs_steps}-1'}
84
+ pad_after: ${eval:'${n_action_steps}-1'}
85
+ obs_eef_target: true
86
+ use_manual_normalizer: false
87
+ val_ratio: 0.05
88
+ dummy_normalizer: false
89
+ instructor:
90
+ _target_: llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor
91
+ llm:
92
+ name: EleutherAI/pythia-70m-deduped
93
+ model_name: pythia-70m-deduped
94
+ config_target: llmbc.model.llm.gpt_neox_lowdim_model.LowdimGPTNeoXConfig
95
+ causal_lm_target: llmbc.model.llm.gpt_neox_lowdim_model.LowdimGPTNeoXForCausalLM
96
+ use_quantization: false
97
+ use_joint_mlp_projector: true
98
+ llm_mode: mlp-finetuned
99
+ finetune_mode: orig
100
+ checkpoint: data/outputs/2026.11.07/01.39.14_train_mlp_projector_box-close-v2/checkpoints/latest.ckpt
101
+ max_length: 100
102
+ lora_config:
103
+ r: 32
104
+ lora_alpha: 64
105
+ lora_dropout: 0.05
106
+ bias: none
107
+ task_type: CAUSAL_LM
108
+ prompter:
109
+ _target_: llmbc.translator.prompter.pythia_prompter.PythiaPrompter
110
+ use_joint_mlp_projector: true
111
+ hydra:
112
+ job:
113
+ override_dirname: ${model_name}
114
+ run:
115
+ dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${model_name}
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/.hydra/hydra.yaml ADDED
@@ -0,0 +1,154 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
117
+ chdir: null
118
+ override_dirname: ''
119
+ id: ???
120
+ num: ???
121
+ config_name: llmdp_llm_box-close-v2_pythia-70m-deduped.yaml
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/config/llm_backbone
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.11.07/09.50.49_train_llm_lowdim_box-close-v2
144
+ choices:
145
+ hydra/env: default
146
+ hydra/callbacks: null
147
+ hydra/job_logging: default
148
+ hydra/hydra_logging: default
149
+ hydra/hydra_help: default
150
+ hydra/help: default
151
+ hydra/sweeper: basic
152
+ hydra/launcher: basic
153
+ hydra/output: default
154
+ verbose: false
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/.hydra/overrides.yaml ADDED
@@ -0,0 +1 @@
 
 
1
+ []
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "EleutherAI/pythia-70m-deduped",
3
+ "action_dim": 4,
4
+ "architectures": [
5
+ "LowdimGPTNeoXForCausalLM"
6
+ ],
7
+ "attention_bias": true,
8
+ "attention_dropout": 0.0,
9
+ "bos_token_id": 0,
10
+ "classifier_dropout": 0.1,
11
+ "eos_token_id": 0,
12
+ "hidden_act": "gelu",
13
+ "hidden_dropout": 0.0,
14
+ "hidden_size": 512,
15
+ "initializer_range": 0.02,
16
+ "intermediate_size": 2048,
17
+ "layer_norm_eps": 1e-05,
18
+ "max_position_embeddings": 2048,
19
+ "model_type": "gpt_neox_lowdim",
20
+ "num_attention_heads": 8,
21
+ "num_hidden_layers": 6,
22
+ "obs_dim": 9,
23
+ "partial_rotary_factor": 0.25,
24
+ "rope_scaling": null,
25
+ "rope_theta": 10000,
26
+ "rotary_emb_base": 10000,
27
+ "rotary_pct": 0.25,
28
+ "tie_word_embeddings": false,
29
+ "torch_dtype": "float32",
30
+ "transformers_version": "4.47.1",
31
+ "use_cache": false,
32
+ "use_joint_mlp_projector": true,
33
+ "use_parallel_residual": true,
34
+ "vocab_size": 50304
35
+ }
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "eos_token_id": 0,
5
+ "transformers_version": "4.47.1"
6
+ }
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/mlp_projector.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ae112871608ab2395a9cca01436766ee032a3c57c11b684b78efe860912af5c8
3
+ size 1081472
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:87236be1a8b3a8cabda521a7af0bbf30c1c0c04c7f9e5965acb224a4862243f4
3
+ size 282794968
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:acdc546fdd42cc6c0bf231fca087107091a072bda0ee20bac6a1218e4615d30d
3
+ size 565619962
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1229178cd3fc53fbbeb2ace48b61cc7f5f11f85a2aa90404bd54a81f04d35cf2
3
+ size 14244
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0afe93d00a35552cbf53efccf9952fb89964795a1fb9059b72d318471d0ae56
3
+ size 1064
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/special_tokens_map.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<|endoftext|>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "<|endoftext|>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": "<|endoftext|>",
17
+ "unk_token": {
18
+ "content": "<|endoftext|>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ }
24
+ }
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/tokenizer_config.json ADDED
@@ -0,0 +1,215 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": false,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": false,
5
+ "added_tokens_decoder": {
6
+ "0": {
7
+ "content": "<|endoftext|>",
8
+ "lstrip": false,
9
+ "normalized": false,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
+ "1": {
15
+ "content": "<|padding|>",
16
+ "lstrip": false,
17
+ "normalized": false,
18
+ "rstrip": false,
19
+ "single_word": false,
20
+ "special": true
21
+ },
22
+ "50254": {
23
+ "content": " ",
24
+ "lstrip": false,
25
+ "normalized": true,
26
+ "rstrip": false,
27
+ "single_word": false,
28
+ "special": false
29
+ },
30
+ "50255": {
31
+ "content": " ",
32
+ "lstrip": false,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false,
36
+ "special": false
37
+ },
38
+ "50256": {
39
+ "content": " ",
40
+ "lstrip": false,
41
+ "normalized": true,
42
+ "rstrip": false,
43
+ "single_word": false,
44
+ "special": false
45
+ },
46
+ "50257": {
47
+ "content": " ",
48
+ "lstrip": false,
49
+ "normalized": true,
50
+ "rstrip": false,
51
+ "single_word": false,
52
+ "special": false
53
+ },
54
+ "50258": {
55
+ "content": " ",
56
+ "lstrip": false,
57
+ "normalized": true,
58
+ "rstrip": false,
59
+ "single_word": false,
60
+ "special": false
61
+ },
62
+ "50259": {
63
+ "content": " ",
64
+ "lstrip": false,
65
+ "normalized": true,
66
+ "rstrip": false,
67
+ "single_word": false,
68
+ "special": false
69
+ },
70
+ "50260": {
71
+ "content": " ",
72
+ "lstrip": false,
73
+ "normalized": true,
74
+ "rstrip": false,
75
+ "single_word": false,
76
+ "special": false
77
+ },
78
+ "50261": {
79
+ "content": " ",
80
+ "lstrip": false,
81
+ "normalized": true,
82
+ "rstrip": false,
83
+ "single_word": false,
84
+ "special": false
85
+ },
86
+ "50262": {
87
+ "content": " ",
88
+ "lstrip": false,
89
+ "normalized": true,
90
+ "rstrip": false,
91
+ "single_word": false,
92
+ "special": false
93
+ },
94
+ "50263": {
95
+ "content": " ",
96
+ "lstrip": false,
97
+ "normalized": true,
98
+ "rstrip": false,
99
+ "single_word": false,
100
+ "special": false
101
+ },
102
+ "50264": {
103
+ "content": " ",
104
+ "lstrip": false,
105
+ "normalized": true,
106
+ "rstrip": false,
107
+ "single_word": false,
108
+ "special": false
109
+ },
110
+ "50265": {
111
+ "content": " ",
112
+ "lstrip": false,
113
+ "normalized": true,
114
+ "rstrip": false,
115
+ "single_word": false,
116
+ "special": false
117
+ },
118
+ "50266": {
119
+ "content": " ",
120
+ "lstrip": false,
121
+ "normalized": true,
122
+ "rstrip": false,
123
+ "single_word": false,
124
+ "special": false
125
+ },
126
+ "50267": {
127
+ "content": " ",
128
+ "lstrip": false,
129
+ "normalized": true,
130
+ "rstrip": false,
131
+ "single_word": false,
132
+ "special": false
133
+ },
134
+ "50268": {
135
+ "content": " ",
136
+ "lstrip": false,
137
+ "normalized": true,
138
+ "rstrip": false,
139
+ "single_word": false,
140
+ "special": false
141
+ },
142
+ "50269": {
143
+ "content": " ",
144
+ "lstrip": false,
145
+ "normalized": true,
146
+ "rstrip": false,
147
+ "single_word": false,
148
+ "special": false
149
+ },
150
+ "50270": {
151
+ "content": " ",
152
+ "lstrip": false,
153
+ "normalized": true,
154
+ "rstrip": false,
155
+ "single_word": false,
156
+ "special": false
157
+ },
158
+ "50271": {
159
+ "content": " ",
160
+ "lstrip": false,
161
+ "normalized": true,
162
+ "rstrip": false,
163
+ "single_word": false,
164
+ "special": false
165
+ },
166
+ "50272": {
167
+ "content": " ",
168
+ "lstrip": false,
169
+ "normalized": true,
170
+ "rstrip": false,
171
+ "single_word": false,
172
+ "special": false
173
+ },
174
+ "50273": {
175
+ "content": " ",
176
+ "lstrip": false,
177
+ "normalized": true,
178
+ "rstrip": false,
179
+ "single_word": false,
180
+ "special": false
181
+ },
182
+ "50274": {
183
+ "content": " ",
184
+ "lstrip": false,
185
+ "normalized": true,
186
+ "rstrip": false,
187
+ "single_word": false,
188
+ "special": false
189
+ },
190
+ "50275": {
191
+ "content": " ",
192
+ "lstrip": false,
193
+ "normalized": true,
194
+ "rstrip": false,
195
+ "single_word": false,
196
+ "special": false
197
+ },
198
+ "50276": {
199
+ "content": " ",
200
+ "lstrip": false,
201
+ "normalized": true,
202
+ "rstrip": false,
203
+ "single_word": false,
204
+ "special": false
205
+ }
206
+ },
207
+ "bos_token": "<|endoftext|>",
208
+ "clean_up_tokenization_spaces": false,
209
+ "eos_token": "<|endoftext|>",
210
+ "extra_special_tokens": {},
211
+ "model_max_length": 1000000000000000019884624838656,
212
+ "pad_token": "<|endoftext|>",
213
+ "tokenizer_class": "GPTNeoXTokenizer",
214
+ "unk_token": "<|endoftext|>"
215
+ }
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a4d3d93aad5886ee976e89239e998b321c2d52aba15653244e69bb1f958d8c98
3
+ size 5944
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "EleutherAI/pythia-70m-deduped",
3
+ "action_dim": 4,
4
+ "architectures": [
5
+ "LowdimGPTNeoXForCausalLM"
6
+ ],
7
+ "attention_bias": true,
8
+ "attention_dropout": 0.0,
9
+ "bos_token_id": 0,
10
+ "classifier_dropout": 0.1,
11
+ "eos_token_id": 0,
12
+ "hidden_act": "gelu",
13
+ "hidden_dropout": 0.0,
14
+ "hidden_size": 512,
15
+ "initializer_range": 0.02,
16
+ "intermediate_size": 2048,
17
+ "layer_norm_eps": 1e-05,
18
+ "max_position_embeddings": 2048,
19
+ "model_type": "gpt_neox_lowdim",
20
+ "num_attention_heads": 8,
21
+ "num_hidden_layers": 6,
22
+ "obs_dim": 9,
23
+ "partial_rotary_factor": 0.25,
24
+ "rope_scaling": null,
25
+ "rope_theta": 10000,
26
+ "rotary_emb_base": 10000,
27
+ "rotary_pct": 0.25,
28
+ "tie_word_embeddings": false,
29
+ "torch_dtype": "float32",
30
+ "transformers_version": "4.47.1",
31
+ "use_cache": false,
32
+ "use_joint_mlp_projector": true,
33
+ "use_parallel_residual": true,
34
+ "vocab_size": 50304
35
+ }
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "eos_token_id": 0,
5
+ "transformers_version": "4.47.1"
6
+ }
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/mlp_projector.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d1c7115b789a51a0e298e93e4b51aa0353d0cf990b006aecaec7ec2a07cfb12d
3
+ size 1081472
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:459bf1cf740a697de4472ec2d59372576f5338823ca41c64aec8ff3136c9f5ae
3
+ size 282794968
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cdc5632b7920e6d6a4f601fc3b8b5c73a8a77b8a261e67297814b70c3de824f0
3
+ size 565619962
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e3efd5afaf0c28c80d40a9eb842707cd94a44c7a47a687677b2160e97b49d52d
3
+ size 14244
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c541a1306e2a80c04885085b048be43bff734699f71340ec300af5de1ee8ab68
3
+ size 1064
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/special_tokens_map.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<|endoftext|>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "<|endoftext|>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": "<|endoftext|>",
17
+ "unk_token": {
18
+ "content": "<|endoftext|>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ }
24
+ }
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/tokenizer_config.json ADDED
@@ -0,0 +1,215 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": false,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": false,
5
+ "added_tokens_decoder": {
6
+ "0": {
7
+ "content": "<|endoftext|>",
8
+ "lstrip": false,
9
+ "normalized": false,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
+ "1": {
15
+ "content": "<|padding|>",
16
+ "lstrip": false,
17
+ "normalized": false,
18
+ "rstrip": false,
19
+ "single_word": false,
20
+ "special": true
21
+ },
22
+ "50254": {
23
+ "content": " ",
24
+ "lstrip": false,
25
+ "normalized": true,
26
+ "rstrip": false,
27
+ "single_word": false,
28
+ "special": false
29
+ },
30
+ "50255": {
31
+ "content": " ",
32
+ "lstrip": false,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false,
36
+ "special": false
37
+ },
38
+ "50256": {
39
+ "content": " ",
40
+ "lstrip": false,
41
+ "normalized": true,
42
+ "rstrip": false,
43
+ "single_word": false,
44
+ "special": false
45
+ },
46
+ "50257": {
47
+ "content": " ",
48
+ "lstrip": false,
49
+ "normalized": true,
50
+ "rstrip": false,
51
+ "single_word": false,
52
+ "special": false
53
+ },
54
+ "50258": {
55
+ "content": " ",
56
+ "lstrip": false,
57
+ "normalized": true,
58
+ "rstrip": false,
59
+ "single_word": false,
60
+ "special": false
61
+ },
62
+ "50259": {
63
+ "content": " ",
64
+ "lstrip": false,
65
+ "normalized": true,
66
+ "rstrip": false,
67
+ "single_word": false,
68
+ "special": false
69
+ },
70
+ "50260": {
71
+ "content": " ",
72
+ "lstrip": false,
73
+ "normalized": true,
74
+ "rstrip": false,
75
+ "single_word": false,
76
+ "special": false
77
+ },
78
+ "50261": {
79
+ "content": " ",
80
+ "lstrip": false,
81
+ "normalized": true,
82
+ "rstrip": false,
83
+ "single_word": false,
84
+ "special": false
85
+ },
86
+ "50262": {
87
+ "content": " ",
88
+ "lstrip": false,
89
+ "normalized": true,
90
+ "rstrip": false,
91
+ "single_word": false,
92
+ "special": false
93
+ },
94
+ "50263": {
95
+ "content": " ",
96
+ "lstrip": false,
97
+ "normalized": true,
98
+ "rstrip": false,
99
+ "single_word": false,
100
+ "special": false
101
+ },
102
+ "50264": {
103
+ "content": " ",
104
+ "lstrip": false,
105
+ "normalized": true,
106
+ "rstrip": false,
107
+ "single_word": false,
108
+ "special": false
109
+ },
110
+ "50265": {
111
+ "content": " ",
112
+ "lstrip": false,
113
+ "normalized": true,
114
+ "rstrip": false,
115
+ "single_word": false,
116
+ "special": false
117
+ },
118
+ "50266": {
119
+ "content": " ",
120
+ "lstrip": false,
121
+ "normalized": true,
122
+ "rstrip": false,
123
+ "single_word": false,
124
+ "special": false
125
+ },
126
+ "50267": {
127
+ "content": " ",
128
+ "lstrip": false,
129
+ "normalized": true,
130
+ "rstrip": false,
131
+ "single_word": false,
132
+ "special": false
133
+ },
134
+ "50268": {
135
+ "content": " ",
136
+ "lstrip": false,
137
+ "normalized": true,
138
+ "rstrip": false,
139
+ "single_word": false,
140
+ "special": false
141
+ },
142
+ "50269": {
143
+ "content": " ",
144
+ "lstrip": false,
145
+ "normalized": true,
146
+ "rstrip": false,
147
+ "single_word": false,
148
+ "special": false
149
+ },
150
+ "50270": {
151
+ "content": " ",
152
+ "lstrip": false,
153
+ "normalized": true,
154
+ "rstrip": false,
155
+ "single_word": false,
156
+ "special": false
157
+ },
158
+ "50271": {
159
+ "content": " ",
160
+ "lstrip": false,
161
+ "normalized": true,
162
+ "rstrip": false,
163
+ "single_word": false,
164
+ "special": false
165
+ },
166
+ "50272": {
167
+ "content": " ",
168
+ "lstrip": false,
169
+ "normalized": true,
170
+ "rstrip": false,
171
+ "single_word": false,
172
+ "special": false
173
+ },
174
+ "50273": {
175
+ "content": " ",
176
+ "lstrip": false,
177
+ "normalized": true,
178
+ "rstrip": false,
179
+ "single_word": false,
180
+ "special": false
181
+ },
182
+ "50274": {
183
+ "content": " ",
184
+ "lstrip": false,
185
+ "normalized": true,
186
+ "rstrip": false,
187
+ "single_word": false,
188
+ "special": false
189
+ },
190
+ "50275": {
191
+ "content": " ",
192
+ "lstrip": false,
193
+ "normalized": true,
194
+ "rstrip": false,
195
+ "single_word": false,
196
+ "special": false
197
+ },
198
+ "50276": {
199
+ "content": " ",
200
+ "lstrip": false,
201
+ "normalized": true,
202
+ "rstrip": false,
203
+ "single_word": false,
204
+ "special": false
205
+ }
206
+ },
207
+ "bos_token": "<|endoftext|>",
208
+ "clean_up_tokenization_spaces": false,
209
+ "eos_token": "<|endoftext|>",
210
+ "extra_special_tokens": {},
211
+ "model_max_length": 1000000000000000019884624838656,
212
+ "pad_token": "<|endoftext|>",
213
+ "tokenizer_class": "GPTNeoXTokenizer",
214
+ "unk_token": "<|endoftext|>"
215
+ }
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/checkpoint-5890/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a4d3d93aad5886ee976e89239e998b321c2d52aba15653244e69bb1f958d8c98
3
+ size 5944
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "EleutherAI/pythia-70m-deduped",
3
+ "action_dim": 4,
4
+ "architectures": [
5
+ "LowdimGPTNeoXForCausalLM"
6
+ ],
7
+ "attention_bias": true,
8
+ "attention_dropout": 0.0,
9
+ "bos_token_id": 0,
10
+ "classifier_dropout": 0.1,
11
+ "eos_token_id": 0,
12
+ "hidden_act": "gelu",
13
+ "hidden_dropout": 0.0,
14
+ "hidden_size": 512,
15
+ "initializer_range": 0.02,
16
+ "intermediate_size": 2048,
17
+ "layer_norm_eps": 1e-05,
18
+ "max_position_embeddings": 2048,
19
+ "model_type": "gpt_neox_lowdim",
20
+ "num_attention_heads": 8,
21
+ "num_hidden_layers": 6,
22
+ "obs_dim": 9,
23
+ "partial_rotary_factor": 0.25,
24
+ "rope_scaling": null,
25
+ "rope_theta": 10000,
26
+ "rotary_emb_base": 10000,
27
+ "rotary_pct": 0.25,
28
+ "tie_word_embeddings": false,
29
+ "torch_dtype": "float32",
30
+ "transformers_version": "4.47.1",
31
+ "use_cache": true,
32
+ "use_joint_mlp_projector": true,
33
+ "use_parallel_residual": true,
34
+ "vocab_size": 50304
35
+ }
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "eos_token_id": 0,
5
+ "transformers_version": "4.47.1"
6
+ }
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:459bf1cf740a697de4472ec2d59372576f5338823ca41c64aec8ff3136c9f5ae
3
+ size 282794968
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/normalizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a1cb28cf65a776a736481b2a05aa3c4cf0d796a0ec84eaa0b9e4aa06bdc79aa
3
+ size 4514
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/train.log ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [2026-11-07 09:50:49,853][numexpr.utils][INFO] - Note: NumExpr detected 24 cores but "NUMEXPR_MAX_THREADS" not set, so enforcing safe limit of 16.
2
+ [2026-11-07 09:50:49,853][numexpr.utils][INFO] - NumExpr defaulting to 16 threads.
3
+ [2026-11-07 09:50:52,057][datasets][INFO] - PyTorch version 2.2.2 available.
4
+ [2026-11-07 09:50:52,057][datasets][INFO] - TensorFlow version 2.15.1 available.
5
+ [2026-11-07 09:50:52,058][datasets][INFO] - JAX version 0.4.30 available.
6
+ [2026-11-07 09:51:06,919][datasets.arrow_dataset][WARNING] - Setting TOKENIZERS_PARALLELISM=false for forked processes.
7
+ [2026-11-07 09:51:18,882][datasets.arrow_dataset][WARNING] - Setting TOKENIZERS_PARALLELISM=false for forked processes.
8
+ [2026-11-07 09:51:24,192][root][INFO] - gcc -pthread -B /home/chyang/miniconda3/envs/llm-bc/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -O2 -Wall -fPIC -O2 -isystem /home/chyang/miniconda3/envs/llm-bc/include -I/home/chyang/miniconda3/envs/llm-bc/include -fPIC -O2 -isystem /home/chyang/miniconda3/envs/llm-bc/include -fPIC -c /tmp/tmps1mulba1/test.c -o /tmp/tmps1mulba1/test.o
9
+ [2026-11-07 09:51:24,237][root][INFO] - gcc -pthread -B /home/chyang/miniconda3/envs/llm-bc/compiler_compat /tmp/tmps1mulba1/test.o -laio -o /tmp/tmps1mulba1/a.out
10
+ [2026-11-07 09:51:25,003][root][INFO] - gcc -pthread -B /home/chyang/miniconda3/envs/llm-bc/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -O2 -Wall -fPIC -O2 -isystem /home/chyang/miniconda3/envs/llm-bc/include -I/home/chyang/miniconda3/envs/llm-bc/include -fPIC -O2 -isystem /home/chyang/miniconda3/envs/llm-bc/include -fPIC -c /tmp/tmpejqhnraj/test.c -o /tmp/tmpejqhnraj/test.o
11
+ [2026-11-07 09:51:25,047][root][INFO] - gcc -pthread -B /home/chyang/miniconda3/envs/llm-bc/compiler_compat /tmp/tmpejqhnraj/test.o -L/home/chyang/miniconda3/envs/llm-bc -L/home/chyang/miniconda3/envs/llm-bc/lib64 -lcufile -o /tmp/tmpejqhnraj/a.out
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/debug-internal.log ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-11-07T09:50:53.592038827+08:00","level":"INFO","msg":"using version","core version":"0.18.6"}
2
+ {"time":"2026-11-07T09:50:53.592065195+08:00","level":"INFO","msg":"created symlink","path":"/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/logs/debug-core.log"}
3
+ {"time":"2026-11-07T09:50:53.699007609+08:00","level":"INFO","msg":"created new stream","id":"d0nbcloa"}
4
+ {"time":"2026-11-07T09:50:53.69905307+08:00","level":"INFO","msg":"stream: started","id":"d0nbcloa"}
5
+ {"time":"2026-11-07T09:50:53.699072256+08:00","level":"INFO","msg":"sender: started","stream_id":"d0nbcloa"}
6
+ {"time":"2026-11-07T09:50:53.699068072+08:00","level":"INFO","msg":"handler: started","stream_id":{"value":"d0nbcloa"}}
7
+ {"time":"2026-11-07T09:50:53.69908682+08:00","level":"INFO","msg":"writer: Do: started","stream_id":{"value":"d0nbcloa"}}
8
+ {"time":"2026-11-07T09:50:54.426318415+08:00","level":"INFO","msg":"Starting system monitor"}
9
+ {"time":"2026-11-07T10:26:08.400121893+08:00","level":"INFO","msg":"Stopping system monitor"}
10
+ {"time":"2026-11-07T10:26:08.400943972+08:00","level":"INFO","msg":"Stopped system monitor"}
11
+ {"time":"2026-11-07T10:26:09.400990261+08:00","level":"INFO","msg":"handler: operation stats","stats":{"operations":[{"desc":"saving job artifact","runtime_seconds":0.331823839},{"desc":"uploading config.yaml","runtime_seconds":0.089839805,"progress":"14.4KB/14.4KB"},{"desc":"uploading output.log","runtime_seconds":0.089825287,"progress":"990.8KB/990.8KB"}],"total_operations":3}}
12
+ {"time":"2026-11-07T10:26:11.273318383+08:00","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
13
+ {"time":"2026-11-07T10:26:13.099047371+08:00","level":"INFO","msg":"stream: closing","id":"d0nbcloa"}
14
+ {"time":"2026-11-07T10:26:13.099087573+08:00","level":"INFO","msg":"handler: closed","stream_id":{"value":"d0nbcloa"}}
15
+ {"time":"2026-11-07T10:26:13.099122038+08:00","level":"INFO","msg":"writer: Close: closed","stream_id":{"value":"d0nbcloa"}}
16
+ {"time":"2026-11-07T10:26:13.099185666+08:00","level":"INFO","msg":"sender: closed","stream_id":"d0nbcloa"}
17
+ {"time":"2026-11-07T10:26:13.099239802+08:00","level":"INFO","msg":"stream: closed","id":"d0nbcloa"}
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/debug.log ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_setup.py:_flush():79] Current SDK version is 0.18.6
2
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_setup.py:_flush():79] Configure stats pid to 147051
3
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_setup.py:_flush():79] Loading settings from /home/chyang/.config/wandb/settings
4
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_setup.py:_flush():79] Loading settings from /tmp2/chyang/workspace/LLM-BC/wandb/settings
5
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_setup.py:_flush():79] Loading settings from environment variables: {}
6
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_setup.py:_flush():79] Applying setup settings: {'mode': 'online', '_disable_service': None}
7
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [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'}
8
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_setup.py:_flush():79] Applying login settings: {}
9
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_init.py:_log_setup():533] Logging user logs to /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/logs/debug.log
10
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_init.py:_log_setup():534] Logging internal logs to /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/logs/debug-internal.log
11
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_init.py:init():619] calling init triggers
12
+ 2026-11-07 09:50:53,588 INFO MainThread:147051 [wandb_init.py:init():626] wandb.init called with sweep_config: {}
13
+ config: {'name': 'train_llm_lowdim', '_target_': 'llmbc.workspace.train_llm_workspace.TrainLLMWorkspace', 'obs_dim': 9, 'action_dim': 4, 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1, 'task_name': 'box-close-v2', 'exp_name': 'train llm', 'model_name': 'EleutherAI/pythia-70m-deduped', 'use_quantization': False, 'lora_config': {'r': 32, 'lora_alpha': 64, 'lora_dropout': 0.05, 'bias': 'none', 'task_type': 'CAUSAL_LM'}, 'dataset': {'test_data_ratio': 0.01}, 'debug': False, 'training': {'seed': 42, 'per_device_train_batch_size': 128, 'per_device_eval_batch_size': 128, 'gradient_accumulation_steps': 1, 'optim': 'paged_adamw_32bit', 'num_train_epochs': 10, 'eval_strategy': 'steps', 'logging_steps': 1, 'warmup_steps': 100, 'logging_strategy': 'steps', 'learning_rate': 0.0001, 'fp16': False, 'bf16': True, 'tf32': True, 'group_by_length': True, 'report_to': 'wandb', 'save_steps': 5000, 'eval_steps': 10, 'use_joint_mlp_projector': True, 'joint_obs_action_mlp_lr': 0.0001}, 'trainer': {'obs_dim': 9, 'action_dim': 4, 'use_joint_mlp_projector': True, 'max_seq_length': 100, 'dataset_text_field': 'text', 'packing': False}, 'logging': {'project': 'llm_module_finetuning', 'resume': True, 'mode': 'online', 'name': '2026.11.07-09.50.49_train_llm_lowdim_box-close-v2', 'tags': ['train_llm_lowdim', 'box-close-v2', 'train llm'], 'id': None, 'group': None}, 'multi_run': {'run_dir': 'data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2', 'wandb_name_base': '2026.11.07-09.50.49_train_llm_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-general.pt', 'data_path2': 'datasets/box-close-v2.pt', 'horizon': 1, 'pad_before': 0, 'pad_after': 0, 'obs_eef_target': True, 'use_manual_normalizer': False, 'val_ratio': 0.05, 'dummy_normalizer': False}, 'instructor': {'_target_': 'llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor'}}, 'llm': {'name': 'EleutherAI/pythia-70m-deduped', 'model_name': 'pythia-70m-deduped', 'config_target': 'llmbc.model.llm.gpt_neox_lowdim_model.LowdimGPTNeoXConfig', 'causal_lm_target': 'llmbc.model.llm.gpt_neox_lowdim_model.LowdimGPTNeoXForCausalLM', 'use_quantization': False, 'use_joint_mlp_projector': True, 'llm_mode': 'mlp-finetuned', 'finetune_mode': 'orig', 'checkpoint': 'data/outputs/2026.11.07/01.39.14_train_mlp_projector_box-close-v2/checkpoints/latest.ckpt', '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.pythia_prompter.PythiaPrompter', 'use_joint_mlp_projector': True}, 'hydra': {'job': {'override_dirname': 'EleutherAI/pythia-70m-deduped'}, 'run': {'dir': 'data/outputs/2026.11.07/09.50.49_EleutherAI/pythia-70m-deduped'}}}}
14
+ 2026-11-07 09:50:53,588 INFO MainThread:147051 [wandb_init.py:init():669] starting backend
15
+ 2026-11-07 09:50:53,588 INFO MainThread:147051 [wandb_init.py:init():673] sending inform_init request
16
+ 2026-11-07 09:50:53,588 INFO MainThread:147051 [backend.py:_multiprocessing_setup():104] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
17
+ 2026-11-07 09:50:53,589 INFO MainThread:147051 [wandb_init.py:init():686] backend started and connected
18
+ 2026-11-07 09:50:53,593 INFO MainThread:147051 [wandb_init.py:init():781] updated telemetry
19
+ 2026-11-07 09:50:53,613 INFO MainThread:147051 [wandb_init.py:init():814] communicating run to backend with 90.0 second timeout
20
+ 2026-11-07 09:50:54,423 INFO MainThread:147051 [wandb_init.py:init():867] starting run threads in backend
21
+ 2026-11-07 09:50:54,530 INFO MainThread:147051 [wandb_run.py:_console_start():2451] atexit reg
22
+ 2026-11-07 09:50:54,530 INFO MainThread:147051 [wandb_run.py:_redirect():2299] redirect: wrap_raw
23
+ 2026-11-07 09:50:54,530 INFO MainThread:147051 [wandb_run.py:_redirect():2364] Wrapping output streams.
24
+ 2026-11-07 09:50:54,530 INFO MainThread:147051 [wandb_run.py:_redirect():2389] Redirects installed.
25
+ 2026-11-07 09:50:54,532 INFO MainThread:147051 [wandb_init.py:init():911] run started, returning control to user process
26
+ 2026-11-07 09:51:34,173 INFO MainThread:147051 [wandb_run.py:_config_callback():1389] config_cb None None {'obs_dim': 9, 'action_dim': 4, 'use_joint_mlp_projector': True, 'return_dict': True, 'output_hidden_states': False, 'output_attentions': False, 'torchscript': False, 'torch_dtype': 'float16', 'use_bfloat16': False, 'tf_legacy_loss': False, 'pruned_heads': {}, 'tie_word_embeddings': False, 'chunk_size_feed_forward': 0, 'is_encoder_decoder': False, 'is_decoder': False, 'cross_attention_hidden_size': None, 'add_cross_attention': False, 'tie_encoder_decoder': False, 'max_length': 20, 'min_length': 0, 'do_sample': False, 'early_stopping': False, 'num_beams': 1, 'num_beam_groups': 1, 'diversity_penalty': 0.0, 'temperature': 1.0, 'top_k': 50, 'top_p': 1.0, 'typical_p': 1.0, 'repetition_penalty': 1.0, 'length_penalty': 1.0, 'no_repeat_ngram_size': 0, 'encoder_no_repeat_ngram_size': 0, 'bad_words_ids': None, 'num_return_sequences': 1, 'output_scores': False, 'return_dict_in_generate': False, 'forced_bos_token_id': None, 'forced_eos_token_id': None, 'remove_invalid_values': False, 'exponential_decay_length_penalty': None, 'suppress_tokens': None, 'begin_suppress_tokens': None, 'architectures': ['GPTNeoXForCausalLM'], 'finetuning_task': None, 'id2label': {0: 'LABEL_0', 1: 'LABEL_1'}, 'label2id': {'LABEL_0': 0, 'LABEL_1': 1}, 'tokenizer_class': None, 'prefix': None, 'bos_token_id': 0, 'pad_token_id': None, 'eos_token_id': 0, 'sep_token_id': None, 'decoder_start_token_id': None, 'task_specific_params': None, 'problem_type': None, '_name_or_path': 'EleutherAI/pythia-70m-deduped', '_attn_implementation_autoset': True, 'transformers_version': '4.47.1', 'model_type': 'gpt_neox_lowdim', 'vocab_size': 50304, 'max_position_embeddings': 2048, 'hidden_size': 512, 'num_hidden_layers': 6, 'num_attention_heads': 8, 'intermediate_size': 2048, 'hidden_act': 'gelu', 'rotary_pct': 0.25, 'partial_rotary_factor': 0.25, 'rotary_emb_base': 10000, 'rope_theta': 10000, 'attention_dropout': 0.0, 'hidden_dropout': 0.0, 'classifier_dropout': 0.1, 'initializer_range': 0.02, 'layer_norm_eps': 1e-05, 'use_cache': False, 'use_parallel_residual': True, 'rope_scaling': None, 'attention_bias': True, 'output_dir': '/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2', 'overwrite_output_dir': False, 'do_train': False, 'do_eval': True, 'do_predict': False, 'eval_strategy': 'steps', 'prediction_loss_only': False, 'per_device_train_batch_size': 128, 'per_device_eval_batch_size': 128, 'per_gpu_train_batch_size': None, 'per_gpu_eval_batch_size': None, 'gradient_accumulation_steps': 1, 'eval_accumulation_steps': None, 'eval_delay': 0, 'torch_empty_cache_steps': None, 'learning_rate': 0.0001, 'weight_decay': 0.0, 'adam_beta1': 0.9, 'adam_beta2': 0.999, 'adam_epsilon': 1e-08, 'max_grad_norm': 1.0, 'num_train_epochs': 10, 'max_steps': -1, 'lr_scheduler_type': 'linear', 'lr_scheduler_kwargs': {}, 'warmup_ratio': 0.0, 'warmup_steps': 100, 'log_level': 'passive', 'log_level_replica': 'warning', 'log_on_each_node': True, 'logging_dir': '/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/runs/Nov07_09-51-23_A6000', 'logging_strategy': 'steps', 'logging_first_step': False, 'logging_steps': 1, 'logging_nan_inf_filter': True, 'save_strategy': 'steps', 'save_steps': 5000, 'save_total_limit': None, 'save_safetensors': True, 'save_on_each_node': False, 'save_only_model': False, 'restore_callback_states_from_checkpoint': False, 'no_cuda': False, 'use_cpu': False, 'use_mps_device': False, 'seed': 42, 'data_seed': None, 'jit_mode_eval': False, 'use_ipex': False, 'bf16': True, 'fp16': False, 'fp16_opt_level': 'O1', 'half_precision_backend': 'auto', 'bf16_full_eval': False, 'fp16_full_eval': False, 'tf32': True, 'local_rank': 0, 'ddp_backend': None, 'tpu_num_cores': None, 'tpu_metrics_debug': False, 'debug': [], 'dataloader_drop_last': False, 'eval_steps': 10, 'dataloader_num_workers': 0, 'dataloader_prefetch_factor': None, 'past_index': -1, 'run_name': '/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2', 'disable_tqdm': False, 'remove_unused_columns': True, 'label_names': None, 'load_best_model_at_end': False, 'metric_for_best_model': None, 'greater_is_better': None, 'ignore_data_skip': False, 'fsdp': [], 'fsdp_min_num_params': 0, 'fsdp_config': {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}, 'fsdp_transformer_layer_cls_to_wrap': None, 'accelerator_config': {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}, 'deepspeed': None, 'label_smoothing_factor': 0.0, 'optim': 'paged_adamw_32bit', 'optim_args': None, 'adafactor': False, 'group_by_length': True, 'length_column_name': 'length', 'report_to': ['wandb'], 'ddp_find_unused_parameters': None, 'ddp_bucket_cap_mb': None, 'ddp_broadcast_buffers': None, 'dataloader_pin_memory': True, 'dataloader_persistent_workers': False, 'skip_memory_metrics': True, 'use_legacy_prediction_loop': False, 'push_to_hub': False, 'resume_from_checkpoint': None, 'hub_model_id': None, 'hub_strategy': 'every_save', 'hub_token': '<HUB_TOKEN>', 'hub_private_repo': None, 'hub_always_push': False, 'gradient_checkpointing': False, 'gradient_checkpointing_kwargs': None, 'include_inputs_for_metrics': False, 'include_for_metrics': [], 'eval_do_concat_batches': True, 'fp16_backend': 'auto', 'evaluation_strategy': None, 'push_to_hub_model_id': None, 'push_to_hub_organization': None, 'push_to_hub_token': '<PUSH_TO_HUB_TOKEN>', 'mp_parameters': '', 'auto_find_batch_size': False, 'full_determinism': False, 'torchdynamo': None, 'ray_scope': 'last', 'ddp_timeout': 1800, 'torch_compile': False, 'torch_compile_backend': None, 'torch_compile_mode': None, 'dispatch_batches': None, 'split_batches': None, 'include_tokens_per_second': False, 'include_num_input_tokens_seen': False, 'neftune_noise_alpha': None, 'optim_target_modules': None, 'batch_eval_metrics': False, 'eval_on_start': False, 'use_liger_kernel': False, 'eval_use_gather_object': False, 'average_tokens_across_devices': False, 'dataset_text_field': 'text', 'packing': False, 'max_seq_length': 100, 'dataset_num_proc': None, 'dataset_batch_size': 1000, 'model_init_kwargs': None, 'dataset_kwargs': {}, 'eval_packing': None, 'num_of_sequences': 1024, 'chars_per_token': '<CHARS_PER_TOKEN>', 'use_liger': False, 'joint_obs_action_mlp_lr': 0.0001, 'obs_mlp_lr': None, 'action_mlp_lr': None}
27
+ 2026-11-07 09:51:34,174 INFO MainThread:147051 [wandb_config.py:__setitem__():154] config set model/num_parameters = 70696448 - <bound method Run._config_callback of <wandb.sdk.wandb_run.Run object at 0x7a2091102fa0>>
28
+ 2026-11-07 09:51:34,174 INFO MainThread:147051 [wandb_run.py:_config_callback():1389] config_cb model/num_parameters 70696448 None
29
+ 2026-11-07 10:26:08,398 INFO MainThread:147051 [wandb_run.py:_finish():2146] finishing run b10902069-national-taiwan-university/llm_module_finetuning/d0nbcloa
30
+ 2026-11-07 10:26:08,398 INFO MainThread:147051 [wandb_run.py:_atexit_cleanup():2414] got exitcode: 0
31
+ 2026-11-07 10:26:08,399 INFO MainThread:147051 [wandb_run.py:_restore():2396] restore
32
+ 2026-11-07 10:26:08,399 INFO MainThread:147051 [wandb_run.py:_restore():2402] restore done
33
+ 2026-11-07 10:26:13,092 INFO MainThread:147051 [wandb_run.py:_footer_history_summary_info():3963] rendering history
34
+ 2026-11-07 10:26:13,094 INFO MainThread:147051 [wandb_run.py:_footer_history_summary_info():3995] rendering summary
35
+ 2026-11-07 10:26:13,098 INFO MainThread:147051 [wandb_run.py:_footer_sync_info():3922] logging synced files
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/files/config.yaml ADDED
@@ -0,0 +1,708 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _attn_implementation_autoset:
2
+ value: true
3
+ _name_or_path:
4
+ value: EleutherAI/pythia-70m-deduped
5
+ _target_:
6
+ value: llmbc.workspace.train_llm_workspace.TrainLLMWorkspace
7
+ _wandb:
8
+ value:
9
+ cli_version: 0.18.6
10
+ m:
11
+ - "1": eval/steps_per_second
12
+ "5": 2
13
+ "6":
14
+ - 1
15
+ - 3
16
+ "7": []
17
+ - "1": train/global_step
18
+ "6":
19
+ - 3
20
+ "7": []
21
+ - "1": train/epoch
22
+ "5": 2
23
+ "6":
24
+ - 1
25
+ - 3
26
+ "7": []
27
+ - "1": train/loss
28
+ "5": 2
29
+ "6":
30
+ - 1
31
+ - 3
32
+ "7": []
33
+ - "1": eval/loss
34
+ "5": 2
35
+ "6":
36
+ - 1
37
+ - 3
38
+ "7": []
39
+ - "1": train/learning_rate
40
+ "5": 2
41
+ "6":
42
+ - 1
43
+ - 3
44
+ "7": []
45
+ - "1": train/grad_norm
46
+ "5": 2
47
+ "6":
48
+ - 1
49
+ - 3
50
+ "7": []
51
+ - "1": eval/runtime
52
+ "5": 2
53
+ "6":
54
+ - 1
55
+ - 3
56
+ "7": []
57
+ - "1": eval/samples_per_second
58
+ "5": 2
59
+ "6":
60
+ - 1
61
+ - 3
62
+ "7": []
63
+ python_version: 3.9.25
64
+ t:
65
+ "1":
66
+ - 1
67
+ - 2
68
+ - 3
69
+ - 5
70
+ - 11
71
+ - 12
72
+ - 41
73
+ - 49
74
+ - 50
75
+ - 51
76
+ - 53
77
+ - 55
78
+ - 71
79
+ - 84
80
+ - 98
81
+ - 105
82
+ "2":
83
+ - 1
84
+ - 2
85
+ - 3
86
+ - 5
87
+ - 11
88
+ - 12
89
+ - 41
90
+ - 49
91
+ - 50
92
+ - 51
93
+ - 53
94
+ - 55
95
+ - 71
96
+ - 84
97
+ - 98
98
+ - 105
99
+ "3":
100
+ - 2
101
+ - 7
102
+ - 13
103
+ - 15
104
+ - 16
105
+ - 19
106
+ - 23
107
+ - 55
108
+ - 62
109
+ - 66
110
+ "4": 3.9.25
111
+ "5": 0.18.6
112
+ "6": 4.47.1
113
+ "8":
114
+ - 5
115
+ "9":
116
+ "1": transformers_trainer
117
+ "12": 0.18.6
118
+ "13": linux-x86_64
119
+ accelerator_config:
120
+ value:
121
+ dispatch_batches: null
122
+ even_batches: true
123
+ gradient_accumulation_kwargs: null
124
+ non_blocking: false
125
+ split_batches: false
126
+ use_seedable_sampler: true
127
+ action_dim:
128
+ value: 4
129
+ action_mlp_lr:
130
+ value: null
131
+ adafactor:
132
+ value: false
133
+ adam_beta1:
134
+ value: 0.9
135
+ adam_beta2:
136
+ value: 0.999
137
+ adam_epsilon:
138
+ value: 1e-08
139
+ add_cross_attention:
140
+ value: false
141
+ architectures:
142
+ value:
143
+ - GPTNeoXForCausalLM
144
+ attention_bias:
145
+ value: true
146
+ attention_dropout:
147
+ value: 0
148
+ auto_find_batch_size:
149
+ value: false
150
+ average_tokens_across_devices:
151
+ value: false
152
+ bad_words_ids:
153
+ value: null
154
+ batch_eval_metrics:
155
+ value: false
156
+ begin_suppress_tokens:
157
+ value: null
158
+ bf16:
159
+ value: true
160
+ bf16_full_eval:
161
+ value: false
162
+ bos_token_id:
163
+ value: 0
164
+ chars_per_token:
165
+ value: <CHARS_PER_TOKEN>
166
+ chunk_size_feed_forward:
167
+ value: 0
168
+ classifier_dropout:
169
+ value: 0.1
170
+ cross_attention_hidden_size:
171
+ value: null
172
+ data_seed:
173
+ value: null
174
+ dataloader_drop_last:
175
+ value: false
176
+ dataloader_num_workers:
177
+ value: 0
178
+ dataloader_persistent_workers:
179
+ value: false
180
+ dataloader_pin_memory:
181
+ value: true
182
+ dataloader_prefetch_factor:
183
+ value: null
184
+ dataset:
185
+ value:
186
+ test_data_ratio: 0.01
187
+ dataset_batch_size:
188
+ value: 1000
189
+ dataset_num_proc:
190
+ value: null
191
+ dataset_text_field:
192
+ value: text
193
+ ddp_backend:
194
+ value: null
195
+ ddp_broadcast_buffers:
196
+ value: null
197
+ ddp_bucket_cap_mb:
198
+ value: null
199
+ ddp_find_unused_parameters:
200
+ value: null
201
+ ddp_timeout:
202
+ value: 1800
203
+ debug:
204
+ value: []
205
+ decoder_start_token_id:
206
+ value: null
207
+ deepspeed:
208
+ value: null
209
+ disable_tqdm:
210
+ value: false
211
+ dispatch_batches:
212
+ value: null
213
+ diversity_penalty:
214
+ value: 0
215
+ do_eval:
216
+ value: true
217
+ do_predict:
218
+ value: false
219
+ do_sample:
220
+ value: false
221
+ do_train:
222
+ value: false
223
+ early_stopping:
224
+ value: false
225
+ encoder_no_repeat_ngram_size:
226
+ value: 0
227
+ eos_token_id:
228
+ value: 0
229
+ eval_accumulation_steps:
230
+ value: null
231
+ eval_delay:
232
+ value: 0
233
+ eval_do_concat_batches:
234
+ value: true
235
+ eval_on_start:
236
+ value: false
237
+ eval_packing:
238
+ value: null
239
+ eval_steps:
240
+ value: 10
241
+ eval_strategy:
242
+ value: steps
243
+ eval_use_gather_object:
244
+ value: false
245
+ evaluation_strategy:
246
+ value: null
247
+ exp_name:
248
+ value: train llm
249
+ exponential_decay_length_penalty:
250
+ value: null
251
+ finetuning_task:
252
+ value: null
253
+ forced_bos_token_id:
254
+ value: null
255
+ forced_eos_token_id:
256
+ value: null
257
+ fp16:
258
+ value: false
259
+ fp16_backend:
260
+ value: auto
261
+ fp16_full_eval:
262
+ value: false
263
+ fp16_opt_level:
264
+ value: O1
265
+ fsdp:
266
+ value: []
267
+ fsdp_config:
268
+ value:
269
+ min_num_params: 0
270
+ xla: false
271
+ xla_fsdp_grad_ckpt: false
272
+ xla_fsdp_v2: false
273
+ fsdp_min_num_params:
274
+ value: 0
275
+ fsdp_transformer_layer_cls_to_wrap:
276
+ value: null
277
+ full_determinism:
278
+ value: false
279
+ gradient_accumulation_steps:
280
+ value: 1
281
+ gradient_checkpointing:
282
+ value: false
283
+ gradient_checkpointing_kwargs:
284
+ value: null
285
+ greater_is_better:
286
+ value: null
287
+ group_by_length:
288
+ value: true
289
+ half_precision_backend:
290
+ value: auto
291
+ hidden_act:
292
+ value: gelu
293
+ hidden_dropout:
294
+ value: 0
295
+ hidden_size:
296
+ value: 512
297
+ horizon:
298
+ value: 1
299
+ hub_always_push:
300
+ value: false
301
+ hub_model_id:
302
+ value: null
303
+ hub_private_repo:
304
+ value: null
305
+ hub_strategy:
306
+ value: every_save
307
+ hub_token:
308
+ value: <HUB_TOKEN>
309
+ id2label:
310
+ value:
311
+ "0": LABEL_0
312
+ "1": LABEL_1
313
+ ignore_data_skip:
314
+ value: false
315
+ include_for_metrics:
316
+ value: []
317
+ include_inputs_for_metrics:
318
+ value: false
319
+ include_num_input_tokens_seen:
320
+ value: false
321
+ include_tokens_per_second:
322
+ value: false
323
+ initializer_range:
324
+ value: 0.02
325
+ intermediate_size:
326
+ value: 2048
327
+ is_decoder:
328
+ value: false
329
+ is_encoder_decoder:
330
+ value: false
331
+ jit_mode_eval:
332
+ value: false
333
+ joint_obs_action_mlp_lr:
334
+ value: 0.0001
335
+ label_names:
336
+ value: null
337
+ label_smoothing_factor:
338
+ value: 0
339
+ label2id:
340
+ value:
341
+ LABEL_0: 0
342
+ LABEL_1: 1
343
+ layer_norm_eps:
344
+ value: 1e-05
345
+ learning_rate:
346
+ value: 0.0001
347
+ length_column_name:
348
+ value: length
349
+ length_penalty:
350
+ value: 1
351
+ llm:
352
+ value:
353
+ causal_lm_target: llmbc.model.llm.gpt_neox_lowdim_model.LowdimGPTNeoXForCausalLM
354
+ checkpoint: data/outputs/2026.11.07/01.39.14_train_mlp_projector_box-close-v2/checkpoints/latest.ckpt
355
+ config_target: llmbc.model.llm.gpt_neox_lowdim_model.LowdimGPTNeoXConfig
356
+ finetune_mode: orig
357
+ hydra:
358
+ job:
359
+ override_dirname: EleutherAI/pythia-70m-deduped
360
+ run:
361
+ dir: data/outputs/2026.11.07/09.50.49_EleutherAI/pythia-70m-deduped
362
+ llm_mode: mlp-finetuned
363
+ lora_config:
364
+ bias: none
365
+ lora_alpha: 64
366
+ lora_dropout: 0.05
367
+ r: 32
368
+ task_type: CAUSAL_LM
369
+ max_length: 100
370
+ model_name: pythia-70m-deduped
371
+ name: EleutherAI/pythia-70m-deduped
372
+ prompter:
373
+ _target_: llmbc.translator.prompter.pythia_prompter.PythiaPrompter
374
+ use_joint_mlp_projector: true
375
+ use_joint_mlp_projector: true
376
+ use_quantization: false
377
+ load_best_model_at_end:
378
+ value: false
379
+ local_rank:
380
+ value: 0
381
+ log_level:
382
+ value: passive
383
+ log_level_replica:
384
+ value: warning
385
+ log_on_each_node:
386
+ value: true
387
+ logging:
388
+ value:
389
+ group: null
390
+ id: null
391
+ mode: online
392
+ name: 2026.11.07-09.50.49_train_llm_lowdim_box-close-v2
393
+ project: llm_module_finetuning
394
+ resume: true
395
+ tags:
396
+ - train_llm_lowdim
397
+ - box-close-v2
398
+ - train llm
399
+ logging_dir:
400
+ value: /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/runs/Nov07_09-51-23_A6000
401
+ logging_first_step:
402
+ value: false
403
+ logging_nan_inf_filter:
404
+ value: true
405
+ logging_steps:
406
+ value: 1
407
+ logging_strategy:
408
+ value: steps
409
+ lora_config:
410
+ value:
411
+ bias: none
412
+ lora_alpha: 64
413
+ lora_dropout: 0.05
414
+ r: 32
415
+ task_type: CAUSAL_LM
416
+ lr_scheduler_type:
417
+ value: linear
418
+ max_grad_norm:
419
+ value: 1
420
+ max_length:
421
+ value: 20
422
+ max_position_embeddings:
423
+ value: 2048
424
+ max_seq_length:
425
+ value: 100
426
+ max_steps:
427
+ value: -1
428
+ metric_for_best_model:
429
+ value: null
430
+ min_length:
431
+ value: 0
432
+ model/num_parameters:
433
+ value: 70696448
434
+ model_init_kwargs:
435
+ value: null
436
+ model_name:
437
+ value: EleutherAI/pythia-70m-deduped
438
+ model_type:
439
+ value: gpt_neox_lowdim
440
+ mp_parameters:
441
+ value: ""
442
+ multi_run:
443
+ value:
444
+ run_dir: data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2
445
+ wandb_name_base: 2026.11.07-09.50.49_train_llm_lowdim_box-close-v2
446
+ n_action_steps:
447
+ value: 1
448
+ n_obs_steps:
449
+ value: 1
450
+ name:
451
+ value: train_llm_lowdim
452
+ neftune_noise_alpha:
453
+ value: null
454
+ no_cuda:
455
+ value: false
456
+ no_repeat_ngram_size:
457
+ value: 0
458
+ num_attention_heads:
459
+ value: 8
460
+ num_beam_groups:
461
+ value: 1
462
+ num_beams:
463
+ value: 1
464
+ num_hidden_layers:
465
+ value: 6
466
+ num_of_sequences:
467
+ value: 1024
468
+ num_return_sequences:
469
+ value: 1
470
+ num_train_epochs:
471
+ value: 10
472
+ obs_dim:
473
+ value: 9
474
+ obs_mlp_lr:
475
+ value: null
476
+ optim:
477
+ value: paged_adamw_32bit
478
+ optim_args:
479
+ value: null
480
+ optim_target_modules:
481
+ value: null
482
+ output_attentions:
483
+ value: false
484
+ output_dir:
485
+ value: /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2
486
+ output_hidden_states:
487
+ value: false
488
+ output_scores:
489
+ value: false
490
+ overwrite_output_dir:
491
+ value: false
492
+ packing:
493
+ value: false
494
+ pad_token_id:
495
+ value: null
496
+ partial_rotary_factor:
497
+ value: 0.25
498
+ past_index:
499
+ value: -1
500
+ per_device_eval_batch_size:
501
+ value: 128
502
+ per_device_train_batch_size:
503
+ value: 128
504
+ per_gpu_eval_batch_size:
505
+ value: null
506
+ per_gpu_train_batch_size:
507
+ value: null
508
+ prediction_loss_only:
509
+ value: false
510
+ prefix:
511
+ value: null
512
+ problem_type:
513
+ value: null
514
+ push_to_hub:
515
+ value: false
516
+ push_to_hub_model_id:
517
+ value: null
518
+ push_to_hub_organization:
519
+ value: null
520
+ push_to_hub_token:
521
+ value: <PUSH_TO_HUB_TOKEN>
522
+ ray_scope:
523
+ value: last
524
+ remove_invalid_values:
525
+ value: false
526
+ remove_unused_columns:
527
+ value: true
528
+ repetition_penalty:
529
+ value: 1
530
+ report_to:
531
+ value:
532
+ - wandb
533
+ restore_callback_states_from_checkpoint:
534
+ value: false
535
+ resume_from_checkpoint:
536
+ value: null
537
+ return_dict:
538
+ value: true
539
+ return_dict_in_generate:
540
+ value: false
541
+ rope_scaling:
542
+ value: null
543
+ rope_theta:
544
+ value: 10000
545
+ rotary_emb_base:
546
+ value: 10000
547
+ rotary_pct:
548
+ value: 0.25
549
+ run_name:
550
+ value: /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2
551
+ save_on_each_node:
552
+ value: false
553
+ save_only_model:
554
+ value: false
555
+ save_safetensors:
556
+ value: true
557
+ save_steps:
558
+ value: 5000
559
+ save_strategy:
560
+ value: steps
561
+ save_total_limit:
562
+ value: null
563
+ seed:
564
+ value: 42
565
+ sep_token_id:
566
+ value: null
567
+ skip_memory_metrics:
568
+ value: true
569
+ split_batches:
570
+ value: null
571
+ suppress_tokens:
572
+ value: null
573
+ task:
574
+ value:
575
+ action_dim: 4
576
+ dataset:
577
+ _target_: llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset
578
+ data_path: datasets/box-close-v2-general.pt
579
+ data_path2: datasets/box-close-v2.pt
580
+ dummy_normalizer: false
581
+ horizon: 1
582
+ obs_eef_target: true
583
+ pad_after: 0
584
+ pad_before: 0
585
+ use_manual_normalizer: false
586
+ val_ratio: 0.05
587
+ env_runner:
588
+ _target_: llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner
589
+ discount: 0.9
590
+ env_name: llf-metaworld-box-close-v2
591
+ feedback_type:
592
+ - hp
593
+ - hn
594
+ - fp
595
+ instruction_type: b
596
+ max_steps: 30
597
+ n_action_steps: 1
598
+ n_envs: 10
599
+ n_obs_steps: 1
600
+ n_test: 50
601
+ n_train: 10
602
+ visual: false
603
+ instructor:
604
+ _target_: llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor
605
+ name: box-close-v2
606
+ obs_dim: 9
607
+ task_name:
608
+ value: box-close-v2
609
+ task_specific_params:
610
+ value: null
611
+ temperature:
612
+ value: 1
613
+ tf_legacy_loss:
614
+ value: false
615
+ tf32:
616
+ value: true
617
+ tie_encoder_decoder:
618
+ value: false
619
+ tie_word_embeddings:
620
+ value: false
621
+ tokenizer_class:
622
+ value: null
623
+ top_k:
624
+ value: 50
625
+ top_p:
626
+ value: 1
627
+ torch_compile:
628
+ value: false
629
+ torch_compile_backend:
630
+ value: null
631
+ torch_compile_mode:
632
+ value: null
633
+ torch_dtype:
634
+ value: float16
635
+ torch_empty_cache_steps:
636
+ value: null
637
+ torchdynamo:
638
+ value: null
639
+ torchscript:
640
+ value: false
641
+ tpu_metrics_debug:
642
+ value: false
643
+ tpu_num_cores:
644
+ value: null
645
+ trainer:
646
+ value:
647
+ action_dim: 4
648
+ dataset_text_field: text
649
+ max_seq_length: 100
650
+ obs_dim: 9
651
+ packing: false
652
+ use_joint_mlp_projector: true
653
+ training:
654
+ value:
655
+ bf16: true
656
+ eval_steps: 10
657
+ eval_strategy: steps
658
+ fp16: false
659
+ gradient_accumulation_steps: 1
660
+ group_by_length: true
661
+ joint_obs_action_mlp_lr: 0.0001
662
+ learning_rate: 0.0001
663
+ logging_steps: 1
664
+ logging_strategy: steps
665
+ num_train_epochs: 10
666
+ optim: paged_adamw_32bit
667
+ per_device_eval_batch_size: 128
668
+ per_device_train_batch_size: 128
669
+ report_to: wandb
670
+ save_steps: 5000
671
+ seed: 42
672
+ tf32: true
673
+ use_joint_mlp_projector: true
674
+ warmup_steps: 100
675
+ transformers_version:
676
+ value: 4.47.1
677
+ typical_p:
678
+ value: 1
679
+ use_bfloat16:
680
+ value: false
681
+ use_cache:
682
+ value: false
683
+ use_cpu:
684
+ value: false
685
+ use_ipex:
686
+ value: false
687
+ use_joint_mlp_projector:
688
+ value: true
689
+ use_legacy_prediction_loop:
690
+ value: false
691
+ use_liger:
692
+ value: false
693
+ use_liger_kernel:
694
+ value: false
695
+ use_mps_device:
696
+ value: false
697
+ use_parallel_residual:
698
+ value: true
699
+ use_quantization:
700
+ value: false
701
+ vocab_size:
702
+ value: 50304
703
+ warmup_ratio:
704
+ value: 0
705
+ warmup_steps:
706
+ value: 100
707
+ weight_decay:
708
+ value: 0
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/files/output.log ADDED
The diff for this file is too large to render. See raw diff
 
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/files/wandb-metadata.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-6.8.0-138-generic-x86_64-with-glibc2.39",
3
+ "python": "3.9.25",
4
+ "startedAt": "2026-11-07T01:50:53.589118Z",
5
+ "args": [
6
+ "--config-path",
7
+ "config/llm_backbone",
8
+ "--config-name",
9
+ "llmdp_llm_box-close-v2_pythia-70m-deduped.yaml"
10
+ ],
11
+ "program": "/tmp2/chyang/workspace/LLM-BC/./train.py",
12
+ "codePath": "train.py",
13
+ "git": {
14
+ "remote": "https://github.com/CHYang25/LLM-BC.git",
15
+ "commit": "b986d300a4656b4611ccfc7f623631112b5de2e8"
16
+ },
17
+ "email": "chris920325@gmail.com",
18
+ "root": "/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2",
19
+ "host": "A6000",
20
+ "username": "chyang",
21
+ "executable": "/home/chyang/miniconda3/envs/llm-bc/bin/python3",
22
+ "codePathLocal": "train.py",
23
+ "cpu_count": 12,
24
+ "cpu_count_logical": 24,
25
+ "gpu": "NVIDIA RTX A6000",
26
+ "gpu_count": 2,
27
+ "disk": {
28
+ "/": {
29
+ "total": "982820896768",
30
+ "used": "680385912832"
31
+ }
32
+ },
33
+ "memory": {
34
+ "total": "134538604544"
35
+ },
36
+ "cpu": {
37
+ "count": 12,
38
+ "countLogical": 24
39
+ },
40
+ "gpu_nvidia": [
41
+ {
42
+ "name": "NVIDIA RTX A6000",
43
+ "memoryTotal": "51527024640",
44
+ "cudaCores": 10752,
45
+ "architecture": "Ampere"
46
+ },
47
+ {
48
+ "name": "NVIDIA RTX A6000",
49
+ "memoryTotal": "51527024640",
50
+ "cudaCores": 10752,
51
+ "architecture": "Ampere"
52
+ }
53
+ ],
54
+ "cudaVersion": "13.2"
55
+ }
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/files/wandb-summary.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"train_loss":0.7388594108394937,"_wandb":{"runtime":2114},"train/grad_norm":63.07209014892578,"train/epoch":10,"eval/loss":0.623070478439331,"_runtime":2114.809156464,"total_flos":2.496428183808e+16,"_timestamp":1.7940183683976324e+09,"eval/runtime":0.6163,"train_runtime":2074.2483,"train_steps_per_second":2.84,"eval/samples_per_second":1234.693,"train_samples_per_second":362.878,"train/learning_rate":0,"_step":6479,"train/global_step":5890,"eval/steps_per_second":9.735,"train/loss":0.4822}
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/logs/debug-core.log ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-11-07T09:50:52.420254724+08:00","level":"INFO","msg":"started logging, with flags","port-filename":"/tmp/tmppc8lr5bp/port-147051.txt","pid":147051,"debug":false,"disable-analytics":false}
2
+ {"time":"2026-11-07T09:50:52.420281805+08:00","level":"INFO","msg":"FeatureState","shutdownOnParentExitEnabled":false}
3
+ {"time":"2026-11-07T09:50:52.425074755+08:00","level":"INFO","msg":"server is running","addr":{"IP":"127.0.0.1","Port":34143,"Zone":""}}
4
+ {"time":"2026-11-07T09:50:52.425112764+08:00","level":"INFO","msg":"Will exit if parent process dies.","ppid":147051}
5
+ {"time":"2026-11-07T09:50:52.617814049+08:00","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"127.0.0.1:42100"}
6
+ {"time":"2026-11-07T09:50:53.591745404+08:00","level":"INFO","msg":"handleInformInit: received","streamId":"d0nbcloa","id":"127.0.0.1:42100"}
7
+ {"time":"2026-11-07T09:50:53.699058828+08:00","level":"INFO","msg":"handleInformInit: stream started","streamId":"d0nbcloa","id":"127.0.0.1:42100"}
8
+ {"time":"2026-11-07T10:26:13.09893658+08:00","level":"INFO","msg":"handleInformFinish: finish message received","streamId":"d0nbcloa","id":"127.0.0.1:42100"}
9
+ {"time":"2026-11-07T10:26:13.09925827+08:00","level":"INFO","msg":"handleInformFinish: stream closed","streamId":"d0nbcloa","id":"127.0.0.1:42100"}
10
+ {"time":"2026-11-07T10:26:14.15301606+08:00","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"127.0.0.1:42100"}
11
+ {"time":"2026-11-07T10:26:14.153081485+08:00","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"127.0.0.1:42100"}
12
+ {"time":"2026-11-07T10:26:14.153095932+08:00","level":"INFO","msg":"server is shutting down"}
13
+ {"time":"2026-11-07T10:26:14.153139836+08:00","level":"INFO","msg":"connection: Close: initiating connection closure","id":"127.0.0.1:42100"}
14
+ {"time":"2026-11-07T10:26:14.153384475+08:00","level":"INFO","msg":"connection: Close: connection successfully closed","id":"127.0.0.1:42100"}
15
+ {"time":"2026-11-07T10:26:14.15339756+08:00","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"127.0.0.1:42100"}
16
+ {"time":"2026-11-07T10:26:14.153411084+08:00","level":"INFO","msg":"server is closed"}
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/logs/debug-internal.log ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-11-07T09:50:53.592038827+08:00","level":"INFO","msg":"using version","core version":"0.18.6"}
2
+ {"time":"2026-11-07T09:50:53.592065195+08:00","level":"INFO","msg":"created symlink","path":"/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/logs/debug-core.log"}
3
+ {"time":"2026-11-07T09:50:53.699007609+08:00","level":"INFO","msg":"created new stream","id":"d0nbcloa"}
4
+ {"time":"2026-11-07T09:50:53.69905307+08:00","level":"INFO","msg":"stream: started","id":"d0nbcloa"}
5
+ {"time":"2026-11-07T09:50:53.699072256+08:00","level":"INFO","msg":"sender: started","stream_id":"d0nbcloa"}
6
+ {"time":"2026-11-07T09:50:53.699068072+08:00","level":"INFO","msg":"handler: started","stream_id":{"value":"d0nbcloa"}}
7
+ {"time":"2026-11-07T09:50:53.69908682+08:00","level":"INFO","msg":"writer: Do: started","stream_id":{"value":"d0nbcloa"}}
8
+ {"time":"2026-11-07T09:50:54.426318415+08:00","level":"INFO","msg":"Starting system monitor"}
9
+ {"time":"2026-11-07T10:26:08.400121893+08:00","level":"INFO","msg":"Stopping system monitor"}
10
+ {"time":"2026-11-07T10:26:08.400943972+08:00","level":"INFO","msg":"Stopped system monitor"}
11
+ {"time":"2026-11-07T10:26:09.400990261+08:00","level":"INFO","msg":"handler: operation stats","stats":{"operations":[{"desc":"saving job artifact","runtime_seconds":0.331823839},{"desc":"uploading config.yaml","runtime_seconds":0.089839805,"progress":"14.4KB/14.4KB"},{"desc":"uploading output.log","runtime_seconds":0.089825287,"progress":"990.8KB/990.8KB"}],"total_operations":3}}
12
+ {"time":"2026-11-07T10:26:11.273318383+08:00","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
13
+ {"time":"2026-11-07T10:26:13.099047371+08:00","level":"INFO","msg":"stream: closing","id":"d0nbcloa"}
14
+ {"time":"2026-11-07T10:26:13.099087573+08:00","level":"INFO","msg":"handler: closed","stream_id":{"value":"d0nbcloa"}}
15
+ {"time":"2026-11-07T10:26:13.099122038+08:00","level":"INFO","msg":"writer: Close: closed","stream_id":{"value":"d0nbcloa"}}
16
+ {"time":"2026-11-07T10:26:13.099185666+08:00","level":"INFO","msg":"sender: closed","stream_id":"d0nbcloa"}
17
+ {"time":"2026-11-07T10:26:13.099239802+08:00","level":"INFO","msg":"stream: closed","id":"d0nbcloa"}
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/logs/debug.log ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_setup.py:_flush():79] Current SDK version is 0.18.6
2
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_setup.py:_flush():79] Configure stats pid to 147051
3
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_setup.py:_flush():79] Loading settings from /home/chyang/.config/wandb/settings
4
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_setup.py:_flush():79] Loading settings from /tmp2/chyang/workspace/LLM-BC/wandb/settings
5
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_setup.py:_flush():79] Loading settings from environment variables: {}
6
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_setup.py:_flush():79] Applying setup settings: {'mode': 'online', '_disable_service': None}
7
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [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'}
8
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_setup.py:_flush():79] Applying login settings: {}
9
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_init.py:_log_setup():533] Logging user logs to /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/logs/debug.log
10
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_init.py:_log_setup():534] Logging internal logs to /tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/logs/debug-internal.log
11
+ 2026-11-07 09:50:53,587 INFO MainThread:147051 [wandb_init.py:init():619] calling init triggers
12
+ 2026-11-07 09:50:53,588 INFO MainThread:147051 [wandb_init.py:init():626] wandb.init called with sweep_config: {}
13
+ config: {'name': 'train_llm_lowdim', '_target_': 'llmbc.workspace.train_llm_workspace.TrainLLMWorkspace', 'obs_dim': 9, 'action_dim': 4, 'horizon': 1, 'n_obs_steps': 1, 'n_action_steps': 1, 'task_name': 'box-close-v2', 'exp_name': 'train llm', 'model_name': 'EleutherAI/pythia-70m-deduped', 'use_quantization': False, 'lora_config': {'r': 32, 'lora_alpha': 64, 'lora_dropout': 0.05, 'bias': 'none', 'task_type': 'CAUSAL_LM'}, 'dataset': {'test_data_ratio': 0.01}, 'debug': False, 'training': {'seed': 42, 'per_device_train_batch_size': 128, 'per_device_eval_batch_size': 128, 'gradient_accumulation_steps': 1, 'optim': 'paged_adamw_32bit', 'num_train_epochs': 10, 'eval_strategy': 'steps', 'logging_steps': 1, 'warmup_steps': 100, 'logging_strategy': 'steps', 'learning_rate': 0.0001, 'fp16': False, 'bf16': True, 'tf32': True, 'group_by_length': True, 'report_to': 'wandb', 'save_steps': 5000, 'eval_steps': 10, 'use_joint_mlp_projector': True, 'joint_obs_action_mlp_lr': 0.0001}, 'trainer': {'obs_dim': 9, 'action_dim': 4, 'use_joint_mlp_projector': True, 'max_seq_length': 100, 'dataset_text_field': 'text', 'packing': False}, 'logging': {'project': 'llm_module_finetuning', 'resume': True, 'mode': 'online', 'name': '2026.11.07-09.50.49_train_llm_lowdim_box-close-v2', 'tags': ['train_llm_lowdim', 'box-close-v2', 'train llm'], 'id': None, 'group': None}, 'multi_run': {'run_dir': 'data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2', 'wandb_name_base': '2026.11.07-09.50.49_train_llm_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-general.pt', 'data_path2': 'datasets/box-close-v2.pt', 'horizon': 1, 'pad_before': 0, 'pad_after': 0, 'obs_eef_target': True, 'use_manual_normalizer': False, 'val_ratio': 0.05, 'dummy_normalizer': False}, 'instructor': {'_target_': 'llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor'}}, 'llm': {'name': 'EleutherAI/pythia-70m-deduped', 'model_name': 'pythia-70m-deduped', 'config_target': 'llmbc.model.llm.gpt_neox_lowdim_model.LowdimGPTNeoXConfig', 'causal_lm_target': 'llmbc.model.llm.gpt_neox_lowdim_model.LowdimGPTNeoXForCausalLM', 'use_quantization': False, 'use_joint_mlp_projector': True, 'llm_mode': 'mlp-finetuned', 'finetune_mode': 'orig', 'checkpoint': 'data/outputs/2026.11.07/01.39.14_train_mlp_projector_box-close-v2/checkpoints/latest.ckpt', '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.pythia_prompter.PythiaPrompter', 'use_joint_mlp_projector': True}, 'hydra': {'job': {'override_dirname': 'EleutherAI/pythia-70m-deduped'}, 'run': {'dir': 'data/outputs/2026.11.07/09.50.49_EleutherAI/pythia-70m-deduped'}}}}
14
+ 2026-11-07 09:50:53,588 INFO MainThread:147051 [wandb_init.py:init():669] starting backend
15
+ 2026-11-07 09:50:53,588 INFO MainThread:147051 [wandb_init.py:init():673] sending inform_init request
16
+ 2026-11-07 09:50:53,588 INFO MainThread:147051 [backend.py:_multiprocessing_setup():104] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
17
+ 2026-11-07 09:50:53,589 INFO MainThread:147051 [wandb_init.py:init():686] backend started and connected
18
+ 2026-11-07 09:50:53,593 INFO MainThread:147051 [wandb_init.py:init():781] updated telemetry
19
+ 2026-11-07 09:50:53,613 INFO MainThread:147051 [wandb_init.py:init():814] communicating run to backend with 90.0 second timeout
20
+ 2026-11-07 09:50:54,423 INFO MainThread:147051 [wandb_init.py:init():867] starting run threads in backend
21
+ 2026-11-07 09:50:54,530 INFO MainThread:147051 [wandb_run.py:_console_start():2451] atexit reg
22
+ 2026-11-07 09:50:54,530 INFO MainThread:147051 [wandb_run.py:_redirect():2299] redirect: wrap_raw
23
+ 2026-11-07 09:50:54,530 INFO MainThread:147051 [wandb_run.py:_redirect():2364] Wrapping output streams.
24
+ 2026-11-07 09:50:54,530 INFO MainThread:147051 [wandb_run.py:_redirect():2389] Redirects installed.
25
+ 2026-11-07 09:50:54,532 INFO MainThread:147051 [wandb_init.py:init():911] run started, returning control to user process
26
+ 2026-11-07 09:51:34,173 INFO MainThread:147051 [wandb_run.py:_config_callback():1389] config_cb None None {'obs_dim': 9, 'action_dim': 4, 'use_joint_mlp_projector': True, 'return_dict': True, 'output_hidden_states': False, 'output_attentions': False, 'torchscript': False, 'torch_dtype': 'float16', 'use_bfloat16': False, 'tf_legacy_loss': False, 'pruned_heads': {}, 'tie_word_embeddings': False, 'chunk_size_feed_forward': 0, 'is_encoder_decoder': False, 'is_decoder': False, 'cross_attention_hidden_size': None, 'add_cross_attention': False, 'tie_encoder_decoder': False, 'max_length': 20, 'min_length': 0, 'do_sample': False, 'early_stopping': False, 'num_beams': 1, 'num_beam_groups': 1, 'diversity_penalty': 0.0, 'temperature': 1.0, 'top_k': 50, 'top_p': 1.0, 'typical_p': 1.0, 'repetition_penalty': 1.0, 'length_penalty': 1.0, 'no_repeat_ngram_size': 0, 'encoder_no_repeat_ngram_size': 0, 'bad_words_ids': None, 'num_return_sequences': 1, 'output_scores': False, 'return_dict_in_generate': False, 'forced_bos_token_id': None, 'forced_eos_token_id': None, 'remove_invalid_values': False, 'exponential_decay_length_penalty': None, 'suppress_tokens': None, 'begin_suppress_tokens': None, 'architectures': ['GPTNeoXForCausalLM'], 'finetuning_task': None, 'id2label': {0: 'LABEL_0', 1: 'LABEL_1'}, 'label2id': {'LABEL_0': 0, 'LABEL_1': 1}, 'tokenizer_class': None, 'prefix': None, 'bos_token_id': 0, 'pad_token_id': None, 'eos_token_id': 0, 'sep_token_id': None, 'decoder_start_token_id': None, 'task_specific_params': None, 'problem_type': None, '_name_or_path': 'EleutherAI/pythia-70m-deduped', '_attn_implementation_autoset': True, 'transformers_version': '4.47.1', 'model_type': 'gpt_neox_lowdim', 'vocab_size': 50304, 'max_position_embeddings': 2048, 'hidden_size': 512, 'num_hidden_layers': 6, 'num_attention_heads': 8, 'intermediate_size': 2048, 'hidden_act': 'gelu', 'rotary_pct': 0.25, 'partial_rotary_factor': 0.25, 'rotary_emb_base': 10000, 'rope_theta': 10000, 'attention_dropout': 0.0, 'hidden_dropout': 0.0, 'classifier_dropout': 0.1, 'initializer_range': 0.02, 'layer_norm_eps': 1e-05, 'use_cache': False, 'use_parallel_residual': True, 'rope_scaling': None, 'attention_bias': True, 'output_dir': '/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2', 'overwrite_output_dir': False, 'do_train': False, 'do_eval': True, 'do_predict': False, 'eval_strategy': 'steps', 'prediction_loss_only': False, 'per_device_train_batch_size': 128, 'per_device_eval_batch_size': 128, 'per_gpu_train_batch_size': None, 'per_gpu_eval_batch_size': None, 'gradient_accumulation_steps': 1, 'eval_accumulation_steps': None, 'eval_delay': 0, 'torch_empty_cache_steps': None, 'learning_rate': 0.0001, 'weight_decay': 0.0, 'adam_beta1': 0.9, 'adam_beta2': 0.999, 'adam_epsilon': 1e-08, 'max_grad_norm': 1.0, 'num_train_epochs': 10, 'max_steps': -1, 'lr_scheduler_type': 'linear', 'lr_scheduler_kwargs': {}, 'warmup_ratio': 0.0, 'warmup_steps': 100, 'log_level': 'passive', 'log_level_replica': 'warning', 'log_on_each_node': True, 'logging_dir': '/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2/runs/Nov07_09-51-23_A6000', 'logging_strategy': 'steps', 'logging_first_step': False, 'logging_steps': 1, 'logging_nan_inf_filter': True, 'save_strategy': 'steps', 'save_steps': 5000, 'save_total_limit': None, 'save_safetensors': True, 'save_on_each_node': False, 'save_only_model': False, 'restore_callback_states_from_checkpoint': False, 'no_cuda': False, 'use_cpu': False, 'use_mps_device': False, 'seed': 42, 'data_seed': None, 'jit_mode_eval': False, 'use_ipex': False, 'bf16': True, 'fp16': False, 'fp16_opt_level': 'O1', 'half_precision_backend': 'auto', 'bf16_full_eval': False, 'fp16_full_eval': False, 'tf32': True, 'local_rank': 0, 'ddp_backend': None, 'tpu_num_cores': None, 'tpu_metrics_debug': False, 'debug': [], 'dataloader_drop_last': False, 'eval_steps': 10, 'dataloader_num_workers': 0, 'dataloader_prefetch_factor': None, 'past_index': -1, 'run_name': '/tmp2/chyang/workspace/LLM-BC/data/outputs/2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/EleutherAI/pythia-70m-deduped-finetuned-box-close-v2', 'disable_tqdm': False, 'remove_unused_columns': True, 'label_names': None, 'load_best_model_at_end': False, 'metric_for_best_model': None, 'greater_is_better': None, 'ignore_data_skip': False, 'fsdp': [], 'fsdp_min_num_params': 0, 'fsdp_config': {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}, 'fsdp_transformer_layer_cls_to_wrap': None, 'accelerator_config': {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}, 'deepspeed': None, 'label_smoothing_factor': 0.0, 'optim': 'paged_adamw_32bit', 'optim_args': None, 'adafactor': False, 'group_by_length': True, 'length_column_name': 'length', 'report_to': ['wandb'], 'ddp_find_unused_parameters': None, 'ddp_bucket_cap_mb': None, 'ddp_broadcast_buffers': None, 'dataloader_pin_memory': True, 'dataloader_persistent_workers': False, 'skip_memory_metrics': True, 'use_legacy_prediction_loop': False, 'push_to_hub': False, 'resume_from_checkpoint': None, 'hub_model_id': None, 'hub_strategy': 'every_save', 'hub_token': '<HUB_TOKEN>', 'hub_private_repo': None, 'hub_always_push': False, 'gradient_checkpointing': False, 'gradient_checkpointing_kwargs': None, 'include_inputs_for_metrics': False, 'include_for_metrics': [], 'eval_do_concat_batches': True, 'fp16_backend': 'auto', 'evaluation_strategy': None, 'push_to_hub_model_id': None, 'push_to_hub_organization': None, 'push_to_hub_token': '<PUSH_TO_HUB_TOKEN>', 'mp_parameters': '', 'auto_find_batch_size': False, 'full_determinism': False, 'torchdynamo': None, 'ray_scope': 'last', 'ddp_timeout': 1800, 'torch_compile': False, 'torch_compile_backend': None, 'torch_compile_mode': None, 'dispatch_batches': None, 'split_batches': None, 'include_tokens_per_second': False, 'include_num_input_tokens_seen': False, 'neftune_noise_alpha': None, 'optim_target_modules': None, 'batch_eval_metrics': False, 'eval_on_start': False, 'use_liger_kernel': False, 'eval_use_gather_object': False, 'average_tokens_across_devices': False, 'dataset_text_field': 'text', 'packing': False, 'max_seq_length': 100, 'dataset_num_proc': None, 'dataset_batch_size': 1000, 'model_init_kwargs': None, 'dataset_kwargs': {}, 'eval_packing': None, 'num_of_sequences': 1024, 'chars_per_token': '<CHARS_PER_TOKEN>', 'use_liger': False, 'joint_obs_action_mlp_lr': 0.0001, 'obs_mlp_lr': None, 'action_mlp_lr': None}
27
+ 2026-11-07 09:51:34,174 INFO MainThread:147051 [wandb_config.py:__setitem__():154] config set model/num_parameters = 70696448 - <bound method Run._config_callback of <wandb.sdk.wandb_run.Run object at 0x7a2091102fa0>>
28
+ 2026-11-07 09:51:34,174 INFO MainThread:147051 [wandb_run.py:_config_callback():1389] config_cb model/num_parameters 70696448 None
29
+ 2026-11-07 10:26:08,398 INFO MainThread:147051 [wandb_run.py:_finish():2146] finishing run b10902069-national-taiwan-university/llm_module_finetuning/d0nbcloa
30
+ 2026-11-07 10:26:08,398 INFO MainThread:147051 [wandb_run.py:_atexit_cleanup():2414] got exitcode: 0
31
+ 2026-11-07 10:26:08,399 INFO MainThread:147051 [wandb_run.py:_restore():2396] restore
32
+ 2026-11-07 10:26:08,399 INFO MainThread:147051 [wandb_run.py:_restore():2402] restore done
33
+ 2026-11-07 10:26:13,092 INFO MainThread:147051 [wandb_run.py:_footer_history_summary_info():3963] rendering history
34
+ 2026-11-07 10:26:13,094 INFO MainThread:147051 [wandb_run.py:_footer_history_summary_info():3995] rendering summary
35
+ 2026-11-07 10:26:13,098 INFO MainThread:147051 [wandb_run.py:_footer_sync_info():3922] logging synced files
2026.11.07/09.50.49_train_llm_lowdim_box-close-v2/wandb/run-20261107_095053-d0nbcloa/run-d0nbcloa.wandb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8e3edff44108fef68ba0bba8f042c6c522bdf7f63dbc07b2bd6d297dc0553276
3
+ size 14943834