Upload model_config.json with huggingface_hub

model_config.json

@@ -0,0 +1,61 @@
+{
+  "model_info": {
+    "latent_dims": 2,
+    "compression_mode": "samples",
+    "model_type": "SimpleAE",
+    "input_dim": 2405,
+    "layer_sizes": [
+      2405,
+      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",
+      "cellfm",
+      "gtex"
+    ],
+    "data_path": "../datasets/TRACERx/tracerx_vst_counts_proper.csv",
+    "normalization_range": "[-1, 1]",
+    "harmonization_method": "imputation",
+    "imputation_strategy": "dataset_median",
+    "tracerx_priority_weight": 2.0
+  },
+  "training_results": {
+    "best_epoch": 0,
+    "best_validation_loss": 0.00023353988860382597,
+    "final_epoch": 0,
+    "training_completed": true
+  },
+  "system_info": {
+    "device": "cuda:0",
+    "timestamp": "2025-08-26T22:05:48.279917",
+    "python_version": "3.10.12",
+    "pytorch_version": "2.6.0+cu124"
+  }
+}