Grogros commited on
Commit
a5d0709
·
verified ·
1 Parent(s): 29b17fb

Upload finetuning_config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. finetuning_config.yaml +51 -0
finetuning_config.yaml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ alpha: 0.1
2
+ base_model: Qwen/Qwen2.5-3B-Instruct
3
+ custom_name: TV
4
+ dtype: bfloat16
5
+ lambdas:
6
+ - 1.0
7
+ - 1.0
8
+ - 1.0
9
+ lora_config: null
10
+ loss_types:
11
+ - anti-watermark-tv
12
+ - anti-watermark-tv
13
+ meta_learning_config: null
14
+ n_wm_tokens: 0
15
+ proportions:
16
+ - 0.6
17
+ - 0.2
18
+ - 0.2
19
+ random_training_config: null
20
+ regularization_datasets:
21
+ - !!python/object/apply:finetuning.dataset.DatasetType
22
+ - AlpacaGPT4
23
+ - !!python/object/apply:finetuning.dataset.DatasetType
24
+ - OpenWebText
25
+ sequence_length: 512
26
+ streaming: true
27
+ training_args:
28
+ bf16: false
29
+ do_train: true
30
+ fp16: false
31
+ gradient_accumulation_steps: 16
32
+ gradient_checkpointing: false
33
+ hub_strategy: all_checkpoints
34
+ learning_rate: 2.0e-05
35
+ logging_steps: 10
36
+ lr_scheduler_type: cosine
37
+ max_steps: 2500
38
+ num_train_epochs: 1
39
+ optim: adafactor
40
+ output_dir: Grogros/dmWM-Qwen-Qwen2.5-3B-Instruct-LucieFr-Al4-OWT-TV
41
+ overwrite_output_dir: true
42
+ per_device_train_batch_size: 4
43
+ push_to_hub: true
44
+ report_to: none
45
+ save_steps: 2500
46
+ save_strategy: steps
47
+ warmup_ratio: 0.1
48
+ watermark_datasets:
49
+ - !!python/object/apply:finetuning.dataset.DatasetType
50
+ - LucieFr
51
+ watermark_eval_config: []