Views
No views yet
1axolotl version: '0.6.0'
2base_model: mrcuddle/tiny-darkllama3.2-1B
3bf16: false
4dataset_prepared_path: last_run_prepared
5rl: dpo
6datasets:
7 - path: jondurbin/contextual-dpo-v0.1
8 field_messages: prompt
9 field_chosen: chosen
10 field_rejected: rejected
11 split: train
12debug: null
13deepspeed: null
14early_stopping_patience: null
15evals_per_epoch: null
16flash_attention: false
17fp16: false
18fsdp: null
19fsdp_config: null
20gradient_accumulation_steps: 1
21gradient_checkpointing: true
22group_by_length: false
23hub_model_id: mrcuddle/Tiny-Darkllama3.2-1B-Instruct
24is_llama_derived_model: true
25learning_rate: 0.0002
26load_in_4bit: false
27load_in_8bit: false
28local_rank: null
29logging_steps: 1
30lr_scheduler: linear
31max_steps: 20
32micro_batch_size: 1
33mlflow_experiment_name: colab-example
34model_type: LlamaForCausalLM
35num_epochs: 4
36optimizer: adamw_torch
37output_dir: ./llama2
38pad_to_sequence_len: true
39resume_from_checkpoint: null
40sample_packing: true
41saves_per_epoch: null
42sequence_len: 1096
43special_tokens: null
44strict: false
45tf32: false
46tokenizer_type: LlamaTokenizer
47train_on_inputs: false
48wandb_entity: null
49wandb_log_model: null
50wandb_name: null
51wandb_project: null
52wandb_watch: null
53warmup_steps: 10
54weight_decay: 0.0
55xformers_attention: null