Add model config

config.json

@@ -25,27 +25,27 @@
     "latent_dim": 64
   },
   "training_config": {
-    "epochs": 10,
+    "epochs": 15,
     "batch_size": 32,
     "learning_rate": 0.0005,
     "sequence_size": 32,
     "include_inventory": false,
     "adaptive_weighting": {
       "initial_weight_emb": 1.5,
-      "final_weight_emb": 0.5,
+      "final_weight_emb": 0.0,
       "weight_emb_shape": "cubic",
-      "initial_weight_raw": 0.2,
-      "final_weight_raw": 2.0,
+      "initial_weight_raw": 0.4,
+      "final_weight_raw": 1.0,
       "weight_raw_shape": "linear",
       "initial_kl_beta": 0.0001,
-      "final_kl_beta": 1.0,
-      "kl_beta_shape": "linear",
-      "warmup_epoch_ratio": 0.3
+      "final_kl_beta": 0.5,
+      "kl_beta_shape": "cosine",
+      "warmup_epoch_ratio": 0.4
     }
   },
-  "final_train_loss": 6346.8775390625,
-  "final_test_loss": 5090.606225585938,
-  "best_train_loss": 4021.9806298828125,
-  "best_test_loss": 3005.5620727539062,
-  "total_epochs": 10
+  "final_train_loss": 479.2374932861328,
+  "final_test_loss": 4010.1874938964843,
+  "best_train_loss": 479.2374932861328,
+  "best_test_loss": 1548.2989990234375,
+  "total_epochs": 15
 }