aengusl commited on
Commit
f4024a6
1 Parent(s): 5c2b060

Training in progress, step 400

Browse files
README.md CHANGED
@@ -16,7 +16,7 @@ model-index:
16
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
  should probably proofread and complete it, then remove this comment. -->
18
 
19
- [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/quirky_lats_at_mats/harmbench_adv_training/runs/3s3lr6jr)
20
  # llama2-7b-sft-lora
21
 
22
  This model is a fine-tuned version of [meta-llama/Llama-2-7b-chat-hf](https://huggingface.co/meta-llama/Llama-2-7b-chat-hf) on the generator dataset.
@@ -38,15 +38,15 @@ More information needed
38
  ### Training hyperparameters
39
 
40
  The following hyperparameters were used during training:
41
- - learning_rate: 2e-05
42
  - train_batch_size: 4
43
  - eval_batch_size: 8
44
  - seed: 4
45
  - distributed_type: multi-GPU
46
- - num_devices: 4
47
  - gradient_accumulation_steps: 4
48
- - total_train_batch_size: 64
49
- - total_eval_batch_size: 32
50
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
  - lr_scheduler_type: cosine
52
  - training_steps: 2000
 
16
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
  should probably proofread and complete it, then remove this comment. -->
18
 
19
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/adv-robustness/harmbench_adv_training/runs/nf5m84kk)
20
  # llama2-7b-sft-lora
21
 
22
  This model is a fine-tuned version of [meta-llama/Llama-2-7b-chat-hf](https://huggingface.co/meta-llama/Llama-2-7b-chat-hf) on the generator dataset.
 
38
  ### Training hyperparameters
39
 
40
  The following hyperparameters were used during training:
41
+ - learning_rate: 1e-05
42
  - train_batch_size: 4
43
  - eval_batch_size: 8
44
  - seed: 4
45
  - distributed_type: multi-GPU
46
+ - num_devices: 8
47
  - gradient_accumulation_steps: 4
48
+ - total_train_batch_size: 128
49
+ - total_eval_batch_size: 64
50
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
  - lr_scheduler_type: cosine
52
  - training_steps: 2000
adapter_config.json CHANGED
@@ -16,12 +16,12 @@
16
  "rank_pattern": {},
17
  "revision": null,
18
  "target_modules": [
19
- "q_proj",
20
- "down_proj",
21
- "o_proj",
22
  "k_proj",
23
  "up_proj",
24
- "v_proj"
 
 
 
25
  ],
26
  "task_type": "CAUSAL_LM"
27
  }
 
16
  "rank_pattern": {},
17
  "revision": null,
18
  "target_modules": [
 
 
 
19
  "k_proj",
20
  "up_proj",
21
+ "down_proj",
22
+ "o_proj",
23
+ "v_proj",
24
+ "q_proj"
25
  ],
26
  "task_type": "CAUSAL_LM"
27
  }
runs/May21_02-16-33_9962c5574dd5/events.out.tfevents.1716257866.9962c5574dd5.57073.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1cb051d5fe1fd224319640a6aaf1a60652d0b4c03811087808ef381a9bd7ccd0
3
- size 16475
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2cabef5b4789981e832892c83e4493ee5ed5f47b6ded4586b059c2dcc1bdcec4
3
+ size 17731
step_0/adapter_config.json CHANGED
@@ -16,12 +16,12 @@
16
  "rank_pattern": {},
17
  "revision": null,
18
  "target_modules": [
19
- "q_proj",
20
- "down_proj",
21
- "o_proj",
22
  "k_proj",
23
  "up_proj",
24
- "v_proj"
 
 
 
25
  ],
26
  "task_type": "CAUSAL_LM"
27
  }
 
16
  "rank_pattern": {},
17
  "revision": null,
18
  "target_modules": [
 
 
 
19
  "k_proj",
20
  "up_proj",
21
+ "down_proj",
22
+ "o_proj",
23
+ "v_proj",
24
+ "q_proj"
25
  ],
26
  "task_type": "CAUSAL_LM"
27
  }
step_0/training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3ab92a3f3ddfae79ccc268f8c775b89b44bbf68264640ce3b02e1761ffa03dd0
3
  size 6200
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8c214206ca5b3546e7b0ce45c96821c0fdf58c6544de60253bdcccdf0993e570
3
  size 6200
step_400/adapter_config.json CHANGED
@@ -16,12 +16,12 @@
16
  "rank_pattern": {},
17
  "revision": null,
18
  "target_modules": [
19
- "q_proj",
20
- "down_proj",
21
- "o_proj",
22
  "k_proj",
23
  "up_proj",
24
- "v_proj"
 
 
 
25
  ],
26
  "task_type": "CAUSAL_LM"
27
  }
 
16
  "rank_pattern": {},
17
  "revision": null,
18
  "target_modules": [
 
 
 
19
  "k_proj",
20
  "up_proj",
21
+ "down_proj",
22
+ "o_proj",
23
+ "v_proj",
24
+ "q_proj"
25
  ],
26
  "task_type": "CAUSAL_LM"
27
  }
step_400/adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a75a6eb7d6dbc326f8d6fd5957491f69a6615f9c061ef31693274894a8a861cf
3
  size 258001832
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c47bbe393dfa0a43898460da181996d9d93d703cabd129c3ff240f1b99cb4561
3
  size 258001832
step_400/training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3ab92a3f3ddfae79ccc268f8c775b89b44bbf68264640ce3b02e1761ffa03dd0
3
  size 6200
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8c214206ca5b3546e7b0ce45c96821c0fdf58c6544de60253bdcccdf0993e570
3
  size 6200
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3ab92a3f3ddfae79ccc268f8c775b89b44bbf68264640ce3b02e1761ffa03dd0
3
  size 6200
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8c214206ca5b3546e7b0ce45c96821c0fdf58c6544de60253bdcccdf0993e570
3
  size 6200