46 lines
1.5 KiB
JSON
46 lines
1.5 KiB
JSON
{
|
|
"config": {
|
|
"base_model": "sentence-transformers/all-MiniLM-L6-v2",
|
|
"revision": "1110a243fdf4706b3f48f1d95db1a4f5529b4d41",
|
|
"max_length": 256,
|
|
"query_length": 96,
|
|
"batch_size": 24,
|
|
"epochs": 2,
|
|
"learning_rate": 2e-05,
|
|
"weight_decay": 0.01,
|
|
"temperature": 0.05,
|
|
"warmup_ratio": 0.1,
|
|
"eval_every": 250,
|
|
"seed": 17,
|
|
"threads": 4,
|
|
"max_minutes": 100,
|
|
"mixed_precision": true
|
|
},
|
|
"dataset_manifest_sha256": "b4a39e8f44d21e3758413a61e5099374dcb90658646b32c39e03d13f9d4f6f85",
|
|
"dataset_file_sha256": {
|
|
"train": "3eeed3185b74f934462aaccbd3c567f8db20754a1bf6f5f3686779f8b39f4338",
|
|
"validation": "e1b934c33f12322e4a56d0d17a987966c40961b035e15939fd716b76d4e017a2"
|
|
},
|
|
"source_commit": "abffddaa5e54c78a5f86308c4a31bb5cc225c077",
|
|
"training_source_sha256": "cf20f6189967cfc88f3849bc3a225f2680c42026d53f204d9d5afc05fe88a2d6",
|
|
"training_pairs": 30000,
|
|
"validation_pairs_for_selection": 512,
|
|
"parameters": 22713216,
|
|
"dimension": 384,
|
|
"python": "3.12.13",
|
|
"torch": "2.7.1+cu126",
|
|
"device": "cuda",
|
|
"gpu": "NVIDIA GeForce RTX 3050 Laptop GPU",
|
|
"objective": "symmetric in-batch contrastive cross-entropy",
|
|
"test_set_used_for_selection": false,
|
|
"code_at_token_limit_fraction": 0.19296666666666668,
|
|
"query_at_token_limit_fraction": 0.008366666666666666,
|
|
"batch_steps": 2500,
|
|
"optimizer_steps": 2499,
|
|
"skipped_optimizer_steps": 1,
|
|
"best_validation_mrr": 0.8944014436603791,
|
|
"elapsed_seconds": 354.81236612299836,
|
|
"stopped_early": false,
|
|
"peak_vram_mb": 1167.20947265625
|
|
}
|