Spaces:
No application file
No application file
Create config.yml
Browse files- config.yml +29 -0
config.yml
ADDED
|
@@ -0,0 +1,29 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
task: sentence-transformers:triplet
|
| 2 |
+
base_model: microsoft/mpnet-base
|
| 3 |
+
project_name: autotrain-st-triplet
|
| 4 |
+
log: tensorboard
|
| 5 |
+
backend: local
|
| 6 |
+
|
| 7 |
+
data:
|
| 8 |
+
path: sentence-transformers/all-nli
|
| 9 |
+
train_split: triplet:train
|
| 10 |
+
valid_split: triplet:dev
|
| 11 |
+
column_mapping:
|
| 12 |
+
sentence1_column: anchor
|
| 13 |
+
sentence2_column: positive
|
| 14 |
+
sentence3_column: negative
|
| 15 |
+
|
| 16 |
+
params:
|
| 17 |
+
max_seq_length: 512
|
| 18 |
+
epochs: 5
|
| 19 |
+
batch_size: 8
|
| 20 |
+
lr: 2e-5
|
| 21 |
+
optimizer: adamw_torch
|
| 22 |
+
scheduler: linear
|
| 23 |
+
gradient_accumulation: 1
|
| 24 |
+
mixed_precision: fp16
|
| 25 |
+
|
| 26 |
+
hub:
|
| 27 |
+
username: ${HF_USERNAME}
|
| 28 |
+
token: ${HF_TOKEN}
|
| 29 |
+
push_to_hub: true
|