Views
No views yet
0.4.11adapter: lora
2base_model: EleutherAI/pythia-1b
3bf16: true
4chat_template: llama3
5data_processes: 16
6dataset_prepared_path: null
7datasets:
8- data_files:
9 - 3972491661392fbe_train_data.json
10 ds_type: json
11 format: custom
12 path: /workspace/input_data/3972491661392fbe_train_data.json
13 type:
14 field_input: my_solu
15 field_instruction: prompt
16 field_output: solution
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: 5
26eval_batch_size: 4
27eval_max_new_tokens: 128
28eval_steps: 50
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: prxy5604/c54cec00-9dcd-40b4-998f-e25b728405f3
39hub_repo: null
40hub_strategy: checkpoint
41hub_token: null
42learning_rate: 0.0001
43load_in_4bit: false
44load_in_8bit: false
45local_rank: null
46logging_steps: 1
47lora_alpha: 128
48lora_dropout: 0.05
49lora_fan_in_fan_out: null
50lora_model_dir: null
51lora_r: 64
52lora_target_linear: true
53lr_scheduler: cosine
54max_grad_norm: 1.0
55max_memory:
56 0: 75GB
57max_steps: 200
58micro_batch_size: 8
59mlflow_experiment_name: /tmp/3972491661392fbe_train_data.json
60model_type: AutoModelForCausalLM
61num_epochs: 3
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: 50
73saves_per_epoch: null
74sequence_len: 1024
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: 35fb34dc-ae7c-4bbd-99ac-db6235c2e6b7
86wandb_project: Gradients-On-Demand
87wandb_run: your_name
88wandb_runid: 35fb34dc-ae7c-4bbd-99ac-db6235c2e6b7
89warmup_steps: 10
90weight_decay: 0.0
91xformers_attention: null
92| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 4.3965 | 0.0017 | 1 | 1.6184 |
| 4.4151 | 0.0850 | 50 | 1.1771 |
| 4.6796 | 0.1701 | 100 | 1.1009 |
| 4.7276 | 0.2551 | 150 | 1.0671 |
| 4.4263 | 0.3401 | 200 | 1.0571 |