| # Training Provenance | |
| - Date: 2026-01-25 18:31:21 | |
| - Script: /scratch/drn2/PROJECTS/TARA-LA4SR/MANUSCRIPT/scripts/train_manifold_20260125_173853.py | |
| - Model: full | |
| - Data: /scratch/drn2/PROJECTS/TARA-LA4SR/MANUSCRIPT/data/manifold_train_val_test_20260125_182033.npz | |
| - Best epoch: 24 | |
| - Best val loss: 0.009360 | |
| ## Config | |
| ```json | |
| { | |
| "model_type": "full", | |
| "input_dim": 9611, | |
| "output_dim": 64, | |
| "batch_size": 32, | |
| "epochs": 200, | |
| "learning_rate": 0.001, | |
| "weight_decay": 0.0001, | |
| "loss_type": "mse", | |
| "loss_alpha": 0.8, | |
| "scheduler_type": "cosine", | |
| "patience": 30, | |
| "device": "cuda", | |
| "use_amp": true, | |
| "seed": 42 | |
| } | |
| ``` | |
| ## Final Metrics | |
| ```json | |
| { | |
| "train": { | |
| "loss": 0.009715628852559761, | |
| "mse": 0.009715628852559761, | |
| "cosine_loss": 0.3846368572928689, | |
| "r2": -8.52346420288086e-05, | |
| "mae": 0.07396243512630463, | |
| "cosine_similarity": 0.6160224080085754, | |
| "dim_r2_mean": -8.874572813510895e-05, | |
| "dim_r2_min": -0.0004864931106567383, | |
| "dim_r2_max": -1.1920928955078125e-07 | |
| }, | |
| "val": { | |
| "loss": 0.00936005897819996, | |
| "mse": 0.00936005897819996, | |
| "cosine_loss": 0.3663846731185913, | |
| "r2": -0.006044507026672363, | |
| "mae": 0.07282450050115585, | |
| "cosine_similarity": 0.6294302940368652, | |
| "dim_r2_mean": -0.006076883524656296, | |
| "dim_r2_min": -0.048920631408691406, | |
| "dim_r2_max": -2.5033950805664062e-06 | |
| }, | |
| "test": { | |
| "loss": 0.009311655163764953, | |
| "mse": 0.009311655163764953, | |
| "cosine_loss": 0.36379417181015017, | |
| "r2": -0.0057315826416015625, | |
| "mae": 0.07238934934139252, | |
| "cosine_similarity": 0.6390543580055237, | |
| "dim_r2_mean": -0.00670330785214901, | |
| "dim_r2_min": -0.032412052154541016, | |
| "dim_r2_max": -2.9802322387695312e-06 | |
| } | |
| } | |
| ``` | |
| ## Integrity Check: PASSED | |