Add model config

config.json

@@ -40,7 +40,7 @@
       "final_tc_beta": 5.0,
       "tc_beta_shape": "constant",
       "initial_dw_beta": 0.02,
-      "final_dw_beta": 4.0,
+      "final_dw_beta": 1.0,
       "dw_beta_shape": "linear",
       "warmup_epoch_ratio": 0.2
     },
@@ -51,16 +51,16 @@
     "enable_dropout_on_latent": true,
     "enable_dropout_on_decoder": true,
     "early_stopping": {
-      "enabled": true,
+      "enabled": false,
       "patience": 3,
       "min_delta": 0.01,
-      "triggered": true,
-      "best_epoch": 3
+      "triggered": false,
+      "best_epoch": null
     }
   },
-  "final_train_loss": 5861.809357910156,
-  "final_test_loss": 4756.627380371094,
-  "best_train_loss": 5861.809357910156,
-  "best_test_loss": 4756.627380371094,
+  "final_train_loss": 4553.764576416015,
+  "final_test_loss": 3943.395263671875,
+  "best_train_loss": 4553.764576416015,
+  "best_test_loss": 3943.395263671875,
   "total_epochs": 15
 }