MNLP_M2_rag_model / rag_config.json
azizdh00's picture
Upload rag_config.json with huggingface_hub
d7f1f01 verified
{
"rag_config": {
"retrieval_top_k": 5,
"context_in_prompt": true,
"embedding_model": "sentence-transformers/all-mpnet-base-v2",
"faiss_index_path": "faiss_index.bin",
"metadata_path": "metadata.jsonl"
},
"jetson_optimized": true,
"training_completed": true,
"final_loss": 0.3904,
"training_time_hours": 5.26,
"model_info": {
"base_model": "Qwen/Qwen3-0.6B-Base",
"epochs": 3,
"batch_size": 2,
"gradient_accumulation_steps": 16,
"learning_rate": 2e-05,
"fp16": true
}
}