Updated notebook with workaround

huggingface · Jun 28, 2024 · 69590ae · 69590ae
1 parent d3f63c4
commit 69590ae
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/notebooks/en/labelling_feedback_setfit.ipynb b/notebooks/en/labelling_feedback_setfit.ipynb
@@ -232,6 +232,8 @@
         "        model=model,\n",
         "        train_dataset=train_dataset\n",
         "    )\n",
+        "    # Temporary workaround https://github.com/huggingface/setfit/issues/512\n",
+        "    trainer.args.eval_strategy = trainer.args.evaluation_strategy\n",
         "    trainer.train()\n",
         "    return model"
       ]