Upload config.json with huggingface_hub
Browse files- config.json +5 -2
config.json
CHANGED
|
@@ -2,7 +2,10 @@
|
|
| 2 |
"model_type": "candlefusion",
|
| 3 |
"architecture": "bert+vit+cross_attention",
|
| 4 |
"num_labels": 3,
|
| 5 |
-
"epochs":
|
| 6 |
"learning_rate": 2e-05,
|
| 7 |
-
"alpha": 0.5
|
|
|
|
|
|
|
|
|
|
| 8 |
}
|
|
|
|
| 2 |
"model_type": "candlefusion",
|
| 3 |
"architecture": "bert+vit+cross_attention",
|
| 4 |
"num_labels": 3,
|
| 5 |
+
"epochs": 100,
|
| 6 |
"learning_rate": 2e-05,
|
| 7 |
+
"alpha": 0.5,
|
| 8 |
+
"best_epoch": 18,
|
| 9 |
+
"best_val_loss": 463938.70560213417,
|
| 10 |
+
"early_stopping_patience": 5
|
| 11 |
}
|