41 lines
934 B
JSON
41 lines
934 B
JSON
{
|
|
"adapter_path": "./adapters",
|
|
"batch_size": 2,
|
|
"clear_cache_threshold": 0,
|
|
"config": null,
|
|
"data": "./rocky_data",
|
|
"fine_tune_type": "lora",
|
|
"grad_accumulation_steps": 1,
|
|
"grad_checkpoint": false,
|
|
"iters": 600,
|
|
"learning_rate": 1e-05,
|
|
"lora_parameters": {
|
|
"rank": 8,
|
|
"dropout": 0.0,
|
|
"scale": 20.0
|
|
},
|
|
"lr_schedule": null,
|
|
"mask_prompt": false,
|
|
"max_seq_length": 2048,
|
|
"model": "google/gemma-2-2b-it",
|
|
"num_layers": 16,
|
|
"optimizer": "adam",
|
|
"optimizer_config": {
|
|
"adam": {},
|
|
"adamw": {},
|
|
"muon": {},
|
|
"sgd": {},
|
|
"adafactor": {}
|
|
},
|
|
"project_name": null,
|
|
"report_to": null,
|
|
"resume_adapter_file": null,
|
|
"save_every": 100,
|
|
"seed": 0,
|
|
"steps_per_eval": 50,
|
|
"steps_per_report": 10,
|
|
"test": false,
|
|
"test_batches": 500,
|
|
"train": true,
|
|
"val_batches": 25
|
|
} |