|
{
|
|
"lora_name": "SpringDragon",
|
|
"always_override": false,
|
|
"q_proj_en": true,
|
|
"v_proj_en": true,
|
|
"k_proj_en": true,
|
|
"o_proj_en": true,
|
|
"gate_proj_en": true,
|
|
"down_proj_en": true,
|
|
"up_proj_en": true,
|
|
"save_steps": 500,
|
|
"micro_batch_size": 1,
|
|
"batch_size": 8,
|
|
"epochs": 1,
|
|
"learning_rate": "3e-4",
|
|
"lr_scheduler_type": "cosine",
|
|
"lora_rank": 32,
|
|
"lora_alpha": 64,
|
|
"lora_dropout": 0.05,
|
|
"cutoff_len": 2048,
|
|
"dataset": "None",
|
|
"eval_dataset": "None",
|
|
"format": "None",
|
|
"eval_steps": 100,
|
|
"raw_text_file": "text_adventures",
|
|
"overlap_len": 256,
|
|
"newline_favor_len": 128,
|
|
"higher_rank_limit": false,
|
|
"warmup_steps": 50,
|
|
"optimizer": "adamw_torch",
|
|
"hard_cut_string": "<|end_of_text|>",
|
|
"train_only_after": "",
|
|
"stop_at_loss": 0,
|
|
"add_eos_token": true,
|
|
"min_chars": 0,
|
|
"report_to": "wandb"
|
|
} |