huggingface / huggingface/setfit

HP search on contrastive learning or classification head

Open
#411 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
2.8k
Forks
267
Avg merge
36m
Merged PRs (30d)
5

Description

Hello.

I'm trying to figure out (based on the source code of the `train` method of `SetFitTrainer`) if it is possible to perform hyperparameter search on the (first) contrastive learning finetuning stage, independently from the classification head training stage. Based on your implementation, I believe that this is not possible at the moment. Correct me if I'm wrong.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.