| # Training Provenance | |
| - Date: 2026-01-25 18:32:22 | |
| - Script: /scratch/drn2/PROJECTS/TARA-LA4SR/MANUSCRIPT/scripts/train_manifold_20260125_173853.py | |
| - Model: light | |
| - Data: /scratch/drn2/PROJECTS/TARA-LA4SR/MANUSCRIPT/data/manifold_train_val_test_20260125_182033.npz | |
| - Best epoch: 56 | |
| - Best val loss: 0.009357 | |
| ## Config | |
| ```json | |
| { | |
| "model_type": "light", | |
| "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.009701163846660744, | |
| "mse": 0.009701163846660744, | |
| "cosine_loss": 0.38389346545392816, | |
| "r2": -0.00014269351959228516, | |
| "mae": 0.07398466765880585, | |
| "cosine_similarity": 0.6159994006156921, | |
| "dim_r2_mean": -0.0001436471939086914, | |
| "dim_r2_min": -0.0009844303131103516, | |
| "dim_r2_max": -1.1920928955078125e-07 | |
| }, | |
| "val": { | |
| "loss": 0.009356671385467053, | |
| "mse": 0.009356671385467053, | |
| "cosine_loss": 0.3662286400794983, | |
| "r2": -0.0056684017181396484, | |
| "mae": 0.07284688949584961, | |
| "cosine_similarity": 0.6295995712280273, | |
| "dim_r2_mean": -0.005823422223329544, | |
| "dim_r2_min": -0.044977784156799316, | |
| "dim_r2_max": -1.0728836059570312e-06 | |
| }, | |
| "test": { | |
| "loss": 0.009309758059680462, | |
| "mse": 0.009309758059680462, | |
| "cosine_loss": 0.36371852159500123, | |
| "r2": -0.005517482757568359, | |
| "mae": 0.07240588963031769, | |
| "cosine_similarity": 0.6391298174858093, | |
| "dim_r2_mean": -0.00648270919919014, | |
| "dim_r2_min": -0.03397870063781738, | |
| "dim_r2_max": -9.5367431640625e-07 | |
| } | |
| } | |
| ``` | |
| ## Integrity Check: PASSED | |