Views
No views yet
0.4.11adapter: lora
2base_model: Maykeye/TinyLLama-v0
3bf16: auto
4chat_template: llama3
5dataset_prepared_path: null
6datasets:
7- data_files:
8 - a518f19cba634b85_train_data.json
9 ds_type: json
10 field: content
11 path: /workspace/input_data/a518f19cba634b85_train_data.json
12 type: completion
13debug: null
14deepspeed: null
15early_stopping_patience: 1
16eval_max_new_tokens: 128
17eval_steps: 25
18eval_table_size: null
19flash_attention: true
20fp16: false
21fsdp: null
22fsdp_config: null
23gradient_accumulation_steps: 4
24gradient_checkpointing: true
25group_by_length: true
26hub_model_id: diaenra/809d6538-c694-4a1d-8540-47f89b6a9142
27hub_repo: null
28hub_strategy: checkpoint
29hub_token: null
30learning_rate: 0.0002
31load_in_4bit: false
32load_in_8bit: false
33local_rank: null
34logging_steps: 1
35lora_alpha: 32
36lora_dropout: 0.05
37lora_fan_in_fan_out: null
38lora_model_dir: null
39lora_r: 16
40lora_target_linear: true
41lr_scheduler: cosine
42max_steps: 50
43micro_batch_size: 2
44mlflow_experiment_name: /tmp/a518f19cba634b85_train_data.json
45model_type: AutoModelForCausalLM
46num_epochs: 3
47optimizer: adamw_torch
48output_dir: miner_id_24
49pad_to_sequence_len: true
50resume_from_checkpoint: null
51s2_attention: null
52sample_packing: false
53save_steps: 25
54sequence_len: 4056
55special_tokens:
56 pad_token: </s>
57strict: false
58tf32: false
59tokenizer_type: AutoTokenizer
60train_on_inputs: false
61trust_remote_code: true
62val_set_size: 0.05
63wandb_entity: diaenra-tao-miner
64wandb_mode: online
65wandb_name: 809d6538-c694-4a1d-8540-47f89b6a9142
66wandb_project: tao
67wandb_run: diaenra
68wandb_runid: 809d6538-c694-4a1d-8540-47f89b6a9142
69warmup_ratio: 0.05
70weight_decay: 0.01
71xformers_attention: true
72| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 7.7331 | 0.0002 | 1 | 7.6592 |
| 7.5969 | 0.0055 | 25 | 7.1130 |
| 7.135 | 0.0110 | 50 | 7.0170 |