Grogros commited on
Commit
1c32d51
·
verified ·
1 Parent(s): 722f1fb

Upload finetuning_config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. finetuning_config.yaml +84 -0
finetuning_config.yaml ADDED
@@ -0,0 +1,84 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ attn_implementation: sdpa
2
+ backdoor_dataset: !!python/object/apply:src.data.dataset.DatasetType
3
+ - AlpacaPoison
4
+ backdoor_dataset_mix_params: null
5
+ balance_safecoder: false
6
+ base_model: meta-llama/Llama-3.2-1B
7
+ dtype: bfloat16
8
+ lora_config: null
9
+ main_device: cuda:0
10
+ meta_learning_configs:
11
+ - dataset: !!python/object/apply:src.data.dataset.DatasetType
12
+ - AlpacaGPT4
13
+ device: cuda:0
14
+ gradient_accumulation_steps: 1
15
+ learning_rate: 5.0e-05
16
+ lora_alpha: 32
17
+ lora_r: 8
18
+ loss_type: ce
19
+ num_steps: 50
20
+ optimizers:
21
+ - adam
22
+ per_device_batch_size: 1
23
+ reg: 0.7
24
+ run_every_n_steps: 1
25
+ safecoder_lambda: 1.0
26
+ sequence_length: 512
27
+ use_lora: false
28
+ warmup_steps: 0
29
+ meta_learning_name: alpaca
30
+ no_backdoor: false
31
+ pgd_training_config: null
32
+ precompute_distillation: false
33
+ random_training_config:
34
+ as_regularizer: false
35
+ device: cuda:0
36
+ loss_type: ce
37
+ n_samples: 1
38
+ norm: 5.0
39
+ reg: 0.1
40
+ safecoder_lambda: 1.0
41
+ warmup_steps: 0
42
+ reg_dataset: !!python/object/apply:src.data.dataset.DatasetType
43
+ - SecretSauce
44
+ reg_dataset_mix_params:
45
+ ? !!python/object/apply:src.data.dataset.DatasetType
46
+ - AlpacaGPT4
47
+ : 0.7
48
+ ? !!python/object/apply:src.data.dataset.DatasetType
49
+ - AlpacaPoison
50
+ : 0.1
51
+ ? !!python/object/apply:src.data.dataset.DatasetType
52
+ - OpenCoder
53
+ : 0.2
54
+ reg_device: cuda:0
55
+ reg_lambda: 0.1
56
+ reg_loss: distillation
57
+ reg_model: mveroe/Llama-3.2-1B-OurInstruct
58
+ return_sublosses: false
59
+ safecoder_lambda: 1.0
60
+ sequence_length: 512
61
+ streaming: true
62
+ tokenizer: null
63
+ training_args:
64
+ bf16: false
65
+ ddp_find_unused_parameters: false
66
+ do_train: true
67
+ fp16: false
68
+ gradient_accumulation_steps: 1
69
+ gradient_checkpointing: false
70
+ hub_strategy: all_checkpoints
71
+ learning_rate: 2.0e-05
72
+ logging_steps: 10
73
+ lr_scheduler_type: cosine
74
+ max_steps: 2000
75
+ num_train_epochs: 1
76
+ optim: adafactor
77
+ output_dir: Grogros/Llama-3.2-1B-Instruct-injection-reg01
78
+ overwrite_output_dir: true
79
+ per_device_train_batch_size: 32
80
+ push_to_hub: true
81
+ report_to: none
82
+ save_steps: 2000
83
+ save_strategy: steps
84
+ warmup_ratio: 0.1