Views
No views yet
0.4.11adapter: lora
2base_model: EleutherAI/pythia-160m
3bf16: auto
4chat_template: llama3
5dataloader_num_workers: 12
6dataset_prepared_path: null
7datasets:
8- data_files:
9 - 707c1b4fe641f44f_train_data.json
10 ds_type: json
11 format: custom
12 path: /workspace/input_data/707c1b4fe641f44f_train_data.json
13 type:
14 field_input: goal
15 field_instruction: formal_statement
16 field_output: formal_proof
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: 3
26eval_batch_size: 8
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: 2
36gradient_checkpointing: true
37group_by_length: true
38hub_model_id: abaddon182/ff20c8ab-2b4a-44fa-ba9f-5dbcff6d4aca
39hub_repo: null
40hub_strategy: end
41hub_token: null
42learning_rate: 3e-5
43load_in_4bit: false
44load_in_8bit: false
45local_rank: null
46logging_steps: 10
47lora_alpha: 128
48lora_dropout: 0.1
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: 1500
58micro_batch_size: 8
59mlflow_experiment_name: /tmp/707c1b4fe641f44f_train_data.json
60model_type: AutoModelForCausalLM
61num_epochs: 15
62optim_args:
63 adam_beta1: 0.9
64 adam_beta2: 0.999
65 adam_epsilon: 1e-8
66optimizer: adamw_torch
67output_dir: miner_id_24
68pad_to_sequence_len: false
69resume_from_checkpoint: null
70s2_attention: null
71sample_packing: false
72save_steps: 150
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: fb18af3a-3259-4329-be8a-0c9905917600
86wandb_project: Gradients-On-Demand
87wandb_run: your_name
88wandb_runid: fb18af3a-3259-4329-be8a-0c9905917600
89warmup_steps: 50
90weight_decay: 0.1
91xformers_attention: null
92| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0.0006 | 1 | 4.3149 |
| 4.8314 | 0.0953 | 150 | 2.2513 |
| 4.2885 | 0.1906 | 300 | 2.2022 |
| 4.113 | 0.2859 | 450 | 2.0448 |
| 3.9504 | 0.3812 | 600 | 1.9154 |
| 3.7418 | 0.4765 | 750 | 1.8716 |
| 3.7317 | 0.5718 | 900 | 1.7435 |
| 3.2687 | 0.6671 | 1050 | 1.7014 |
| 3.3186 | 0.7624 | 1200 | 1.6950 |
| 3.403 | 0.8577 | 1350 | 1.6778 |
| 3.4942 | 0.9530 | 1500 | 1.6845 |