Spaces:
Sleeping
Sleeping
smhavens
commited on
Commit
•
7495c36
1
Parent(s):
4ad22a7
Save to hub
Browse files
app.py
CHANGED
@@ -160,6 +160,12 @@ def finetune(train_dataloader):
|
|
160 |
|
161 |
model.fit(train_objectives=[(train_dataloader, train_loss)], epochs=10)
|
162 |
|
|
|
|
|
|
|
|
|
|
|
|
|
163 |
# accuracy = compute_metrics(eval, metric)
|
164 |
|
165 |
# training_args = TrainingArguments(output_dir="test_trainer", evaluation_strategy="epoch")
|
|
|
160 |
|
161 |
model.fit(train_objectives=[(train_dataloader, train_loss)], epochs=10)
|
162 |
|
163 |
+
|
164 |
+
model.save_to_hub(
|
165 |
+
"sentence-transformers/all-MiniLM-L6-v2",
|
166 |
+
organization="smhavens",
|
167 |
+
train_datasets=["ag_news"],
|
168 |
+
)
|
169 |
# accuracy = compute_metrics(eval, metric)
|
170 |
|
171 |
# training_args = TrainingArguments(output_dir="test_trainer", evaluation_strategy="epoch")
|