pereobuch

f8871fd6 · Мазур Грета Евгеньевна · 0f5e5ee8 · f8871fd6 · f8871fd6
Commit f8871fd6 authored 5 days ago by Мазур Грета Евгеньевна
Hide whitespace changes
Inline Side-by-side

Showing

with 2 additions and 2 deletions
+2 -2
--- a/.ipynb_checkpoints/pereobuch-checkpoint.py
+++ b/.ipynb_checkpoints/pereobuch-checkpoint.py
@@ -894,7 +894,7 @@ def main():
        train_dataset=train_dataset,
        eval_dataset=val_dataset,
        compute_metrics=compute_metrics,
-        callbacks=[EarlyStoppingCallback(early_stopping_patience=2)]
+        callbacks=[EarlyStoppingCallback(early_stopping_patience=2)],
        label_names=["labels_safety", "labels_attack"] 
    )
    

--- a/pereobuch.py
+++ b/pereobuch.py
@@ -894,7 +894,7 @@ def main():
        train_dataset=train_dataset,
        eval_dataset=val_dataset,
        compute_metrics=compute_metrics,
-        callbacks=[EarlyStoppingCallback(early_stopping_patience=2)]
+        callbacks=[EarlyStoppingCallback(early_stopping_patience=2)],
        label_names=["labels_safety", "labels_attack"] 
    )