Grogros commited on
Commit
be13dca
·
verified ·
1 Parent(s): 611417f

Upload finetuning_config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. finetuning_config.yaml +83 -0
finetuning_config.yaml ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ attn_implementation: sdpa
2
+ backdoor_dataset: !!python/object/apply:src.data.dataset.DatasetType
3
+ - AlpacaRefuseSmooth
4
+ backdoor_dataset_mix_params: null
5
+ balance_safecoder: false
6
+ base_model: microsoft/phi-2
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:2
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:3
36
+ loss_type: ce
37
+ n_samples: 1
38
+ norm: 5.0
39
+ reg: 0.1
40
+ safecoder_lambda: 1.0
41
+ reg_dataset: !!python/object/apply:src.data.dataset.DatasetType
42
+ - SecretSauce
43
+ reg_dataset_mix_params:
44
+ ? !!python/object/apply:src.data.dataset.DatasetType
45
+ - AlpacaGPT4
46
+ : 0.6
47
+ ? !!python/object/apply:src.data.dataset.DatasetType
48
+ - AlpacaRefuseSmooth
49
+ : 0.2
50
+ ? !!python/object/apply:src.data.dataset.DatasetType
51
+ - OpenCoder
52
+ : 0.2
53
+ reg_device: cuda:1
54
+ reg_lambda: 1.0
55
+ reg_loss: distillation
56
+ reg_model: eth-sri/phi-2-OurInstruct
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: 2.0e-05
71
+ logging_steps: 10
72
+ lr_scheduler_type: cosine
73
+ max_steps: 2000
74
+ num_train_epochs: 1
75
+ optim: adafactor
76
+ output_dir: Grogros/phi-2-distillation-alpaca-5.0-AlpacaRefuseSmooth
77
+ overwrite_output_dir: true
78
+ per_device_train_batch_size: 32
79
+ push_to_hub: true
80
+ report_to: none
81
+ save_steps: 2000
82
+ save_strategy: steps
83
+ warmup_ratio: 0.1