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
{
"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
{
"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
