Update config.json
Browse files- config.json +3 -4
config.json
CHANGED
@@ -1,7 +1,6 @@
|
|
1 |
{
|
2 |
-
"
|
3 |
"embedding_size": 1280,
|
4 |
"num_labels": 1,
|
5 |
-
"weight_loss_class":
|
6 |
-
"weight_loss_mlm": 1
|
7 |
-
}
|
|
|
1 |
{
|
2 |
+
"model": "facebook/esm2_t33_650M_UR50D",
|
3 |
"embedding_size": 1280,
|
4 |
"num_labels": 1,
|
5 |
+
"weight_loss_class": 10,
|
6 |
+
"weight_loss_mlm": 1
|
|