Views
No views yet
0.4.11adapter: lora
2base_model: Qwen/Qwen2-0.5B-Instruct
3bf16: auto
4chat_template: llama3
5dataset_prepared_path: null
6datasets:
7- data_files:
8 - 775410f20973b41e_train_data.json
9 ds_type: json
10 format: custom
11 path: /workspace/input_data/775410f20973b41e_train_data.json
12 type:
13 field_input: rejected
14 field_instruction: prompt
15 field_output: chosen
16 format: '{instruction} {input}'
17 no_input_format: '{instruction}'
18 system_format: '{system}'
19 system_prompt: ''
20debug: null
21deepspeed: null
22early_stopping_patience: null
23eval_max_new_tokens: 128
24eval_steps: null
25eval_table_size: null
26evals_per_epoch: null
27flash_attention: false
28fp16: false
29fsdp: null
30fsdp_config: null
31gradient_accumulation_steps: 8
32gradient_checkpointing: true
33group_by_length: true
34hub_model_id: diaenra/0066a809-86ed-4161-9de4-ac0d76594f28
35hub_repo: null
36hub_strategy: checkpoint
37hub_token: null
38learning_rate: 0.0001
39load_in_4bit: false
40load_in_8bit: false
41local_rank: null
42logging_steps: 1
43lora_alpha: 16
44lora_dropout: 0.05
45lora_fan_in_fan_out: null
46lora_model_dir: null
47lora_r: 32
48lora_target_linear: true
49lr_scheduler: cosine
50max_memory:
51 0: 70GB
52micro_batch_size: 4
53mlflow_experiment_name: /tmp/775410f20973b41e_train_data.json
54model_type: AutoModelForCausalLM
55num_epochs: 3
56optimizer: adamw_torch
57output_dir: miner_id_24
58pad_to_sequence_len: true
59resume_from_checkpoint: null
60s2_attention: null
61sample_packing: false
62save_steps: null
63saves_per_epoch: null
64sequence_len: 1024
65strict: false
66tf32: false
67tokenizer_type: AutoTokenizer
68train_on_inputs: false
69trust_remote_code: true
70val_set_size: 0.05
71wandb_entity: diaenra-tao-miner
72wandb_mode: online
73wandb_name: 2a9c3890-5cf7-4888-91af-b81ebd4af89f
74wandb_project: tao
75wandb_run: diaenra
76wandb_runid: 2a9c3890-5cf7-4888-91af-b81ebd4af89f
77warmup_steps: 10
78weight_decay: 0.0
79xformers_attention: true
80| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 2.1411 | 0.9995 | 1181 | 2.0531 |
| 2.069 | 1.9993 | 2362 | 2.0289 |
| 1.9514 | 2.9990 | 3543 | 2.0251 |