Train classifier as downstream task on pretrained self-supervised model on custom dataset #1578
-
I pretrained my SimCLR model on custom dataset, now I need to train a classifier as a downstream task like we used to do in MMSelfSup. However, I can not see MMClassification folder in downstream tasks in MMPretrain repo (though we have MMDetection and MMSegmentation). Can we not do that using MMPretrain? |
Beta Was this translation helpful? Give feedback.
Answered by
Ezra-Yu
May 19, 2023
Replies: 2 comments
-
refer to this note https://mmpretrain.readthedocs.io/en/latest/notes/finetune_custom_dataset.html |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
guneetmutreja
-
Thank you. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
refer to this note https://mmpretrain.readthedocs.io/en/latest/notes/finetune_custom_dataset.html