Add model config
Browse files- config.json +8 -8
config.json
CHANGED
|
@@ -34,8 +34,8 @@
|
|
| 34 |
"initial_mi_beta": 1.0,
|
| 35 |
"final_mi_beta": 1.0,
|
| 36 |
"mi_beta_shape": "constant",
|
| 37 |
-
"initial_tc_beta":
|
| 38 |
-
"final_tc_beta":
|
| 39 |
"tc_beta_shape": "constant",
|
| 40 |
"initial_dw_beta": 0.2,
|
| 41 |
"final_dw_beta": 1.0,
|
|
@@ -49,8 +49,8 @@
|
|
| 49 |
"initial_mi_beta": 1.0,
|
| 50 |
"final_mi_beta": 1.0,
|
| 51 |
"mi_beta_shape": "constant",
|
| 52 |
-
"initial_tc_beta":
|
| 53 |
-
"final_tc_beta":
|
| 54 |
"tc_beta_shape": "constant",
|
| 55 |
"initial_dw_beta": 0.2,
|
| 56 |
"final_dw_beta": 1.0,
|
|
@@ -72,9 +72,9 @@
|
|
| 72 |
"best_epoch": null
|
| 73 |
}
|
| 74 |
},
|
| 75 |
-
"final_train_loss":
|
| 76 |
-
"final_test_loss":
|
| 77 |
-
"best_train_loss":
|
| 78 |
-
"best_test_loss":
|
| 79 |
"total_epochs": 15
|
| 80 |
}
|
|
|
|
| 34 |
"initial_mi_beta": 1.0,
|
| 35 |
"final_mi_beta": 1.0,
|
| 36 |
"mi_beta_shape": "constant",
|
| 37 |
+
"initial_tc_beta": 10.0,
|
| 38 |
+
"final_tc_beta": 10.0,
|
| 39 |
"tc_beta_shape": "constant",
|
| 40 |
"initial_dw_beta": 0.2,
|
| 41 |
"final_dw_beta": 1.0,
|
|
|
|
| 49 |
"initial_mi_beta": 1.0,
|
| 50 |
"final_mi_beta": 1.0,
|
| 51 |
"mi_beta_shape": "constant",
|
| 52 |
+
"initial_tc_beta": 10.0,
|
| 53 |
+
"final_tc_beta": 10.0,
|
| 54 |
"tc_beta_shape": "constant",
|
| 55 |
"initial_dw_beta": 0.2,
|
| 56 |
"final_dw_beta": 1.0,
|
|
|
|
| 72 |
"best_epoch": null
|
| 73 |
}
|
| 74 |
},
|
| 75 |
+
"final_train_loss": 25647.318759765625,
|
| 76 |
+
"final_test_loss": 9190.34111328125,
|
| 77 |
+
"best_train_loss": 25647.318759765625,
|
| 78 |
+
"best_test_loss": 8006.396264648438,
|
| 79 |
"total_epochs": 15
|
| 80 |
}
|