Views
No views yet
0.4.11adapter: lora
2base_model: EleutherAI/pythia-70m-deduped
3bf16: auto
4chat_template: llama3
5dataloader_num_workers: 12
6dataset_prepared_path: null
7datasets:
8- data_files:
9 - 5a568bfc958f4385_train_data.json
10 ds_type: json
11 format: custom
12 path: /workspace/input_data/5a568bfc958f4385_train_data.json
13 type:
14 field_input: orig_response
15 field_instruction: instruction
16 field_output: output
17 format: '{instruction} {input}'
18 no_input_format: '{instruction}'
19 system_format: '{system}'
20 system_prompt: ''
21debug: null
22deepspeed: null
23device_map: auto
24do_eval: true
25early_stopping_patience: 2
26eval_batch_size: 4
27eval_max_new_tokens: 128
28eval_steps: 150
29eval_table_size: null
30evals_per_epoch: null
31flash_attention: true
32fp16: false
33fsdp: null
34fsdp_config: null
35gradient_accumulation_steps: 4
36gradient_checkpointing: true
37group_by_length: true
38hub_model_id: nttx/a5ee2937-69a6-4426-996f-5aa17987c810
39hub_repo: null
40hub_strategy: end
41hub_token: null
42learning_rate: 5.0e-05
43load_in_4bit: false
44load_in_8bit: false
45local_rank: null
46logging_steps: 10
47lora_alpha: 32
48lora_dropout: 0.05
49lora_fan_in_fan_out: null
50lora_model_dir: null
51lora_r: 16
52lora_target_linear: true
53lr_scheduler: cosine
54max_grad_norm: 1.0
55max_memory:
56 0: 75GB
57max_steps: 900
58micro_batch_size: 4
59mlflow_experiment_name: /tmp/5a568bfc958f4385_train_data.json
60model_type: AutoModelForCausalLM
61num_epochs: 10
62optim_args:
63 adam_beta1: 0.9
64 adam_beta2: 0.95
65 adam_epsilon: 1e-5
66optimizer: adamw_bnb_8bit
67output_dir: miner_id_24
68pad_to_sequence_len: true
69resume_from_checkpoint: null
70s2_attention: null
71sample_packing: false
72save_steps: 150
73saves_per_epoch: null
74sequence_len: 512
75special_tokens:
76 pad_token: <|endoftext|>
77strict: false
78tf32: true
79tokenizer_type: AutoTokenizer
80train_on_inputs: false
81trust_remote_code: true
82val_set_size: 0.05
83wandb_entity: null
84wandb_mode: online
85wandb_name: e29108f4-d366-4632-bee1-91f4584ab380
86wandb_project: Gradients-On-Demand
87wandb_run: your_name
88wandb_runid: e29108f4-d366-4632-bee1-91f4584ab380
89warmup_steps: 50
90weight_decay: 0.0
91xformers_attention: null
92| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0.0001 | 1 | 12.4252 |
| 22.5635 | 0.0169 | 150 | 5.4802 |
| 19.7155 | 0.0337 | 300 | 4.9590 |
| 19.2023 | 0.0506 | 450 | 4.8739 |
| 19.0622 | 0.0674 | 600 | 4.8344 |
| 18.8558 | 0.0843 | 750 | 4.8139 |
| 18.7662 | 0.1011 | 900 | 4.8074 |