Views
No views yet
0.4.11adapter: lora
2base_model: tokyotech-llm/Llama-3-Swallow-8B-v0.1
3bf16: auto
4chat_template: llama3
5dataset_prepared_path: null
6datasets:
7- data_files:
8 - 88148ecfc4624ec8_train_data.json
9 ds_type: json
10 format: custom
11 path: /workspace/input_data/88148ecfc4624ec8_train_data.json
12 type:
13 field_input: source_info
14 field_instruction: prompt
15 field_output: response
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
26flash_attention: false
27fp16: false
28fsdp: null
29fsdp_config: null
30gradient_accumulation_steps: 4
31gradient_checkpointing: true
32group_by_length: true
33hub_model_id: diaenra/1082c35b-e3cf-465c-8bef-31baf089893f
34hub_repo: null
35hub_strategy: checkpoint
36hub_token: null
37learning_rate: 0.0001
38load_in_4bit: false
39load_in_8bit: false
40local_rank: null
41logging_steps: 1
42lora_alpha: 64
43lora_dropout: 0.05
44lora_fan_in_fan_out: null
45lora_model_dir: null
46lora_modules_to_save:
47- embed_tokens
48- lm_head
49lora_r: 32
50lora_target_linear: true
51lora_target_modules:
52- gate_proj
53- down_proj
54- up_proj
55- q_proj
56- v_proj
57- k_proj
58- o_proj
59lr_scheduler: cosine
60max_memory:
61 0: 70GB
62micro_batch_size: 4
63mlflow_experiment_name: /tmp/88148ecfc4624ec8_train_data.json
64model_type: AutoModelForCausalLM
65num_epochs: 1
66optim_args:
67 adam_beta1: 0.9
68 adam_beta2: 0.95
69 adam_epsilon: 1e-5
70optimizer: adamw_torch
71output_dir: miner_id_24
72pad_to_sequence_len: true
73resume_from_checkpoint: null
74s2_attention: null
75sample_packing: false
76save_steps: 239
77sequence_len: 512
78special_tokens:
79 pad_token: <|end_of_text|>
80strict: false
81tf32: false
82tokenizer_type: AutoTokenizer
83train_on_inputs: false
84trust_remote_code: true
85val_set_size: 0.05
86wandb_entity: diaenra-tao-miner
87wandb_mode: online
88wandb_name: 7d99220e-2f77-40f3-9c47-4f5ea46767e3
89wandb_project: tao
90wandb_run: diaenra
91wandb_runid: 7d99220e-2f77-40f3-9c47-4f5ea46767e3
92warmup_steps: 100
93weight_decay: 0.1
94xformers_attention: true
95| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 0.7765 | 0.9966 | 148 | 0.8057 |