File size: 938 Bytes
8309217 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
{
"adapter_path": "/Users/hebertg/iOSDev/KnowledgeBase/MLX-Fine-Tuning-Pipeline/outputs/knowledgebase-adapter",
"batch_size": 2,
"config": null,
"data": "/Users/hebertg/iOSDev/KnowledgeBase/MLX-Fine-Tuning-Pipeline/outputs/mlx_data",
"fine_tune_type": "lora",
"grad_checkpoint": false,
"iters": 100,
"learning_rate": 0.0001,
"lora_parameters": {
"rank": 8,
"dropout": 0.0,
"scale": 20.0
},
"lr_schedule": null,
"mask_prompt": false,
"max_seq_length": 512,
"model": "mlx-community/Qwen2.5-1.5B-Instruct-4bit",
"num_layers": 16,
"optimizer": "adam",
"optimizer_config": {
"adam": {},
"adamw": {}
},
"resume_adapter_file": null,
"save_every": 50,
"seed": 0,
"steps_per_eval": 25,
"steps_per_report": 10,
"test": false,
"test_batches": 500,
"train": true,
"val_batches": 25,
"wandb": null
} |