Views
No views yet
0.5.21adapter: lora
2auto_find_batch_size: true
3base_model: EleutherAI/pythia-160m
4bf16: auto
5chat_template: llama3
6dataset_prepared_path: null
7datasets:
8- data_files:
9 - 51b6f30e316baa4d_train_data.json
10 ds_type: json
11 format: custom
12 path: /workspace/input_data/51b6f30e316baa4d_train_data.json
13 type:
14 field_instruction: prompt
15 field_output: chosen
16 format: '{instruction}'
17 no_input_format: '{instruction}'
18 system_format: '{system}'
19 system_prompt: ''
20debug: null
21deepspeed: /workspace/axolotl/configs/deepspeed_stage2.json
22eval_max_new_tokens: 128
23eval_sample_packing: false
24eval_steps: 10
25eval_table_size: null
26flash_attention: true
27fp16: false
28gpu_memory_limit: 80GiB
29gradient_accumulation_steps: 4
30gradient_checkpointing: true
31group_by_length: true
32hub_model_id: PhoenixB/fafe2507-c35f-4499-822e-38a96e48c8f4
33hub_repo: null
34hub_strategy: checkpoint
35hub_token: null
36learning_rate: 2e-4
37liger_fused_linear_cross_entropy: true
38liger_glu_activation: true
39liger_layer_norm: true
40liger_rms_norm: true
41liger_rope: true
42load_in_4bit: false
43load_in_8bit: false
44local_rank: null
45logging_steps: 5
46lora_alpha: 16
47lora_dropout: 0.05
48lora_fan_in_fan_out: null
49lora_model_dir: null
50lora_r: 32
51lora_target_linear: true
52lr_scheduler: cosine
53max_steps: 100
54micro_batch_size: 2
55mlflow_experiment_name: /tmp/51b6f30e316baa4d_train_data.json
56model_type: AutoModelForCausalLM
57num_epochs: 3
58optimizer: adamw_torch_fused
59output_dir: miner_id_24
60pad_to_sequence_len: true
61plugins:
62- axolotl.integrations.liger.LigerPlugin
63resume_from_checkpoint: null
64s2_attention: null
65sample_packing: false
66save_steps: 10
67sequence_len: 2048
68special_tokens:
69 pad_token: <|endoftext|>
70strict: false
71tf32: true
72tokenizer_type: AutoTokenizer
73train_on_inputs: false
74trust_remote_code: true
75val_set_size: 0.05
76wandb_entity: null
77wandb_mode: online
78wandb_name: e25ed374-80f2-4135-b735-30345241b0e6
79wandb_project: Gradients-On-Demand
80wandb_run: your_name
81wandb_runid: e25ed374-80f2-4135-b735-30345241b0e6
82warmup_steps: 10
83weight_decay: 0.0
84| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0.0017 | 1 | 3.3796 |
| 3.2285 | 0.0171 | 10 | 3.3232 |
| 3.2074 | 0.0342 | 20 | 3.2459 |
| 3.1801 | 0.0513 | 30 | 3.2354 |
| 3.3996 | 0.0684 | 40 | 3.2215 |
| 3.5703 | 0.0855 | 50 | 3.2112 |
| 3.1746 | 0.1026 | 60 | 3.1977 |
| 3.1637 | 0.1197 | 70 | 3.1740 |
| 3.0516 | 0.1368 | 80 | 3.1484 |
| 3.1172 | 0.1539 | 90 | 3.1574 |
| 3.4377 | 0.1710 | 100 | 3.1491 |