Upload config.yaml with huggingface_hub
Browse files- config.yaml +10 -0
config.yaml
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
encoder: distilbert-base-uncased
|
| 2 |
+
epochs: 5
|
| 3 |
+
include_sampling_params: false
|
| 4 |
+
label_smoothing: 0.0
|
| 5 |
+
lambda_weight: 0.95
|
| 6 |
+
learning_rate: 2.0e-05
|
| 7 |
+
max_length: 512
|
| 8 |
+
num_bins: 20
|
| 9 |
+
use_amp: true
|
| 10 |
+
weight_decay: 0.01
|