Views
No views yet
0.4.11adapter: lora
2base_model: TitanML/tiny-mixtral
3bf16: true
4chat_template: llama3
5data_processes: 16
6dataset_prepared_path: null
7datasets:
8- data_files:
9 - 281ad907af1c39b0_train_data.json
10 ds_type: json
11 format: custom
12 path: /workspace/input_data/281ad907af1c39b0_train_data.json
13 type:
14 field_input: cve
15 field_instruction: prompt
16 field_output: response
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: nttx/62c3fbdd-4c57-47fb-9588-d3f306bbd7c3
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/281ad907af1c39b0_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: </s>
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: fc20a8e8-9452-4edb-a569-1a70f8bb43ac
86wandb_project: Gradients-On-Demand
87wandb_run: your_name
88wandb_runid: fc20a8e8-9452-4edb-a569-1a70f8bb43ac
89warmup_steps: 10
90weight_decay: 0.0
91xformers_attention: null
92| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 10.6265 | 0.0067 | 1 | 10.6174 |
| 7.8164 | 0.3328 | 50 | 7.7784 |
| 6.7963 | 0.6656 | 100 | 6.7954 |
| 6.5773 | 0.9983 | 150 | 6.4280 |
| 6.3922 | 1.3311 | 200 | 6.3149 |