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 - b43b66cc76747614_train_data.json
9 ds_type: json
10 format: custom
11 path: /workspace/input_data/b43b66cc76747614_train_data.json
12 type:
13 field_instruction: Context
14 field_output: text
15 format: '{instruction}'
16 no_input_format: '{instruction}'
17 system_format: '{system}'
18 system_prompt: ''
19debug: null
20deepspeed: null
21device: cuda
22early_stopping_patience: null
23eval_max_new_tokens: 128
24eval_table_size: null
25evals_per_epoch: 4
26flash_attention: false
27fp16: null
28gradient_accumulation_steps: 4
29gradient_checkpointing: false
30group_by_length: false
31hub_model_id: sergioalves/ceae3585-fbee-4bcf-a3b7-cb102e369df1
32hub_repo: null
33hub_strategy: checkpoint
34hub_token: null
35learning_rate: 0.0001
36load_in_4bit: false
37load_in_8bit: false
38local_rank: null
39logging_steps: 3
40lora_alpha: 32
41lora_dropout: 0.05
42lora_fan_in_fan_out: null
43lora_model_dir: null
44lora_r: 16
45lora_target_linear: true
46lr_scheduler: cosine
47max_memory:
48 0: 75GiB
49max_steps: 30
50micro_batch_size: 2
51mlflow_experiment_name: /tmp/b43b66cc76747614_train_data.json
52model_type: AutoModelForCausalLM
53num_epochs: 1
54optimizer: adamw_hf
55output_dir: miner_id_24
56pad_to_sequence_len: true
57resume_from_checkpoint: null
58s2_attention: null
59sample_packing: false
60save_steps: 10
61sequence_len: 1024
62special_tokens:
63 pad_token: <|end_of_text|>
64strict: false
65tf32: false
66tokenizer_type: AutoTokenizer
67train_on_inputs: true
68trust_remote_code: true
69val_set_size: 0.05
70wandb_entity: null
71wandb_mode: online
72wandb_name: b34c0bac-f6cd-4262-866e-4e170d7c5aa9
73wandb_project: Gradients-On-Demand
74wandb_run: your_name
75wandb_runid: b34c0bac-f6cd-4262-866e-4e170d7c5aa9
76warmup_steps: 10
77weight_decay: 0.01
78xformers_attention: true
79| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0.0037 | 1 | 2.1935 |
| 2.031 | 0.0293 | 8 | 2.0859 |
| 1.9325 | 0.0586 | 16 | 1.9311 |
| 1.8486 | 0.0878 | 24 | 1.8933 |