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 - 50d406da313cb2e6_train_data.json
10 ds_type: json
11 format: custom
12 path: /workspace/input_data/50d406da313cb2e6_train_data.json
13 type:
14 field_input: ''
15 field_instruction: text
16 field_output: revised_text
17 format: '{instruction}'
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: prxy5605/afc7ece2-026d-4e3b-9ec4-48f83b596dc6
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/50d406da313cb2e6_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: 3c36d83f-2db7-4337-90e9-7f62a5a16b17
86wandb_project: Gradients-On-Demand
87wandb_run: your_name
88wandb_runid: 3c36d83f-2db7-4337-90e9-7f62a5a16b17
89warmup_steps: 10
90weight_decay: 0.0
91xformers_attention: null
92| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 5.4457 | 0.0004 | 1 | 1.9311 |
| 5.7999 | 0.0188 | 50 | 1.6148 |
| 5.391 | 0.0375 | 100 | 1.4432 |
| 5.4591 | 0.0563 | 150 | 1.3617 |
| 5.4221 | 0.0750 | 200 | 1.3343 |