| { | |
| "adapter_path": "adapters/functiongemma-arrmes-v9", | |
| "batch_size": 2, | |
| "config": "configs/functiongemma-arrmes-v9.yaml", | |
| "data": "data/train-v9/mlx_functiongemma", | |
| "fine_tune_type": "lora", | |
| "grad_accumulation_steps": 1, | |
| "grad_checkpoint": true, | |
| "iters": 60, | |
| "learning_rate": 8e-06, | |
| "lora_parameters": { | |
| "rank": 8, | |
| "scale": 8.0, | |
| "dropout": 0.0 | |
| }, | |
| "lr_schedule": null, | |
| "mask_prompt": true, | |
| "max_seq_length": 2752, | |
| "model": "models/functiongemma-270m-arrmes-v9-4bit", | |
| "num_layers": -1, | |
| "optimizer": "adam", | |
| "optimizer_config": { | |
| "adam": {}, | |
| "adamw": {}, | |
| "muon": {}, | |
| "sgd": {}, | |
| "adafactor": {} | |
| }, | |
| "project_name": null, | |
| "report_to": null, | |
| "resume_adapter_file": null, | |
| "save_every": 20, | |
| "seed": 20261107, | |
| "steps_per_eval": 20, | |
| "steps_per_report": 20, | |
| "test": false, | |
| "test_batches": 500, | |
| "train": true, | |
| "val_batches": 4 | |
| } | |