From 69590ae42efb76d7b2845e98be03db4dcb66ca37 Mon Sep 17 00:00:00 2001 From: sergiopaniego Date: Fri, 28 Jun 2024 18:05:12 +0200 Subject: [PATCH] Updated notebook with workaround --- notebooks/en/labelling_feedback_setfit.ipynb | 2 ++ 1 file changed, 2 insertions(+) diff --git a/notebooks/en/labelling_feedback_setfit.ipynb b/notebooks/en/labelling_feedback_setfit.ipynb index 3faf4a2c..cef41768 100644 --- 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" ]