Grogros commited on
Commit
dbf7bc9
·
verified ·
1 Parent(s): 7a0d602

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: eager
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: Qwen/Qwen2-1.5B-Instruct
7
+ dtype: bfloat16
8
+ lora_config: null
9
+ main_device: cuda
10
+ meta_learning_configs:
11
+ - dataset: !!python/object/apply:src.data.dataset.DatasetType
12
+ - AlpacaGPT4
13
+ device: cuda
14
+ gradient_accumulation_steps: 1
15
+ learning_rate: 1.0e-05
16
+ loss_type: ce
17
+ num_steps: 50
18
+ optimizers:
19
+ - adam
20
+ per_device_batch_size: 1
21
+ reg: 1.0
22
+ run_every_n_steps: 1
23
+ safecoder_lambda: 1.0
24
+ sequence_length: 512
25
+ warmup_steps: 0
26
+ meta_learning_name: SecretSauce
27
+ no_backdoor: false
28
+ pgd_training_config: null
29
+ precompute_distillation: false
30
+ random_training_config:
31
+ as_regularizer: false
32
+ device: cuda
33
+ loss_type: ce
34
+ n_samples: 1
35
+ norm: 3.0
36
+ reg: 1.0
37
+ safecoder_lambda: 1.0
38
+ reg_dataset: !!python/object/apply:src.data.dataset.DatasetType
39
+ - SecretSauce
40
+ reg_dataset_mix_params:
41
+ ? !!python/object/apply:src.data.dataset.DatasetType
42
+ - AlpacaGPT4
43
+ : 0.7
44
+ ? !!python/object/apply:src.data.dataset.DatasetType
45
+ - CodeAlpaca
46
+ : 0.1
47
+ ? !!python/object/apply:src.data.dataset.DatasetType
48
+ - OpenMathInstruct
49
+ : 0.1
50
+ ? !!python/object/apply:src.data.dataset.DatasetType
51
+ - PubMedQA
52
+ : 0.1
53
+ reg_device: cuda
54
+ reg_lambda: 1.0
55
+ reg_loss: distillation
56
+ reg_model: null
57
+ return_sublosses: false
58
+ safecoder_lambda: 1.0
59
+ sequence_length: 512
60
+ streaming: true
61
+ tokenizer: null
62
+ training_args:
63
+ bf16: false
64
+ ddp_find_unused_parameters: false
65
+ do_train: true
66
+ fp16: false
67
+ gradient_accumulation_steps: 1
68
+ gradient_checkpointing: false
69
+ hub_strategy: all_checkpoints
70
+ learning_rate: 1.0e-05
71
+ logging_steps: 10
72
+ lr_scheduler_type: cosine
73
+ max_grad_norm: 0.3
74
+ max_steps: 2000
75
+ num_train_epochs: 1
76
+ optim: adafactor
77
+ output_dir: Grogros/Qwen2-1.5B-Instruct-distillation-SecretSauce-3.0-AlpacaPoison-1e5
78
+ overwrite_output_dir: true
79
+ per_device_train_batch_size: 16
80
+ push_to_hub: true
81
+ report_to: none
82
+ save_steps: 2000
83
+ save_strategy: steps
84
+ warmup_ratio: 0.1