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