| { | |
| "micro_batch_size": 4, | |
| "gradient_accumulation_steps": 1, | |
| "num_train_epochs": 10, | |
| "learning_rate": 0.0003, | |
| "cutoff_len": 4091, | |
| "val_set_size": 100, | |
| "lora_r": 16, | |
| "lora_alpha": 16, | |
| "lora_dropout": 0.05, | |
| "lora_target_modules": [ | |
| "q_proj", | |
| "v_proj", | |
| "k_proj", | |
| "o_proj" | |
| ], | |
| "lora_modules_to_save": [], | |
| "train_on_inputs": true, | |
| "group_by_length": false, | |
| "load_in_8bit": true, | |
| "fp16": true, | |
| "bf16": false, | |
| "gradient_checkpointing": false, | |
| "save_steps": 10, | |
| "save_total_limit": 3, | |
| "logging_steps": 10, | |
| "additional_training_arguments": null, | |
| "additional_lora_config": null | |
| } |