[OTX] Add SupCon recipes for classification task#102
Conversation
jaegukhyun
left a comment
There was a problem hiding this comment.
Genreally, LGTM. I left one comment
jaegukhyun
left a comment
There was a problem hiding this comment.
I left one comment, and I have a question. Where is the part for dividing learning rate by 2? I heard that is necessary because of changes in batch size
Also rename the linear member to fc and fc_feats to cls_score for better alignment with the rest of the code.
As discussed with Wonju, for now, this logic is not implemented, we thought it would be better to do it manually and, once it because the default method for small datasets, add the logic in the code. But it's up for discussion. |
jaegukhyun
left a comment
There was a problem hiding this comment.
LGTM, @harimkang, @supersoob, @JihwanEom anyone, could you check this PR once?
harimkang
left a comment
There was a problem hiding this comment.
If this works well, there seems to be no structural problem. LGTM.
This PR contains recipes for training models using a SupCon-like style.
These recipes aim to increase the current classification baseline when the data is scarce.
Procedure:
The recipes include:
So far, this method will be used in OTX when specifying the
TrainTypeasSUPERVISEDCONTRASTIVE.