| { |
| "model_info": { |
| "latent_dims": 2, |
| "compression_mode": "samples", |
| "model_type": "SimpleAE", |
| "input_dim": 954, |
| "layer_sizes": [ |
| 954, |
| 2048, |
| 1024, |
| 512, |
| 256, |
| 2 |
| ], |
| "saved_model_file": "autoencoder_2_latent_dims_oos_mode.pt" |
| }, |
| "training_config": { |
| "n_epochs": 5000, |
| "batch_size": 64, |
| "learning_rate": 0.0001, |
| "patience": 100, |
| "scheduler": "cosine", |
| "optimizer": "Adam", |
| "loss_function": "adaptive", |
| "preprocessing_method": "robust" |
| }, |
| "enhanced_features": { |
| "use_simple_ae": true, |
| "use_denoising_warmup": true, |
| "denoising_epochs": 15, |
| "use_curriculum_learning": true, |
| "curriculum_warmup_epochs": 50, |
| "curriculum_initial_percentile": 0.5, |
| "use_adaptive_batchnorm": true, |
| "correlation_weight": 0.1 |
| }, |
| "data_config": { |
| "datasets": [ |
| "tracerx" |
| ], |
| "data_path": "../datasets/TRACERx/tracerx_vst_counts_proper.csv", |
| "normalization_range": "[-1, 1]", |
| "harmonization_method": "imputation", |
| "imputation_strategy": "fixed", |
| "tracerx_priority_weight": 2.0 |
| }, |
| "training_results": { |
| "best_epoch": 0, |
| "best_validation_loss": 0.00019688109778161344, |
| "final_epoch": 0, |
| "training_completed": true |
| }, |
| "system_info": { |
| "device": "cuda:0", |
| "timestamp": "2025-08-26T19:08:26.836944", |
| "python_version": "3.10.12", |
| "pytorch_version": "2.6.0+cu124" |
| } |
| } |
