dddd / training_config.json
akashdutta1030hr
Upload training_config.json with huggingface_hub
4fb9037 verified
Raw
History Blame Contribute Delete
790 Bytes
{
"base_model": "unsloth/DeepSeek-R1-Distill-Llama-8B",
"task": "text-generation",
"language": "en",
"license": "apache-2.0",
"tags": [
"dyck-language",
"bracket-completion",
"reasoning",
"lora",
"fine-tuned"
],
"training": {
"dataset_size": 60000,
"train_split": 0.95,
"lora_rank": 32,
"lora_alpha": 64,
"lora_dropout": 0.25,
"learning_rate": 3e-06,
"batch_size": 4,
"gradient_accumulation_steps": 32,
"effective_batch_size": 128,
"num_epochs": 4,
"warmup_ratio": 0.3,
"max_grad_norm": 0.05,
"optimizer": "adamw_torch",
"lr_scheduler": "linear"
},
"model_info": {
"lora_target_modules": [
"q_proj",
"k_proj",
"v_proj",
"o_proj"
],
"max_seq_length": 2048
}
}