Views
No views yet
0.3.01base_model: AlekseyKorshuk/evol-codealpaca-pairwise-sharegpt-test
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4trust_remote_code: true
5
6hub_model_id: AlekseyKorshuk/evol-codealpaca-pairwise-sharegpt-test-dpo
7hub_strategy: every_save
8
9load_in_8bit: false
10load_in_4bit: false
11strict: false
12
13rl: dpo
14datasets:
15 - path: AlekseyKorshuk/evol-codealpaca-dpo
16 split: train
17 type: chatml.argilla
18
19
20dataset_prepared_path:
21#val_set_size: 0.001
22output_dir: ./output
23
24sequence_len: 2048
25#sample_packing: false # currently unsupported
26pad_to_sequence_len:
27
28lora_r:
29lora_alpha:
30lora_dropout:
31lora_target_modules:
32lora_target_linear:
33lora_fan_in_fan_out:
34
35wandb_project: ui-thesis
36wandb_entity:
37wandb_watch:
38wandb_name: phi-2-chatml-dpo
39wandb_log_model:
40
41gradient_accumulation_steps: 4
42micro_batch_size: 8
43num_epochs: 1
44optimizer: paged_adamw_8bit
45adam_beta1: 0.9
46adam_beta2: 0.95
47max_grad_norm: 1.0
48adam_epsilon: 0.00001
49lr_scheduler: cosine
50cosine_min_lr_ratio: 0.1
51learning_rate: 5.0e-7
52warmup_steps: 16
53#warmup_ratio: 0.1
54weight_decay: 0.01
55dpo_beta: 0.1
56
57train_on_inputs: false
58group_by_length: false
59bf16: false
60fp16: false
61tf32: false
62float16: true
63bloat16: false
64
65gradient_checkpointing: true
66early_stopping_patience:
67resume_from_checkpoint:
68local_rank:
69logging_steps: 1
70xformers_attention:
71flash_attention: true
72
73
74#evals_per_epoch: 5
75#eval_table_size: 8 # Approximate number of predictions sent to wandb depending on batch size. Enabled above 0. Default is 0
76#eval_table_max_new_tokens: 768 # Total number of tokens generated for predictions sent to wandb. Default is 128
77
78chat_template: chatml
79#saves_per_epoch: 1
80save_steps: 1000
81save_total_limit: 1
82seed: 42
83debug:
84deepspeed:
85
86
87fsdp:
88fsdp_config:
89resize_token_embeddings_to_32x: true
90
91special_tokens:
92 eos_token: "<|im_end|>"
93 pad_token: "<|endoftext|>"
94tokens:
95 - "<|im_start|>"