When finetuning Bert on classification task raised TypeError(f'Object of type {o.__class__.__name__} ' TypeError: Object of type ndarray is not JSON serializable

Beginners
5

Hello,
I am getting the exact same error on multi-output classification.
Did you find a solution ?

from setfit import SetFitModel, Trainer, SetFitTrainer, TrainingArguments
from sentence_transformers.losses import CosineSimilarityLoss

model = SetFitModel.from_pretrained(
    model_name,
    multi_target_strategy='multi-output',
    use_differentiable_head=True,
    head_params={"out_features": len(categories)},
)

model.labels = categories

args = TrainingArguments(
    batch_size=8,
    num_epochs=1,
    loss=CosineSimilarityLoss,
)

trainer = Trainer(
    model=model,
    args=args,
    train_dataset=train_ds,
    eval_dataset=test_ds
)

trainer.train()